X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/878c28f3ac56f6dce9647993278e36cacb994bcc..7698f560bce1d097c887cbee427e0389fa85eee6:/RecordTimer.py diff --git a/RecordTimer.py b/RecordTimer.py index cf49df23..37aecdbd 100644 --- a/RecordTimer.py +++ b/RecordTimer.py @@ -1,107 +1,161 @@ import time import codecs #from time import datetime -from Tools import Directories +from Tools import Directories, Notifications +from Components.config import config 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, begin, end, serviceref, epg, description): + def __init__(self, serviceref, begin, end, name, description, eit): timer.TimerEntry.__init__(self, int(begin), int(end)) assert isinstance(serviceref, ServiceReference) self.service_ref = serviceref - if epg is not None: - self.epg_data = "" - #str(epg.m_event_name) - else: - self.epg_data = "" + self.eit = eit + 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 = "" - if self.epg_data is not None: - description = " - " + self.epg_data - else: - description = "" - print "begin_date: " + str(begin_date) - print "service_name: " + str(service_name) - print "description: " + str(description) + 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) - # build filename from epg - - # pff das geht noch nicht... -# if epg == None: -# self.Filename = "recording.ts" -# else: -# self.Filename = "record_" + str(epg.m_event_name) + ".ts" -# -# print "------------ record filename: %s" % (self.Filename) - + 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.calculateFilename() - self.record_service = NavigationInstance.instance.recordService(self.service_ref) - if self.record_service == None: - print "timer record failed." - else: - self.record_service.prepare(self.Filename + ".ts") - f = open(self.Filename + ".ts.meta", "w") - f.write(str(self.service_ref) + "\n") - f.write(self.epg_data + "\n") - del f - - elif self.record_service == 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. + if config.recording.asktozap.value == 0: + Notifications.AddNotificationWithCallback(self.failureCB, MessageBox, _("A timer failed to record!\nDisable TV and try again?\n")) + else: # zap without asking + self.failureCB(True) elif event == self.EventStart: - self.record_service.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.record_service.stop() - self.record_service = 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." + def abort(): + # fixme + pass + + def failureCB(self, answer): + if answer == True: + #NavigationInstance.instance.stopUserServices() + print "[RecordTimer] zapping to", self.service_ref + NavigationInstance.instance.playService(self.service_ref.ref) + 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") - epgdata = xml.getAttribute("epgdata") - #filename = xml.getAttribute("filename") - return RecordTimerEntry(begin, end, serviceref, epgdata, description) + 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.Timer.__init__(self) - self.Filename = Directories.resolveFilename(Directories.SCOPE_USERETC, "timers.xml") + self.Filename = Directories.resolveFilename(Directories.SCOPE_CONFIG, "timers.xml") try: self.loadTimer() - except: + 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): - # TODO: PATH! doc = xml.dom.minidom.parse(self.Filename) @@ -116,39 +170,61 @@ class RecordTimer(timer.Timer): 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("epgdata", timer.) + 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(codecs.getwriter('UTF-8')(file)) + 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): + 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." - - self.calcNextActivation() - + + 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) def shutdown(self): self.saveTimer()