99 lines
3.4 KiB
Python
99 lines
3.4 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
|
|
|
|
#TVRECORDINGSDIR = "/Volumes/TV Recordings/"
|
|
#TVRECORDINGSDIR = "/srv/storage2/videos/TVRecordings/" # TODO move this to settings
|
|
|
|
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
|
|
# tempShowList = ["Thomas and Friends", "Thomas the Tank Engine & Friends",
|
|
# "Chuggington", "Mike the Knight", "Octonauts",
|
|
# "The Octonauts", "In the Night Garden",
|
|
# "Raa Raa! The Noisy Lion"] # TODO get from settings
|
|
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:])
|