127 lines
3.8 KiB
Python
127 lines
3.8 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
|
|
import libhandbrake
|
|
from libtvdatasource import TVData
|
|
from collections import namedtuple
|
|
from termcolor import colored
|
|
|
|
|
|
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(shows, filemanager):
|
|
"""
|
|
Prints he details of the shows.
|
|
"""
|
|
|
|
existing = []
|
|
|
|
for showdata in shows:
|
|
if filemanager.checkfileexists(showdata.outputfile):
|
|
existing.append(showdata)
|
|
|
|
print showdata
|
|
|
|
if len(existing) > 0:
|
|
print colored("The following shows have existing output files that "
|
|
"need to be fixed before proceeding:\n", 'red')
|
|
|
|
for showdata in existing:
|
|
print colored(showdata, 'red')
|
|
|
|
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(showdata, filemanager)
|
|
else:
|
|
# Generate the list of files to process
|
|
shows = filemanager.getfilestoprepare(inputoptions.numfiles)
|
|
print "num results: {0}".format(len(shows))
|
|
print_shows(shows, filemanager)
|
|
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:
|
|
result = libhandbrake.encode(settings.handbrakecommand(),
|
|
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:])
|