timer-add-support

[enigma2.git] / lib / python / Screens / TimerEdit.py

diff --git a/lib/python/Screens/TimerEdit.py b/lib/python/Screens/TimerEdit.py
index f8c9f20797c5df37132841347368a3ac955e4c55..233bfe2d3000d5e2d73b4e785f24713d004cf26c 100644 (file)
--- a/lib/python/Screens/TimerEdit.py
+++ b/lib/python/Screens/TimerEdit.py
@@ -1,10 +1,14 @@
 from Screen import Screen
-from Components.TimerList import TimerList, TimerEntry
+from Components.TimerList import TimerList, TimerEntryComponent
 from Components.ActionMap import ActionMap
 from Components.TimeInput import TimeInput
 from Components.Label import Label
 from Components.Button import Button
 from Components.TextInput import TextInput
+from TimerEntry import TimerEntry
+from RecordTimer import RecordTimerEntry
+from time import *
+from ServiceReference import ServiceReference
 
 class TimerEdit(Screen):
        def __init__(self, session, entry):
@@ -55,18 +59,61 @@ class TimerEditList(Screen):
                
                list = [ ]
                for timer in session.nav.RecordTimer.timer_list:
-                       list.append(TimerEntry(timer, 0))
+                       list.append(TimerEntryComponent(timer, 0))
                
                for timer in session.nav.RecordTimer.processed_timers:
-                       list.append(TimerEntry(timer, 1))
+                       list.append(TimerEntryComponent(timer, 1))
                
                self["timerlist"] = TimerList(list)
+               
+               self["key_red"] = Button("Delete")
+               self["key_green"] = Button("Add")
+               self["key_yellow"] = Button("")
+               self["key_blue"] = Button("")
 
-               self["actions"] = ActionMap(["OkCancelActions"], 
+               self["actions"] = ActionMap(["OkCancelActions", "ShortcutActions"], 
                        {
                                "ok": self.openEdit,
-                               "cancel": self.close
+                               "cancel": self.close,
+                               "red": self.removeTimer,
+                               "green": self.addTimer
                        })
 
        def openEdit(self):
-               self.session.open(TimerEdit, self["timerlist"].getCurrent()[0])
+               self.session.open(TimerEntry, self["timerlist"].getCurrent()[0])
+               #self.session.open(TimerEdit, self["timerlist"].getCurrent()[0])
+               
+       def removeTimer(self):
+               self.session.nav.RecordTimer.removeEntry(self["timerlist"].getCurrent()[0])
+       
+       def addTimer(self):
+               begin = time()
+               end = time() + 60
+               
+               epg = None
+               try:
+                       service = self.session.nav.getCurrentService()
+                       info = service.info()
+                       ev = info.getEvent(0)
+                       epg = ev
+               except:
+                       pass
+               
+               if (epg == None):
+                       description = "unknown event"
+               else:
+                       description = ev.getEventName()
+                       # FIXME we need a timestamp here:
+                       begin = ev.getBeginTime()
+                       
+                       print begin
+                       print ev.getDuration()
+                       end = begin + ev.getDuration()
+
+
+               # FIXME only works if already playing a service
+               serviceref = ServiceReference(self.session.nav.getCurrentlyPlayingServiceReference())
+               
+               newEntry = RecordTimerEntry(begin, end, serviceref, epg, description)
+               self.session.open(TimerEntry, newEntry)
+