99939bd8e1563bc5fdccb569242d1965001df85c
[enigma2.git] / lib / python / Screens / TimerEdit.py
1 from Screen import Screen
2 from Components.TimerList import TimerList, TimerEntryComponent
3 from Components.ActionMap import ActionMap
4 from Components.TimeInput import TimeInput
5 from Components.Label import Label
6 from Components.Button import Button
7 from Screens.MessageBox import MessageBox
8 from TimerEntry import TimerEntry, TimerLog
9 from RecordTimer import RecordTimerEntry, parseEvent
10 from time import *
11 from ServiceReference import ServiceReference
12 from Components.config import *
13
14 class TimerEditList(Screen):
15         def __init__(self, session):
16                 Screen.__init__(self, session)
17                 
18                 list = [ ]
19                 self.list = list
20                 self.fillTimerList()
21
22                 self["timerlist"] = TimerList(list)
23                 
24                 self["key_red"] = Button(_("Delete"))
25                 self["key_green"] = Button(_("Add"))
26                 self["key_yellow"] = Button("")
27                 self["key_blue"] = Button(_("Cleanup"))
28
29                 self["actions"] = ActionMap(["OkCancelActions", "ShortcutActions", "TimerEditActions"], 
30                         {
31                                 "ok": self.openEdit,
32                                 "cancel": self.leave,
33                                 "red": self.removeTimer,
34                                 "green": self.addCurrentTimer,
35                                 "blue": self.cleanupQuestion,
36                                 "log": self.showLog
37                         })
38                 self.session.nav.RecordTimer.on_state_change.append(self.onStateChange)
39
40         def fillTimerList(self):
41                 del self.list[:]
42                 
43                 for timer in self.session.nav.RecordTimer.timer_list:
44                         self.list.append(TimerEntryComponent(timer, processed=False))
45                 
46                 for timer in self.session.nav.RecordTimer.processed_timers:
47                         self.list.append(TimerEntryComponent(timer, processed=True))
48
49         def showLog(self):
50                 self.session.openWithCallback(self.finishedEdit, TimerLog, self["timerlist"].getCurrent()[0])
51
52         def openEdit(self):
53                 self.session.openWithCallback(self.finishedEdit, TimerEntry, self["timerlist"].getCurrent()[0])
54                 #self.session.open(TimerEdit, self["timerlist"].getCurrent()[0])
55                 
56         def cleanupQuestion(self):
57                 self.session.openWithCallback(self.cleanupTimer, MessageBox, _("Really delete done timers?"))
58         
59         def cleanupTimer(self, delete):
60                 if delete:
61                         self.session.nav.RecordTimer.cleanup()
62                         self.refill()
63                 
64         def removeTimer(self):
65                 list = self["timerlist"]
66                 currentIndex = list.getCurrentIndex()
67                 list.moveDown()
68                 if list.getCurrentIndex() == currentIndex:
69                         currentIndex -= 1
70                         list.moveToIndex(currentIndex)
71                 self.session.nav.RecordTimer.removeEntry(list.getCurrent()[0])
72                 self.refill()
73         
74         def refill(self):
75                 self.fillTimerList()
76                 self["timerlist"].invalidate()
77         
78         def addCurrentTimer(self):
79                 event = None
80                 service = self.session.nav.getCurrentService()
81                 if service is not None:
82                         info = service.info()
83                         if info is not None:
84                                 event = info.getEvent(0)
85
86                 # FIXME only works if already playing a service
87                 serviceref = ServiceReference(self.session.nav.getCurrentlyPlayingServiceReference())
88                 
89                 if event is None:       
90                         data = (int(time()), int(time() + 60), "unknown event", "", None)
91                 else:
92                         data = parseEvent(event)
93
94                 self.addTimer(RecordTimerEntry(serviceref, *data))
95                 
96         def addTimer(self, timer):
97                 self.session.openWithCallback(self.finishedAdd, TimerEntry, timer)
98                 
99         def finishedEdit(self, answer):
100                 print "finished edit"
101                 if answer[0]:
102                         print "Edited timer"
103                         self.session.nav.RecordTimer.timeChanged(answer[1])
104                         self.fillTimerList()
105                 else:
106                         print "Timeredit aborted"
107
108         def finishedAdd(self, answer):
109                 print "finished add"
110                 if answer[0]:
111                         entry = answer[1]
112                         self.session.nav.RecordTimer.record(entry)
113                         self.fillTimerList()
114                 else:
115                         print "Timeredit aborted"               
116
117         def leave(self):
118                 self.session.nav.RecordTimer.saveTimer()
119                 self.session.nav.RecordTimer.on_state_change.remove(self.onStateChange)
120                 self.close()
121
122         def onStateChange(self, entry):
123                 self.refill()