X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/8a6e44c53a3fb23ca87645fdfad73652a202b905..a27631694342ad6f77090743b5affddef5e155e0:/RecordTimer.py diff --git a/RecordTimer.py b/RecordTimer.py index e375c9f2..d8312493 100644 --- a/RecordTimer.py +++ b/RecordTimer.py @@ -7,11 +7,13 @@ from Components.config import config import timer import xml.dom.minidom +from enigma import quitMainloop, eEPGCache + from Screens.MessageBox import MessageBox -from Screens.SubserviceSelection import SubserviceSelection import NavigationInstance +from time import localtime -from Tools.XMLTools import elementsWithTag, mergeText +from Tools.XMLTools import elementsWithTag, mergeText, stringToXML from ServiceReference import ServiceReference # ok, for descriptions etc we have: @@ -22,19 +24,34 @@ from ServiceReference import ServiceReference # 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 * 60 + end += config.recording.margin_after.value * 60 return (begin, end, name, description, eit) +class AFTEREVENT: + NONE = 0 + STANDBY = 1 + DEEPSTANDBY = 2 + # please do not translate log messages class RecordTimerEntry(timer.TimerEntry): - def __init__(self, serviceref, begin, end, name, description, eit): + def __init__(self, serviceref, begin, end, name, description, eit, disabled = False, justplay = False, afterEvent = AFTEREVENT.NONE, checkOldTimers = False): timer.TimerEntry.__init__(self, int(begin), int(end)) + if checkOldTimers == True: + if self.begin < time.time() - 1209600: + self.begin = int(time.time()) + + if self.end < self.begin: + self.end = self.begin + assert isinstance(serviceref, ServiceReference) self.service_ref = serviceref @@ -42,8 +59,13 @@ class RecordTimerEntry(timer.TimerEntry): self.dontSave = False self.name = name self.description = description + self.disabled = disabled self.timer = None self.record_service = None + self.start_prepare = 0 + self.justplay = justplay + self.afterEvent = afterEvent + self.session = None self.log_entries = [] self.resetState() @@ -54,47 +76,63 @@ class RecordTimerEntry(timer.TimerEntry): def resetState(self): self.state = self.StateWaiting + self.cancelled = False self.first_try_prepare = True self.timeChanged() def calculateFilename(self): service_name = self.service_ref.getServiceName() -# begin_date = datetime.fromtimestamp(begin).strf... - begin_date = "" + begin_date = time.strftime("%Y%m%d %H%M", time.localtime(self.begin)) print "begin_date: ", begin_date print "service_name: ", service_name print "name:", self.name print "description: ", self.description + + filename = begin_date + " - " + service_name + if self.name: + filename += " - " + self.name - self.Filename = Directories.getRecordingFilename(service_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 + if self.justplay: + return True else: - prep_res = self.record_service.prepare(self.Filename + ".ts") - if prep_res: - self.log(2, "'prepare' failed: error %d" % prep_res) - self.record_service = None + 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: + 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 - 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 + if self.repeated: + epgcache = eEPGCache.getInstance() + queryTime=self.begin+(self.end-self.begin)/2 + evt = epgcache.lookupEventTime(self.service_ref.ref, queryTime) + if evt: + self.description = evt.getShortDescription() + 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 IOError: + self.log(4, "failed to write meta information") + return True def do_backoff(self): if self.backoff == 0: @@ -120,11 +158,12 @@ class RecordTimerEntry(timer.TimerEntry): self.log(7, "prepare failed") if self.first_try_prepare: self.first_try_prepare = False - if config.recording.asktozap.value == 0: + if not config.recording.asktozap.value: self.log(8, "asking user to zap away") - Notifications.AddNotificationWithCallback(self.failureCB, MessageBox, _("A timer failed to record!\nDisable TV and try again?\n")) + Notifications.AddNotificationWithCallback(self.failureCB, MessageBox, _("A timer failed to record!\nDisable TV and try again?\n"), timeout=20) else: # zap without asking self.log(9, "zap without asking") + Notifications.AddNotification(MessageBox, _("In order to record a timer, the TV was switched to the recording service!\n"), type=MessageBox.TYPE_INFO, timeout=20) self.failureCB(True) self.do_backoff() @@ -132,21 +171,36 @@ class RecordTimerEntry(timer.TimerEntry): 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 + # if this timer has been cancelled, just go to "end" state. + if self.cancelled: + return True + + if self.justplay: + self.log(11, "zapping") + NavigationInstance.instance.playService(self.service_ref.ref) + return True + else: + 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 + if not self.justplay: + self.record_service.stop() + self.record_service = None + if self.afterEvent == AFTEREVENT.STANDBY: + if self.session is not None: + self.session.open(Standby, self) + elif self.afterEvent == AFTEREVENT.DEEPSTANDBY: + quitMainloop(1) return True def getNextActivation(self): @@ -168,26 +222,37 @@ class RecordTimerEntry(timer.TimerEntry): 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 - self.log(15, "record time changed, start prepare is now: %s" % time.ctime(self.start_prepare)) + + 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"))) + serviceref = ServiceReference(xml.getAttribute("serviceref").encode("utf-8")) description = xml.getAttribute("description").encode("utf-8") repeated = xml.getAttribute("repeated").encode("utf-8") - eit = xml.getAttribute("eit").encode("utf-8") + disabled = long(xml.getAttribute("disabled") or "0") + justplay = long(xml.getAttribute("justplay") or "0") + afterevent = str(xml.getAttribute("afterevent") or "nothing") + afterevent = { "nothing": AFTEREVENT.NONE, "standby": AFTEREVENT.STANDBY, "deepstandby": AFTEREVENT.DEEPSTANDBY }[afterevent] + 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) + entry = RecordTimerEntry(serviceref, begin, end, name, description, eit, disabled, justplay, afterevent) 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() + msg = mergeText(l.childNodes).strip().encode("utf-8") entry.log_entries.append((time, code, msg)) return entry @@ -202,11 +267,11 @@ class RecordTimer(timer.Timer): 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(): + if timer.isRunning() and not timer.justplay: isRunning = True return isRunning @@ -217,45 +282,87 @@ class RecordTimer(timer.Timer): root = doc.childNodes[0] for timer in elementsWithTag(root.childNodes, "timer"): self.record(createTimer(timer)) - + 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(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)) + + list.append('\n') - 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) + #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(file) - file.write("\n") + for x in list: + file.write(x) file.close() def record(self, entry): @@ -263,6 +370,41 @@ class RecordTimer(timer.Timer): print "[Timer] Record " + str(entry) entry.Timer = self self.addTimerEntry(entry) + + def isInTimer(self, eventid, begin, duration, service): + time_match = 0 + chktime = None + chktimecmp = None + chktimecmp_end = None + end = begin + duration + 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: + if chktime is None: + chktime = localtime(begin) + chktimecmp = chktime.tm_wday * 1440 + chktime.tm_hour * 60 + chktime.tm_min + chktimecmp_end = chktimecmp + (duration / 60) + time = localtime(x.begin) + 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_end: + time_match = (chktimecmp_end - timecmp) * 60 + else: #if x.eit is None: + 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)