recordingprocessing/libfilemanager.py

# -*- coding: utf-8 -*-
"""
Created on Fri Jul  5 14:11:31 2013

@author: shanef
"""

import glob
import libtvdatasource as TVData
import os
import shutil

#move this to settings
TVRECORDINGSDIR = "/srv/storage2/videos/TVRecordings/"

class EncodeData:
    inputFile = ''
    show = None
    outputFile = ''
    
    def ToString(self):
        return "Input: {0}/tOutput: {2}".format(self.inputFile, self.outputFile)


def __GetInputFilesToEncode(shows):
    fileList = []
    
    for show in shows:
        for r,d,f in os.walk(show.inputDirectory):
            for files in f:
                if files.endswith(".mpg"):
                    data = EncodeData()
                    data.show = show
                    data.inputFile = os.path.join(r,files)
                    fileList.append(data)
    
    return fileList

def __FindSeason(path, fileName, readOnly):
    season = "Season {0}".format(fileName[1:3])
    seasonPath = os.path.join(path, season)

    if not readOnly:    
        if not os.path.exists(seasonPath):
            os.makedirs(seasonPath)
    
    return seasonPath

def __GetEncodeOutputFile(showData, readOnly):
    inFile = os.path.basename(showData.inputFile)
    outFilename = inFile[:-3]+"mkv"
    outPath = __FindSeason(showData.show.outputDirectory, outFilename)
    showData.outputFile = os.path.join(outPath, outFilename)
    
    return showData

def GetEncodingFiles(shows, readOnly=True):
    showsData = __GetInputFilesToEncode(shows)
    for showData in showsData:
        showsData = __GetEncodeOutputFile(showData, readOnly)
    
    return showsData

def CheckFileExists(file):
    return os.path.isfile(file)
    
def __GetRecordingFile(fileName):
    return os.path.join(TVRECORDINGSDIR, os.path.dirname(fileName).split("/")[-1] + ".mpg")
    
def PerformPostEncodeFileOperations(inputFileName, outputFileName):
    shutil.rmtree(os.path.dirname(inputFileName))

    linkAddress = __GetRecordingFile(inputFileName)

    os.remove(linkAddress)

    os.symlink(outputFileName, linkAddress)

def GetFilesToPrepare(path, numberofFiles, shows):
    files = glob.glob("{0}*.mpg".format(path))
    files = sorted(files, key=os.path.getctime)
    files = filter(lambda file: not os.path.islink(file), files)
    
    #files is now a list of unprocessed files, but contains shows other than those we are interested in
    
    showsToProcess = []
    i = 0
    print "Found {0} potential files".format(len(files))
    for file in files:
        # TODO get these from settings
        #if TVData.CheckTitleIsInList('localhost', 'script', 'script', 'mythconverg', file):
        showData = TVData.RetrieveEpisodeData('192.168.0.2', 'script', 'script', 'mythconverg', file, shows, '192.168.0.2', '8081', '3678177136222bf5002be209220ccb20') # TODO all these settings need to move to settings
        if showData:
            showsToProcess.append(showData)
            i = i + 1
            if i == numberofFiles:
                return showsToProcess
    
    return showsToProcess #will reach here if there were less than numberofFiles found