shane/recordingprocessing
1
0
Fork 0
Code Issues Pull Requests Releases 2 Wiki Activity

added settings file

Browse Source
This commit is contained in:
Shane Frischkorn
2013-07-08 17:21:42 +10:00
parent ccb103ad36
commit 7a34fd01c8
8 changed files with 418 additions and 230 deletions
Download Patch File Download Diff File Expand all files Collapse all files

143
libfilemanager.py
View File

@@ -6,12 +6,12 @@ Created on Fri Jul 5 14:11:31 2013
"""
import glob
import libtvdatasource as TVData
from libtvdatasource import TVData
import os
import shutil
#move this to settings
TVRECORDINGSDIR = "/srv/storage2/videos/TVRecordings/"
#TVRECORDINGSDIR = "/srv/storage2/videos/TVRecordings/"
class EncodeData:
inputFile = ''
@@ -21,80 +21,85 @@ class EncodeData:
def ToString(self):
return "Input: {0}/tOutput: {2}".format(self.inputFile, self.outputFile)
def __GetInputFilesToEncode(shows):
fileList = []
class FileManager:
def __init__(self, settings):
self.settings = settings
def __GetInputFilesToEncode(self):
fileList = []
for show in self.settings.GetShowNames():
for r,d,f in os.walk(show.GetShowInputDirectory(show)):
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
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)
def __FindSeason(self, path, fileName, readOnly):
season = "Season {0}".format(fileName[1:3])
seasonPath = os.path.join(path, season)
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)
if not readOnly:
if not os.path.exists(seasonPath):
os.makedirs(seasonPath)
return 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)
def __GetEncodeOutputFile(self, inputFile, showName, readOnly):
inFile = os.path.basename(inputFile)
outFilename = inFile[:-3]+"mkv"
outPath = self.__FindSeason(self.settings.GetShowOutputFile(showName), outFilename)
return os.path.join(outPath, outFilename)
return showData
def GetEncodingFiles(shows, readOnly=True):
showsData = __GetInputFilesToEncode(shows)
for showData in showsData:
showsData = __GetEncodeOutputFile(showData, readOnly)
def GetEncodingFiles(self, readOnly=True):
showsData = self.__GetInputFilesToEncode(self.settings.GetShowNames())
for showData in showsData:
showData.outputFile = self.__GetEncodeOutputFile(showData.inputFile, showData.name, readOnly)
return showsData
return showsData
def CheckFileExists(file):
return os.path.isfile(file)
def CheckFileExists(self, file):
return os.path.isfile(file)
def __GetRecordingFile(self, fileName):
return os.path.join(self.settings.TVRecordingDirectory, os.path.dirname(fileName).split("/")[-1] + ".mpg")
def PerformPostEncodeFileOperations(self, inputFileName, outputFileName):
shutil.rmtree(os.path.dirname(inputFileName))
def __GetRecordingFile(fileName):
return os.path.join(TVRECORDINGSDIR, os.path.dirname(fileName).split("/")[-1] + ".mpg")
linkAddress = self.__GetRecordingFile(inputFileName)
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)
os.remove(linkAddress)
#files is now a list of unprocessed files, but contains shows other than those we are interested in
os.symlink(outputFileName, linkAddress)
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 == int(numberofFiles):
return showsToProcess
return showsToProcess #will reach here if there were less than numberofFiles found
def GetFilesToPrepare(self, numberofFiles):
path = self.settings.TVRecordingDirectory()
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))
tvData = TVData(self.settings)
for file in files:
# TODO get these from settings
#if TVData.CheckTitleIsInList('localhost', 'script', 'script', 'mythconverg', file):
showData = tvData.RetrieveEpisodeData(file)
if showData:
showsToProcess.append(showData)
i = i + 1
if i == int(numberofFiles):
return showsToProcess
return showsToProcess #will reach here if there were less than numberofFiles found