Merge branch 'bug_518_add_demux_reserve_from_python'

[enigma2.git] / RecordTimer.py

diff --git a/RecordTimer.py b/RecordTimer.py
old mode 100644 (file)
new mode 100755 (executable)
index 8449a2d..4ece9c5
--- a/RecordTimer.py
+++ b/RecordTimer.py
@@ -1,25 +1,23 @@
-import time
-#from time import datetime
-from Tools import Directories, Notifications
-
-from Components.config import config
-import timer
-import xml.etree.cElementTree
-

 from enigma import eEPGCache, getBestPlayableServiceReference, \
        eServiceReference, iRecordableService, quitMainloop
 
 from enigma import eEPGCache, getBestPlayableServiceReference, \
        eServiceReference, iRecordableService, quitMainloop
 

-from Screens.MessageBox import MessageBox
+from Components.config import config
+from Components.UsageConfig import defaultMoviePath

 from Components.TimerSanityCheck import TimerSanityCheck
 from Components.TimerSanityCheck import TimerSanityCheck

-import NavigationInstance

 
 

+from Screens.MessageBox import MessageBox

 import Screens.Standby
 import Screens.Standby

-
-from time import localtime
-
+from Tools import Directories, Notifications, ASCIItranslit

 from Tools.XMLTools import stringToXML
 from Tools.XMLTools import stringToXML

+
+import timer
+import xml.etree.cElementTree
+import NavigationInstance

 from ServiceReference import ServiceReference
 
 from ServiceReference import ServiceReference
 

+from time import localtime, strftime, ctime, time
+from bisect import insort
+

 # ok, for descriptions etc we have:
 # service reference  (to get the service name)
 # name               (title)
 # ok, for descriptions etc we have:
 # service reference  (to get the service name)
 # name               (title)


@@ -63,10 +61,10 @@ class RecordTimerEntry(timer.TimerEntry, object):
                if event == iRecordableService.evEnd:
                        print "RecordTimer.staticGotRecordEvent(iRecordableService.evEnd)"
                        recordings = NavigationInstance.instance.getRecordings()
                if event == iRecordableService.evEnd:
                        print "RecordTimer.staticGotRecordEvent(iRecordableService.evEnd)"
                        recordings = NavigationInstance.instance.getRecordings()

-                       if not len(recordings): # no more recordings exist
+                       if not recordings: # no more recordings exist

                                rec_time = NavigationInstance.instance.RecordTimer.getNextRecordingTime()
                                rec_time = NavigationInstance.instance.RecordTimer.getNextRecordingTime()

-                               if rec_time > 0 and (rec_time - time.time()) < 360:
-                                       print "another recording starts in", rec_time - time.time(), "seconds... do not shutdown yet"
+                               if rec_time > 0 and (rec_time - time()) < 360:
+                                       print "another recording starts in", rec_time - time(), "seconds... do not shutdown yet"

                                else:
                                        print "no starting records in the next 360 seconds... immediate shutdown"
                                        RecordTimerEntry.shutdown() # immediate shutdown
                                else:
                                        print "no starting records in the next 360 seconds... immediate shutdown"
                                        RecordTimerEntry.shutdown() # immediate shutdown


@@ -96,15 +94,18 @@ class RecordTimerEntry(timer.TimerEntry, object):
                timer.TimerEntry.__init__(self, int(begin), int(end))
 
                if checkOldTimers == True:
                timer.TimerEntry.__init__(self, int(begin), int(end))
 
                if checkOldTimers == True:

-                       if self.begin < time.time() - 1209600:
-                               self.begin = int(time.time())
+                       if self.begin < time() - 1209600:
+                               self.begin = int(time())

                
                if self.end < self.begin:
                        self.end = self.begin
                
                assert isinstance(serviceref, ServiceReference)
                
                
                if self.end < self.begin:
                        self.end = self.begin
                
                assert isinstance(serviceref, ServiceReference)
                

-               self.service_ref = serviceref
+               if serviceref.isRecordable():
+                       self.service_ref = serviceref
+               else:
+                       self.service_ref = ServiceReference(None)

                self.eit = eit
                self.dontSave = False
                self.name = name
                self.eit = eit
                self.dontSave = False
                self.name = name


@@ -118,18 +119,20 @@ class RecordTimerEntry(timer.TimerEntry, object):
                self.dirname = dirname
                self.dirnameHadToFallback = False
                self.autoincrease = False
                self.dirname = dirname
                self.dirnameHadToFallback = False
                self.autoincrease = False

+               self.autoincreasetime = 3600 * 24 # 1 day

                self.tags = tags or []
 
                self.log_entries = []
                self.resetState()
        
        def log(self, code, msg):
                self.tags = tags or []
 
                self.log_entries = []
                self.resetState()
        
        def log(self, code, msg):

-               self.log_entries.append((int(time.time()), code, msg))
+               self.log_entries.append((int(time()), code, msg))

                print "[TIMER]", msg
 
        def calculateFilename(self):
                service_name = self.service_ref.getServiceName()
                print "[TIMER]", msg
 
        def calculateFilename(self):
                service_name = self.service_ref.getServiceName()

-               begin_date = time.strftime("%Y%m%d %H%M", time.localtime(self.begin))
+               begin_date = strftime("%Y%m%d %H%M", localtime(self.begin))
+               begin_shortdate = strftime("%Y%m%d", localtime(self.begin))

                
                print "begin_date: ", begin_date
                print "service_name: ", service_name
                
                print "begin_date: ", begin_date
                print "service_name: ", service_name


@@ -138,13 +141,26 @@ class RecordTimerEntry(timer.TimerEntry, object):
                
                filename = begin_date + " - " + service_name
                if self.name:
                
                filename = begin_date + " - " + service_name
                if self.name:

-                       filename += " - " + self.name
+                       if config.usage.setup_level.index >= 2: # expert+
+                               if config.recording.filename_composition.value == "short":
+                                       filename = begin_shortdate + " - " + self.name
+                               elif config.recording.filename_composition.value == "long":
+                                       filename += " - " + self.name + " - " + self.description
+                               else:
+                                       filename += " - " + self.name # standard
+                       else:
+                               filename += " - " + self.name

 
 

-               if self.dirname and not Directories.pathExists(self.dirname):
-                       self.dirnameHadToFallback = True
-                       self.Filename = Directories.getRecordingFilename(filename, None)
+               if config.recording.ascii_filenames.value:
+                       filename = ASCIItranslit.legacyEncode(filename)
+
+               if not self.dirname or not Directories.fileExists(self.dirname, 'w'):
+                       if self.dirname:
+                               self.dirnameHadToFallback = True
+                       dirname = defaultMoviePath()

                else:
                else:

-                       self.Filename = Directories.getRecordingFilename(filename, self.dirname)
+                       dirname = self.dirname
+               self.Filename = Directories.getRecordingFilename(filename, dirname)

                self.log(0, "Filename calculated as: '%s'" % self.Filename)
                #begin_date + " - " + service_name + description)
 
                self.log(0, "Filename calculated as: '%s'" % self.Filename)
                #begin_date + " - " + service_name + description)
 


@@ -180,24 +196,18 @@ class RecordTimerEntry(timer.TimerEntry, object):
                                if event_id is None:
                                        event_id = -1
 
                                if event_id is None:
                                        event_id = -1
 

-                       prep_res=self.record_service.prepare(self.Filename + ".ts", self.begin, self.end, event_id)
+                       prep_res=self.record_service.prepare(self.Filename + ".ts", self.begin, self.end, event_id, self.name.replace("\n", ""), self.description.replace("\n", ""), ' '.join(self.tags))

                        if prep_res:
                        if prep_res:

-                               self.log(2, "'prepare' failed: error %d" % prep_res)
-                               NavigationInstance.instance.stopRecordService(self.record_service)
-                               self.record_service = None
-                               return False
+                               if prep_res == -255:
+                                       self.log(4, "failed to write meta information")
+                               else:
+                                       self.log(2, "'prepare' failed: error %d" % prep_res)
+
+                               # we must calc nur start time before stopRecordService call because in Screens/Standby.py TryQuitMainloop tries to get
+                               # the next start time in evEnd event handler...
+                               self.do_backoff()
+                               self.start_prepare = time() + self.backoff

 
 

-                       self.log(3, "prepare ok, writing meta information to %s" % self.Filename)
-                       try:
-                               f = open(self.Filename + ".ts.meta", "w")
-                               f.write(rec_ref.toString() + "\n")
-                               f.write(self.name + "\n")
-                               f.write(self.description + "\n")
-                               f.write(str(self.begin) + "\n")
-                               f.write(' '.join(self.tags))
-                               f.close()
-                       except IOError:
-                               self.log(4, "failed to write meta information")

                                NavigationInstance.instance.stopRecordService(self.record_service)
                                self.record_service = None
                                return False
                                NavigationInstance.instance.stopRecordService(self.record_service)
                                self.record_service = None
                                return False


@@ -215,29 +225,37 @@ class RecordTimerEntry(timer.TimerEntry, object):
        def activate(self):
                next_state = self.state + 1
                self.log(5, "activating state %d" % next_state)
        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")
                if next_state == self.StatePrepared:
                        if self.tryPrepare():
                                self.log(6, "prepare ok, waiting for begin")

+                               # create file to "reserve" the filename
+                               # because another recording at the same time on another service can try to record the same event
+                               # i.e. cable / sat.. then the second recording needs an own extension... when we create the file
+                               # here than calculateFilename is happy
+                               if not self.justplay:
+                                       open(self.Filename + ".ts", "w").close() 

                                # fine. it worked, resources are allocated.
                                self.next_activation = self.begin
                                self.backoff = 0
                                return True
                                # 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
                        self.log(7, "prepare failed")
                        if self.first_try_prepare:
                                self.first_try_prepare = False

-                               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"), 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()
-                       # retry
-                       self.start_prepare = time.time() + self.backoff
+                               cur_ref = NavigationInstance.instance.getCurrentlyPlayingServiceReference()
+                               if cur_ref and not cur_ref.getPath():
+                                       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"), 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)
+                               elif cur_ref:
+                                       self.log(8, "currently running service is not a live service.. so stop it makes no sense")
+                               else:
+                                       self.log(8, "currently no service running... so we dont need to stop it")

                        return False
                elif next_state == self.StateRunning:
                        # if this timer has been cancelled, just go to "end" state.
                        return False
                elif next_state == self.StateRunning:
                        # if this timer has been cancelled, just go to "end" state.


@@ -263,11 +281,16 @@ class RecordTimerEntry(timer.TimerEntry, object):
                                        self.log(13, "start record returned %d" % record_res)
                                        self.do_backoff()
                                        # retry
                                        self.log(13, "start record returned %d" % record_res)
                                        self.do_backoff()
                                        # retry

-                                       self.begin = time.time() + self.backoff
+                                       self.begin = time() + self.backoff

                                        return False
 
                                return True
                elif next_state == self.StateEnded:
                                        return False
 
                                return True
                elif next_state == self.StateEnded:

+                       old_end = self.end
+                       if self.setAutoincreaseEnd():
+                               self.log(12, "autoincrase recording %d minute(s)" % int((self.end - old_end)/60))
+                               self.state -= 1
+                               return True

                        self.log(12, "stop recording")
                        if not self.justplay:
                                NavigationInstance.instance.stopRecordService(self.record_service)
                        self.log(12, "stop recording")
                        if not self.justplay:
                                NavigationInstance.instance.stopRecordService(self.record_service)


@@ -283,6 +306,29 @@ class RecordTimerEntry(timer.TimerEntry, object):
                                                Notifications.AddNotificationWithCallback(self.sendTryQuitMainloopNotification, MessageBox, _("A finished record timer wants to shut down\nyour Dreambox. Shutdown now?"), timeout = 20)
                        return True
 
                                                Notifications.AddNotificationWithCallback(self.sendTryQuitMainloopNotification, MessageBox, _("A finished record timer wants to shut down\nyour Dreambox. Shutdown now?"), timeout = 20)
                        return True
 

+       def setAutoincreaseEnd(self, entry = None):
+               if not self.autoincrease:
+                       return False
+               if entry is None:
+                       new_end =  int(time()) + self.autoincreasetime
+               else:
+                       new_end = entry.begin -30
+
+               dummyentry = RecordTimerEntry(self.service_ref, self.begin, new_end, self.name, self.description, self.eit, disabled=True, justplay = self.justplay, afterEvent = self.afterEvent, dirname = self.dirname, tags = self.tags)
+               dummyentry.disabled = self.disabled
+               timersanitycheck = TimerSanityCheck(NavigationInstance.instance.RecordTimer.timer_list, dummyentry)
+               if not timersanitycheck.check():
+                       simulTimerList = timersanitycheck.getSimulTimerList()
+                       new_end = simulTimerList[1].begin
+                       del simulTimerList
+                       new_end -= 30                           # 30 Sekunden Prepare-Zeit lassen
+               del dummyentry
+               if new_end <= time():
+                       return False
+               self.end = new_end
+               return True
+       
+       

        def sendStandbyNotification(self, answer):
                if answer:
                        Notifications.AddNotification(Screens.Standby.Standby)
        def sendStandbyNotification(self, answer):
                if answer:
                        Notifications.AddNotification(Screens.Standby.Standby)


@@ -315,7 +361,7 @@ class RecordTimerEntry(timer.TimerEntry, object):
                self.backoff = 0
                
                if int(old_prepare) != int(self.start_prepare):
                self.backoff = 0
                
                if int(old_prepare) != int(self.start_prepare):

-                       self.log(15, "record time changed, start prepare is now: %s" % time.ctime(self.start_prepare))
+                       self.log(15, "record time changed, start prepare is now: %s" % ctime(self.start_prepare))

 
        def gotRecordEvent(self, record, event):
                # TODO: this is not working (never true), please fix. (comparing two swig wrapped ePtrs)
 
        def gotRecordEvent(self, record, event):
                # TODO: this is not working (never true), please fix. (comparing two swig wrapped ePtrs)


@@ -334,10 +380,10 @@ class RecordTimerEntry(timer.TimerEntry, object):
                elif event == iRecordableService.evStart:
                        text = _("A record has been started:\n%s") % self.name
                        if self.dirnameHadToFallback:
                elif event == iRecordableService.evStart:
                        text = _("A record has been started:\n%s") % self.name
                        if self.dirnameHadToFallback:

-                               text = '\n'.join([text, _("Please note that the previously selected media could not be accessed and therefore the default directory is being used instead.")])
+                               text = '\n'.join((text, _("Please note that the previously selected media could not be accessed and therefore the default directory is being used instead.")))

 
 

-                       # maybe this should be configurable?
-                       Notifications.AddPopup(text = text, type = MessageBox.TYPE_INFO, timeout = 3)
+                       if config.usage.show_message_when_recording_starts.value:
+                               Notifications.AddPopup(text = text, type = MessageBox.TYPE_INFO, timeout = 3)

 
        # we have record_service as property to automatically subscribe to record service events
        def setRecordService(self, service):
 
        # we have record_service as property to automatically subscribe to record service events
        def setRecordService(self, service):


@@ -407,7 +453,36 @@ class RecordTimer(timer.Timer):
                        self.loadTimer()
                except IOError:
                        print "unable to load timers from file!"
                        self.loadTimer()
                except IOError:
                        print "unable to load timers from file!"

-                       
+
+       def doActivate(self, w):
+               # when activating a timer which has already passed,
+               # simply abort the timer. don't run trough all the stages.
+               if w.shouldSkip():
+                       w.state = RecordTimerEntry.StateEnded
+               else:
+                       # when active returns true, this means "accepted".
+                       # otherwise, the current state is kept.
+                       # the timer entry itself will fix up the delay then.
+                       if w.activate():
+                               w.state += 1
+
+               self.timer_list.remove(w)
+
+               # did this timer reached the last state?
+               if w.state < RecordTimerEntry.StateEnded:
+                       # no, sort it into active list
+                       insort(self.timer_list, w)
+               else:
+                       # yes. Process repeated, and re-add.
+                       if w.repeated:
+                               w.processRepeated()
+                               w.state = RecordTimerEntry.StateWaiting
+                               self.addTimerEntry(w)
+                       else:
+                               insort(self.processed_timers, w)
+               
+               self.stateChanged(w)
+

        def isRecording(self):
                isRunning = False
                for timer in self.timer_list:
        def isRecording(self):
                isRunning = False
                for timer in self.timer_list:


@@ -539,7 +614,7 @@ class RecordTimer(timer.Timer):
                file.close()
 
        def getNextZapTime(self):
                file.close()
 
        def getNextZapTime(self):

-               now = time.time()
+               now = time()

                for timer in self.timer_list:
                        if not timer.justplay or timer.begin < now:
                                continue
                for timer in self.timer_list:
                        if not timer.justplay or timer.begin < now:
                                continue


@@ -547,15 +622,16 @@ class RecordTimer(timer.Timer):
                return -1
 
        def getNextRecordingTime(self):
                return -1
 
        def getNextRecordingTime(self):

-               now = time.time()
+               now = time()

                for timer in self.timer_list:
                for timer in self.timer_list:

-                       if timer.justplay or timer.begin < now:
+                       next_act = timer.getNextActivation()
+                       if timer.justplay or next_act < now:

                                continue
                                continue

-                       return timer.begin
+                       return next_act

                return -1
 
        def isNextRecordAfterEventActionAuto(self):
                return -1
 
        def isNextRecordAfterEventActionAuto(self):

-               now = time.time()
+               now = time()

                t = None
                for timer in self.timer_list:
                        if timer.justplay or timer.begin < now:
                t = None
                for timer in self.timer_list:
                        if timer.justplay or timer.begin < now:


@@ -577,6 +653,7 @@ class RecordTimer(timer.Timer):
                                print "ignore timer conflict"
                elif timersanitycheck.doubleCheck():
                        print "ignore double timer"
                                print "ignore timer conflict"
                elif timersanitycheck.doubleCheck():
                        print "ignore double timer"

+                       return None

                entry.timeChanged()
                print "[Timer] Record " + str(entry)
                entry.Timer = self
                entry.timeChanged()
                print "[Timer] Record " + str(entry)
                entry.Timer = self


@@ -584,15 +661,16 @@ class RecordTimer(timer.Timer):
                if dosave:
                        self.saveTimer()
                return None
                if dosave:
                        self.saveTimer()
                return None

-               
+

        def isInTimer(self, eventid, begin, duration, service):
                time_match = 0
                chktime = None
                chktimecmp = None
                chktimecmp_end = None
                end = begin + duration
        def isInTimer(self, eventid, begin, duration, service):
                time_match = 0
                chktime = None
                chktimecmp = None
                chktimecmp_end = None
                end = begin + duration

+               refstr = str(service)

                for x in self.timer_list:
                for x in self.timer_list:

-                       check = x.service_ref.ref.toCompareString() == str(service)
+                       check = x.service_ref.ref.toString() == refstr

                        if not check:
                                sref = x.service_ref.ref
                                parent_sid = sref.getUnsignedData(5)
                        if not check:
                                sref = x.service_ref.ref
                                parent_sid = sref.getUnsignedData(5)


@@ -604,7 +682,7 @@ class RecordTimer(timer.Timer):
                                        sref.setUnsignedData(2, parent_tsid)
                                        sref.setUnsignedData(5, 0)
                                        sref.setUnsignedData(6, 0)
                                        sref.setUnsignedData(2, parent_tsid)
                                        sref.setUnsignedData(5, 0)
                                        sref.setUnsignedData(6, 0)

-                                       check = x.service_ref.ref.toCompareString() == str(service)
+                                       check = sref.toCompareString() == refstr

                                        num = 0
                                        if check:
                                                check = False
                                        num = 0
                                        if check:
                                                check = False


@@ -620,17 +698,14 @@ class RecordTimer(timer.Timer):
                                                        check = True
                                                        break
                        if check:
                                                        check = True
                                                        break
                        if check:

-                               #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)
                                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):
+                                       for y in (0, 1, 2, 3, 4, 5, 6):
+                                               if x.repeated & (1 << y) and (x.begin <= begin or begin <= x.begin <= end):

                                                        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
                                                        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


@@ -645,6 +720,8 @@ class RecordTimer(timer.Timer):
                                                diff = x.end - begin
                                                if time_match < diff:
                                                        time_match = diff
                                                diff = x.end - begin
                                                if time_match < diff:
                                                        time_match = diff

+                               if time_match:
+                                       break

                return time_match
 
        def removeEntry(self, entry):
                return time_match
 
        def removeEntry(self, entry):


@@ -655,6 +732,7 @@ class RecordTimer(timer.Timer):
 
                # abort timer.
                # this sets the end time to current time, so timer will be stopped.
 
                # abort timer.
                # this sets the end time to current time, so timer will be stopped.

+               entry.autoincrease = False

                entry.abort()
                
                if entry.state != entry.StateEnded:
                entry.abort()
                
                if entry.state != entry.StateEnded:


@@ -663,6 +741,11 @@ class RecordTimer(timer.Timer):
                print "state: ", entry.state
                print "in processed: ", entry in self.processed_timers
                print "in running: ", entry in self.timer_list
                print "state: ", entry.state
                print "in processed: ", entry in self.processed_timers
                print "in running: ", entry in self.timer_list

+               # autoincrease instanttimer if possible
+               if not entry.dontSave:
+                       for x in self.timer_list:
+                               if x.setAutoincreaseEnd():
+                                       self.timeChanged(x)

                # now the timer should be in the processed_timers list. remove it from there.
                self.processed_timers.remove(entry)
                self.saveTimer()
                # now the timer should be in the processed_timers list. remove it from there.
                self.processed_timers.remove(entry)
                self.saveTimer()