123 lines
3.6 KiB
Python
123 lines
3.6 KiB
Python
# -*- 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
|
|
from collections import namedtuple
|
|
|
|
|
|
def showhelp():
|
|
"""
|
|
Prints the command lines switches that are valid for the program.
|
|
"""
|
|
|
|
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 print_shows_to_encode(shows):
|
|
"""
|
|
Prints he details of the shows that have been selected for encoding.
|
|
"""
|
|
|
|
for showdata in shows:
|
|
print showdata
|
|
|
|
|
|
def print_shows_to_prepare(shows):
|
|
"""
|
|
Prints he details of the shows that have been selected for preparation.
|
|
"""
|
|
|
|
for showdata in shows:
|
|
showdata.Print()
|
|
|
|
|
|
def processarguments(options):
|
|
"""
|
|
Determine the actions required from the input flags
|
|
"""
|
|
|
|
inputoptions = namedtuple("inputoptions",
|
|
"numfiles doencode readonly dolist")
|
|
|
|
for opt, arg in options:
|
|
if opt == '-h':
|
|
showhelp()
|
|
sys.exit()
|
|
elif opt == "-p":
|
|
inputoptions.doencode = False
|
|
elif opt == "-e":
|
|
inputoptions.doencode = True
|
|
elif opt == "-n":
|
|
inputoptions.numfiles = arg
|
|
elif opt == "-l":
|
|
inputoptions.readonly = True
|
|
inputoptions.dolist = True
|
|
|
|
return inputoptions
|
|
|
|
|
|
def main(argv):
|
|
"""
|
|
The main program for TVEncoder.
|
|
"""
|
|
|
|
try:
|
|
opts, args = getopt.getopt(argv, "hlpen:")
|
|
except getopt.GetoptError:
|
|
showhelp()
|
|
sys.exit(2)
|
|
|
|
inputoptions = processarguments(opts)
|
|
|
|
settings = Settings("settings.cfg")
|
|
filemanager = FileManager(settings)
|
|
|
|
if inputoptions.readonly and inputoptions.dolist:
|
|
if inputoptions.doencode:
|
|
#Generate the list of files that would be encoded
|
|
showdata = filemanager.getencodingfiles(inputoptions.readonly)
|
|
print_shows_to_encode(showdata)
|
|
else:
|
|
# Generate the list of files to process
|
|
shows = filemanager.getfilestoprepare(inputoptions.numfiles)
|
|
print "num results: {0}".format(len(shows))
|
|
print_shows_to_prepare(shows)
|
|
else:
|
|
if inputoptions.doencode:
|
|
#Encode the files and move them to their final destination
|
|
showdata = filemanager.getencodingfiles(inputoptions.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)
|
|
# TODO do something with the result
|
|
filemanager.performpostencodefileoperations(
|
|
show.inputFile, show.outputFile)
|
|
else:
|
|
# Process files for encoding
|
|
shows = filemanager.getfilestoprepare(inputoptions.numfiles)
|
|
tvdata = TVData(settings)
|
|
tvdata.PrepareEpisodes(shows)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1:])
|