interim commit

libfilemanager.py

@@ -5,3 +5,70 @@ Created on Fri Jul  5 14:11:31 2013
 @author: shanef
 """
 
+import os
+import shutil
+
+#move this to settings
+TVRECORDINGSDIR = "/srv/storage2/videos/TVRecordings/"
+
+class EncodeData:
+    inputFile = ''
+    show = None
+    outputFile = ''
+    
+    def ToString():
+        return "Input: {0}/tOutput: {2}".format(inputFile, outputFile)
+
+def __GetInputFilesToEncode(shows):
+    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)
+    
+    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)
+    
+    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)
+    
+    return showData
+
+def GetEncodingFiles(shows, readOnly=True):
+    showsData = __GetInputFilesToEncode(shows)
+    for showData in showsData:
+        showsData = __GetEncodeOutputFile(showData, readOnly)
+    
+    return showsData
+
+def CheckFileExists(file):
+    return os.path.isfile(file)
+    
+def __GetRecordingFile(fileName):
+    return os.path.join(TVRECORDINGSDIR, os.path.dirname(fileName).split("/")[-1] + ".mpg")
+    
+def PerformPostEncodeFileOperations(inputFileName, outputFileName):
+    shutil.rmtree(os.path.dirname(inputFileName))
+
+    linkAddress = __GetRecordingFile(inputFileName)
+
+    os.remove(linkAddress)
+
+    os.symlink(outputFileName, linkAddress)

libhandbrake.py

@@ -32,4 +32,5 @@ class Encoder:
                 logger.info("Handbrake completed with return code {0}".format(process.returncode))
             return process.returncode
             
-        return None
+        return None
+    

libsettings.py

@@ -5,3 +5,18 @@ Created on Fri Jul  5 20:14:15 2013
 @author: shanef
 """
 
+from libtvdatasource import TVShow
+from xml.etree import ElementTree
+
+class Settings:
+    def __init__(self, settingsFile):
+        self.shows = self.LoadSettings(settingsFile)
+    
+    def LoadSettings(source):
+        shows = []
+        settingsXml = ElementTree.parse(source).getroot()
+
+        for show in settingsXml.findall('show'):     
+            newShow = TVShow(show[0].text, show[1].text, show[2].text)
+            shows.append(newShow)
+        return shows

libtvdatasource.py

@@ -5,3 +5,10 @@ Created on Fri Jul  5 14:42:47 2013
 @author: shanef
 """
 
+class TVShow:
+    def __init__(self, episode, season, title, subtitle, description):
+        self.episode = episode
+        self.season = season
+        self.title = title
+        self.subtitle = subtitle
+        self.description = description