X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/8c446ddb6c42ea330318ae62347b4b86beeb3004..8d8279d975474b1cb8befad8d67df26de4104f20:/RecordTimer.py diff --git a/RecordTimer.py b/RecordTimer.py index 4724876a..2fb923ba 100644 --- a/RecordTimer.py +++ b/RecordTimer.py @@ -3,121 +3,213 @@ import codecs #from time import datetime from Tools import Directories, Notifications +from Components.config import config import timer import xml.dom.minidom from Screens.MessageBox import MessageBox +from Screens.SubserviceSelection import SubserviceSelection import NavigationInstance +from time import localtime -from Tools.XMLTools import elementsWithTag +from Tools.XMLTools import elementsWithTag, mergeText 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. +# begin and end will be corrected +def parseEvent(ev): + name = ev.getEventName() + description = ev.getShortDescription() + begin = ev.getBeginTime() + end = begin + ev.getDuration() + eit = ev.getEventId() + begin -= config.recording.margin_before.value[0] * 60 + end += config.recording.margin_after.value[0] * 60 + return (begin, end, name, description, eit) + +# please do not translate log messages class RecordTimerEntry(timer.TimerEntry): - def __init__(self, begin, end, serviceref, epg, description): + def __init__(self, serviceref, begin, end, name, description, eit, disabled = False): 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.disabled = disabled self.timer = None self.record_service = None + self.start_prepare = 0 + self.log_entries = [] + self.resetState() + + def log(self, code, msg): + self.log_entries.append((int(time.time()), code, msg)) + print "[TIMER]", msg + + def resetState(self): + self.state = self.StateWaiting + self.first_try_prepare = True + self.timeChanged() + 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) - self.Filename = Directories.getRecordingFilename(service_name) - #begin_date + " - " + service_name + description) + begin_date = time.strftime("%Y%m%d %H%M", time.localtime(self.begin)) - # build filename from epg + print "begin_date: ", begin_date + print "service_name: ", service_name + print "name:", self.name + print "description: ", self.description - # 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) + filename = begin_date + " - " + service_name + if self.name: + filename += " - " + self.name + + self.Filename = Directories.getRecordingFilename(filename) + self.log(0, "Filename calculated as: '%s'" % self.Filename) + #begin_date + " - " + service_name + description) def tryPrepare(self): self.calculateFilename() self.record_service = NavigationInstance.instance.recordService(self.service_ref) if self.record_service == None: + self.log(1, "'record service' failed") return False - else: - if self.record_service.prepare(self.Filename + ".ts"): + else: + event_id = self.eit + if event_id is None: + event_id = -1 + prep_res = self.record_service.prepare(self.Filename + ".ts", self.begin, self.end, event_id ) + if prep_res: + self.log(2, "'prepare' failed: error %d" % prep_res) self.record_service = None return False - f = open(self.Filename + ".ts.meta", "w") - f.write(str(self.service_ref) + "\n") - f.write(self.epg_data + "\n") - del f + self.log(3, "prepare ok, writing meta information to %s" % self.Filename) + try: + 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") + f.close() + except: + self.log(4, "failed to write meta information") return True - def activate(self, event): - if event == self.EventPrepare: - self.prepareOK = False + def do_backoff(self): + if self.backoff == 0: + self.backoff = 5 + else: + self.backoff *= 2 + if self.backoff > 100: + self.backoff = 100 + self.log(10, "backoff: retry in %d seconds" % self.backoff) + + def activate(self): + next_state = self.state + 1 + self.log(5, "activating state %d" % next_state) + + if next_state == self.StatePrepared: 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 self.record_service == None: - if event != self.EventAbort: - print "timer record start failed, can't finish recording." - elif event == self.EventStart: - if self.prepareOK: - self.record_service.start() - print "timer started!" - else: - print "prepare failed, thus start failed, too." - elif event == self.EventEnd or event == self.EventAbort: - if self.prepareOK: - self.record_service.stop() - self.record_service = None - print "Timer successfully ended" - else: - print "prepare failed, thus nothing was recorded." + self.log(6, "prepare ok, waiting for begin") + # fine. it worked, resources are allocated. + self.next_activation = self.begin + self.backoff = 0 + return True + + self.log(7, "prepare failed") + if self.first_try_prepare: + self.first_try_prepare = False + if config.recording.asktozap.value == 0: + self.log(8, "asking user to zap away") + Notifications.AddNotificationWithCallback(self.failureCB, MessageBox, _("A timer failed to record!\nDisable TV and try again?\n")) + else: # zap without asking + self.log(9, "zap without asking") + self.failureCB(True) - def abort(): - # fixme - pass + self.do_backoff() + # retry + self.start_prepare = time.time() + self.backoff + return False + elif next_state == self.StateRunning: + self.log(11, "start recording") + record_res = self.record_service.start() + + if record_res: + self.log(13, "start record returned %d" % record_res) + self.do_backoff() + # retry + self.begin = time.time() + self.backoff + return False + + return True + elif next_state == self.StateEnded: + self.log(12, "stop recording") + self.record_service.stop() + self.record_service = None + return True + + def getNextActivation(self): + if self.state == self.StateEnded: + return self.end + + next_state = self.state + 1 + + return {self.StatePrepared: self.start_prepare, + self.StateRunning: self.begin, + self.StateEnded: self.end }[next_state] def failureCB(self, answer): if answer == True: - NavigationInstance.instance.stopUserServices() - self.activate(self.EventPrepare) + self.log(13, "ok, zapped away") + #NavigationInstance.instance.stopUserServices() + NavigationInstance.instance.playService(self.service_ref.ref) else: - print "user killed record" + self.log(14, "user didn't want to zap away, record will probably fail") + + def timeChanged(self): + old_prepare = self.start_prepare + self.start_prepare = self.begin - self.prepare_time + self.backoff = 0 + + if old_prepare != self.start_prepare: + self.log(15, "record time changed, start prepare is now: %s" % time.ctime(self.start_prepare)) def createTimer(xml): begin = int(xml.getAttribute("begin")) end = int(xml.getAttribute("end")) serviceref = ServiceReference(str(xml.getAttribute("serviceref"))) - description = xml.getAttribute("description") - repeated = xml.getAttribute("repeated") - epgdata = xml.getAttribute("epgdata") - #filename = xml.getAttribute("filename") - entry = RecordTimerEntry(begin, end, serviceref, epgdata, description) + description = xml.getAttribute("description").encode("utf-8") + repeated = xml.getAttribute("repeated").encode("utf-8") + disabled = long(xml.getAttribute("disabled") or "0") + if xml.hasAttribute("eit") and xml.getAttribute("eit") != "None": + eit = long(xml.getAttribute("eit")) + else: + eit = None + + name = xml.getAttribute("name").encode("utf-8") + #filename = xml.getAttribute("filename").encode("utf-8") + entry = RecordTimerEntry(serviceref, begin, end, name, description, eit, disabled) entry.repeated = int(repeated) + + for l in elementsWithTag(xml.childNodes, "log"): + time = int(l.getAttribute("time")) + code = int(l.getAttribute("code")) + msg = mergeText(l.childNodes).strip().encode("utf-8") + entry.log_entries.append((time, code, msg)) + return entry class RecordTimer(timer.Timer): @@ -128,7 +220,7 @@ class RecordTimer(timer.Timer): try: self.loadTimer() - except: + except IOError: print "unable to load timers from file!" def isRecording(self): @@ -145,63 +237,142 @@ class RecordTimer(timer.Timer): root = doc.childNodes[0] for timer in elementsWithTag(root.childNodes, "timer"): self.record(createTimer(timer)) - + + def strToXML(self, str): + return str.replace('&', '&').replace('<', '<').replace('>', '>').replace("'", '''). replace('"', '&qout;') + def saveTimer(self): - doc = xml.dom.minidom.Document() - root_element = doc.createElement('timers') - doc.appendChild(root_element) - root_element.appendChild(doc.createTextNode("\n")) + #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: + #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(int(timer.begin))) + #t.setAttribute("end", str(int(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)) + + #for time, code, msg in timer.log_entries: + #t.appendChild(doc.createTextNode("\t\t")) + #l = doc.createElement('log') + #l.setAttribute("time", str(time)) + #l.setAttribute("code", str(code)) + #l.appendChild(doc.createTextNode(msg)) + #t.appendChild(l) + #t.appendChild(doc.createTextNode("\n")) + + #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() + + list = [] + + list.append('\n') + list.append('\n') + + for timer in self.timer_list + self.processed_timers: 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("epgdata", timer.) - t.setAttribute("description", timer.description) - root_element.appendChild(t) - t = doc.createTextNode("\n") - root_element.appendChild(t) + + list.append('\n') + + for time, code, msg in timer.log_entries: + list.append('') + list.append(str(msg)) + list.append('\n') + + + list.append('\n') + + list.append('\n') file = open(self.Filename, "w") - doc.writexml(codecs.getwriter('UTF-8')(file)) - file.write("\n") + for x in list: + file.write(x) file.close() - + def record(self, entry): + entry.timeChanged() print "[Timer] Record " + str(entry) entry.Timer = self self.addTimerEntry(entry) + + def isInTimer(self, eventid, begin, duration, service): + time_match = 0 + for x in self.timer_list: + if str(x.service_ref) == str(service): + #if x.eit is not None and x.repeated == 0: + # if x.eit == eventid: + # return duration + if x.repeated != 0: + chktime = localtime(begin) + time = localtime(x.begin) + chktimecmp = chktime.tm_wday * 1440 + chktime.tm_hour * 60 + chktime.tm_min + for y in range(7): + if x.repeated & (2 ** y): + timecmp = y * 1440 + time.tm_hour * 60 + time.tm_min + if timecmp <= chktimecmp < (timecmp + ((x.end - x.begin) / 60)): + time_match = ((timecmp + ((x.end - x.begin) / 60)) - chktimecmp) * 60 + elif chktimecmp <= timecmp < (chktimecmp + (duration / 60)): + time_match = ((chktimecmp + (duration / 60)) - timecmp) * 60 + elif x.eit is None: + end = begin + duration + if begin <= x.begin <= end: + diff = end - x.begin + if time_match < diff: + time_match = diff + elif x.begin <= begin <= x.end: + diff = x.end - begin + if time_match < diff: + time_match = diff + return time_match + + + + def removeEntry(self, entry): print "[Timer] Remove " + str(entry) + # avoid re-enqueuing entry.repeated = False - entry.repeated = False - - if entry.state == timer.TimerEntry.StateRunning: - print "remove running timer." - entry.end = time.time() + # abort timer. + # this sets the end time to current time, so timer will be stopped. + entry.abort() + + if entry.state != entry.StateEnded: 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