X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/f0273520285e28b22053eda706690774b0c8448c..eb47577c32b49a28eb7986b0fd86ed0d76207150:/RecordTimer.py diff --git a/RecordTimer.py b/RecordTimer.py index ad7fa565..f1f4e14b 100644 --- a/RecordTimer.py +++ b/RecordTimer.py @@ -1,66 +1,222 @@ -from timer import * import time +import codecs +#from time import datetime +from Tools import Directories, Notifications -class RecordTimerEntry(TimerEntry): - def __init__(self, begin, end, nav, serviceref, epg): - TimerEntry.__init__(self, begin, end) - self.ServiceRef = serviceref - self.EpgData = epg - self.Timer = None - self.Nav = nav - self.RecordService = None +import timer +import xml.dom.minidom + +from Screens.MessageBox import MessageBox +import NavigationInstance + +from Tools.XMLTools import elementsWithTag +from ServiceReference import ServiceReference + +# ok, for descriptions etc we have: +# service reference (to get the service name) +# name (title) +# description (description) +# event data (ONLY for time adjustments etc.) + + +# parses an event, and gives out a (begin, end, name, duration, eit)-tuple. +def parseEvent(ev): + name = ev.getEventName() + description = ev.getShortDescription() + begin = ev.getBeginTime() + end = begin + ev.getDuration() + eit = None + return (begin, end, name, description, eit) + +class RecordTimerEntry(timer.TimerEntry): + def __init__(self, serviceref, begin, end, name, description, eit): + timer.TimerEntry.__init__(self, int(begin), int(end)) - # build filename from epg + assert isinstance(serviceref, ServiceReference) - # pff das geht noch nicht... - if epg == None: - self.Filename = "recording.ts" - else: - self.Filename = "record_" + str(epg.m_event_name) + ".ts" + self.service_ref = serviceref + + self.eit = eit - print "------------ record filename: %s" % (self.Filename) + self.dontSave = False + self.name = name + self.description = description + self.timer = None + self.record_service = None + self.wantStart = False + self.prepareOK = False + def calculateFilename(self): + service_name = self.service_ref.getServiceName() +# begin_date = datetime.fromtimestamp(begin).strf... + begin_date = "" + + print "begin_date: ", begin_date + print "service_name: ", service_name + print "name:", self.name + print "description: ", self.description + + self.Filename = Directories.getRecordingFilename(service_name) + #begin_date + " - " + service_name + description) + + + def tryPrepare(self): + self.calculateFilename() + self.record_service = NavigationInstance.instance.recordService(self.service_ref) + if self.record_service == None: + return False + else: + if self.record_service.prepare(self.Filename + ".ts"): + self.record_service = None + return False + + f = open(self.Filename + ".ts.meta", "w") + f.write(str(self.service_ref) + "\n") + f.write(self.name + "\n") + f.write(self.description + "\n") + f.write(str(self.begin) + "\n") + del f + return True + def activate(self, event): if event == self.EventPrepare: - self.RecordService = self.Nav.recordService(self.ServiceRef) - if self.RecordService == None: - print "timer record failed." - else: - self.RecordService.prepare() - elif self.RecordService == None: - if event != self.EventAbort: - print "timer record start failed, can't finish recording." + self.prepareOK = False + if self.tryPrepare(): + self.prepareOK = True + else: + # error. + Notifications.AddNotificationWithCallback(self.failureCB, MessageBox, _("A timer failed to record!\nDisable TV and try again?\n")) elif event == self.EventStart: - self.RecordService.start() - print "timer started!" + if self.prepareOK: + self.record_service.start() + print "timer started!" + else: + print "prepare failed, thus start failed, too." + self.wantStart = True elif event == self.EventEnd or event == self.EventAbort: - self.RecordService.stop() - self.RecordService = None - print "Timer successfully ended" + self.wantStart = False + if self.prepareOK: + self.record_service.stop() + self.record_service = None + print "Timer successfully ended" + else: + print "prepare failed, thus nothing was recorded." -class RecordTimer(Timer): + def abort(): + # fixme + pass + + def failureCB(self, answer): + if answer == True: + NavigationInstance.instance.stopUserServices() + self.activate(self.EventPrepare) + if self.wantStart: + print "post-activating record" + self.activate(self.EventStart) + else: + print "user killed record" + +def createTimer(xml): + begin = int(xml.getAttribute("begin")) + end = int(xml.getAttribute("end")) + serviceref = ServiceReference(str(xml.getAttribute("serviceref"))) + description = xml.getAttribute("description").encode("utf-8") + repeated = xml.getAttribute("repeated").encode("utf-8") + eit = xml.getAttribute("eit").encode("utf-8") + name = xml.getAttribute("name").encode("utf-8") + #filename = xml.getAttribute("filename").encode("utf-8") + entry = RecordTimerEntry(serviceref, begin, end, name, description, eit) + entry.repeated = int(repeated) + return entry + +class RecordTimer(timer.Timer): def __init__(self): - Timer.__init__(self) + timer.Timer.__init__(self) + self.Filename = Directories.resolveFilename(Directories.SCOPE_CONFIG, "timers.xml") + + try: + self.loadTimer() + except IOError: + print "unable to load timers from file!" + + def isRecording(self): + isRunning = False + for timer in self.timer_list: + if timer.isRunning(): + isRunning = True + return isRunning + def loadTimer(self): - print "TODO: load timers from xml" + # TODO: PATH! + doc = xml.dom.minidom.parse(self.Filename) + + root = doc.childNodes[0] + for timer in elementsWithTag(root.childNodes, "timer"): + self.record(createTimer(timer)) def saveTimer(self): - print "TODO: save timers to xml" + doc = xml.dom.minidom.Document() + root_element = doc.createElement('timers') + doc.appendChild(root_element) + root_element.appendChild(doc.createTextNode("\n")) + + for timer in self.timer_list + self.processed_timers: + # some timers (instant records) don't want to be saved. + # skip them + if timer.dontSave: + continue + t = doc.createTextNode("\t") + root_element.appendChild(t) + t = doc.createElement('timer') + t.setAttribute("begin", str(timer.begin)) + t.setAttribute("end", str(timer.end)) + t.setAttribute("serviceref", str(timer.service_ref)) + t.setAttribute("repeated", str(timer.repeated)) + t.setAttribute("name", timer.name) + t.setAttribute("description", timer.description) + t.setAttribute("eit", str(timer.eit)) + + root_element.appendChild(t) + t = doc.createTextNode("\n") + root_element.appendChild(t) + + file = open(self.Filename, "w") + doc.writexml(file) + file.write("\n") + file.close() def record(self, entry): + print "[Timer] Record " + str(entry) entry.Timer = self self.addTimerEntry(entry) def removeEntry(self, entry): - if entry.State == TimerEntry.StateRunning: - entry.End = time.time() - print "aborting timer" - elif entry.State != TimerEntry.StateEnded: - entry.activate(TimerEntry.EventAbort) - self.TimerList.remove(entry) + print "[Timer] Remove " + str(entry) + + entry.repeated = False + + if entry.state == timer.TimerEntry.StateRunning: + print "remove running timer." + entry.end = time.time() + self.timeChanged(entry) + elif entry.state != timer.TimerEntry.StateEnded: + entry.activate(timer.TimerEntry.EventAbort) + self.timer_list.remove(entry) + + self.calcNextActivation() print "timer did not yet start - removing" + + # the timer was aborted, and removed. + return else: print "timer did already end - doing nothing." + + print "state: ", entry.state + print "in processed: ", entry in self.processed_timers + print "in running: ", entry in self.timer_list + # now the timer should be in the processed_timers list. remove it from there. + self.processed_timers.remove(entry) - self.calcNextActivation() + def shutdown(self): + self.saveTimer()