-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
+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, mergeText, stringToXML
+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, serviceref, begin, end, name, description, eit, disabled = False, justplay = False):
+ 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"
+ self.service_ref = serviceref
+ 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.justplay = justplay
+
+ 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.cancelled = False
+ self.first_try_prepare = True
+ self.timeChanged()
+
+ def calculateFilename(self):
+ service_name = self.service_ref.getServiceName()
+ 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(filename)
+ self.log(0, "Filename calculated as: '%s'" % self.Filename)
+ #begin_date + " - " + service_name + description)
+
+ def tryPrepare(self):
+ if self.justplay:
+ return True
else:
- self.Filename = "record_" + str(epg.m_event_name) + ".ts"
-
- print "------------ record filename: %s" % (self.Filename)
-
- 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."
- elif event == self.EventStart:
- self.RecordService.start()
- print "timer started!"
- elif event == self.EventEnd or event == self.EventAbort:
- self.RecordService.stop()
- self.RecordService = None
- print "Timer successfully ended"
-
-class RecordTimer(Timer):
+ 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
+
+ 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.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)
+
+ self.do_backoff()
+ # retry
+ self.start_prepare = time.time() + self.backoff
+ return False
+ elif next_state == self.StateRunning:
+ # 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")
+ if not self.justplay:
+ 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:
+ self.log(13, "ok, zapped away")
+ #NavigationInstance.instance.stopUserServices()
+ NavigationInstance.instance.playService(self.service_ref.ref)
+ else:
+ 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").encode("utf-8")
+ repeated = xml.getAttribute("repeated").encode("utf-8")
+ disabled = long(xml.getAttribute("disabled") or "0")
+ justplay = long(xml.getAttribute("justplay") 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, justplay)
+ 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):
def __init__(self):
- Timer.__init__(self)
+ timer.Timer.__init__(self)
- def loadTimer(self):
- print "TODO: load timers from xml"
+ 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):
+ # 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(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('<?xml version="1.0" ?>\n')
+ list.append('<timers>\n')
+
+ for timer in self.timer_list + self.processed_timers:
+ if timer.dontSave:
+ continue
+
+ list.append('<timer')
+ list.append(' begin="' + str(int(timer.begin)) + '"')
+ list.append(' end="' + str(int(timer.end)) + '"')
+ list.append(' serviceref="' + str(timer.service_ref) + '"')
+ list.append(' repeated="' + str(int(timer.repeated)) + '"')
+ list.append(' name="' + str(stringToXML(timer.name)) + '"')
+ list.append(' description="' + str(stringToXML(timer.description)) + '"')
+ if timer.eit is not None:
+ list.append(' eit="' + str(timer.eit) + '"')
+ list.append(' disabled="' + str(int(timer.disabled)) + '"')
+ list.append(' justplay="' + str(int(timer.justplay)) + '"')
+ list.append('>\n')
+
+ #for time, code, msg in timer.log_entries:
+ #list.append('<log')
+ #list.append(' code="' + str(code) + '"')
+ #list.append(' time="' + str(time) + '"')
+ #list.append('>')
+ #list.append(str(msg))
+ #list.append('</log>\n')
+
+
+ list.append('</timer>\n')
+
+ list.append('</timers>\n')
+
+ file = open(self.Filename, "w")
+ 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
+ else: #if 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):
- 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 did not yet start - removing"
- else:
- print "timer did already end - doing nothing."
+ print "[Timer] Remove " + str(entry)
+
+ # avoid re-enqueuing
+ entry.repeated = False
+
+ # 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)
+
+ 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()