recordingprocessing/TVEncoder.py

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

@author: shanef
"""

import sys
import getopt
from libfilemanager import FileManager
from libsettings import Settings
from libhandbrake import Encoder
from libtvdatasource import TVData


def ShowHelp():
    print 'TVEncoder.py -p -n <number of files to prepare for processing> ' \
          '- prepare n recordings'
    print 'TVEncoder.py -p -l -n <number of files to process> - lists the ' \
          'files that will be processed without actually encoding them'
    print 'TVEncoder.py -e - encode the files that have been processed'
    print 'TVEncoder.py -e -l - list the files that would be encoded'


def PrintShowsToEncode(showsData):
#    print "/n".join(showData)
    for showData in showsData:
        print showData.ToString()


def PrintShowsToPrepare(showsData):
    for showData in showsData:
        showData.Print()


def main(argv):
    numFiles = 0
    doEncode = False
    readOnly = False
    doList = False

    try:
        opts, args = getopt.getopt(argv, "hlpen:")
    except getopt.GetoptError:
        ShowHelp()
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            ShowHelp()
            sys.exit()
        elif opt == "-p":
            doEncode = False
        elif opt == "-e":
            doEncode = True
        elif opt == "-n":
            numFiles = arg
        elif opt == "-l":
            readOnly = True
            doList = True

    settings = Settings("settings.cfg")

    if readOnly and doList:
        if doEncode:
            #Generate the list of files that would be encoded
            fileManager = FileManager(settings)
            showData = fileManager.GetEncodingFiles(readOnly)
            PrintShowsToEncode(showData)
        else:
            # Generate the list of files to process
            fileManager = FileManager(settings)
            shows = fileManager.GetFilesToPrepare(numFiles)
            print "num results: {0}".format(len(shows))
            PrintShowsToPrepare(shows)
    else:
        if doEncode:
            #Encode the files and move them to their final destination
            fileManager = FileManager(settings)
            showData = fileManager.GetEncodingFiles(readOnly)

            for show in showData:
                if fileManager.CheckFileExists(show.outputFile):
                    print "File {0} already exists. Cannot process." \
                        .format(show.outputFile)
                else:
                    encoder = Encoder(settings.HandbrakeCommand())
                    result = encoder.Encode(show.inputFile, show.outputFile)

                    fileManager.PerformPostEncodeFileOperations(
                        show.inputFile, show.outputFile)
        else:
            # TODO Process files for encoding
            fileManager = FileManager(settings)
            shows = fileManager.GetFilesToPrepare(numFiles)
            tvData = TVData(settings)
            tvData.PrepareEpisodes(shows)


if __name__ == "__main__":
    main(sys.argv[1:])