update scrollbar in timerlist when a entry is removed
[enigma2.git] / lib / python / Screens / TimerEdit.py
1 from Components.ActionMap import ActionMap
2 from Components.Button import Button
3 from Components.MenuList import MenuList
4 from Components.TimerList import TimerList
5 from Components.TimerSanityCheck import TimerSanityCheck
6 from RecordTimer import RecordTimerEntry, parseEvent, AFTEREVENT
7 from Screen import Screen
8 from Screens.ChoiceBox import ChoiceBox
9 from Screens.MessageBox import MessageBox
10 from ServiceReference import ServiceReference
11 from TimerEntry import TimerEntry, TimerLog
12 from Tools.BoundFunction import boundFunction
13 from time import time
14
15 class TimerEditList(Screen):
16         def __init__(self, session):
17                 Screen.__init__(self, session)
18                 
19                 list = [ ]
20                 self.list = list
21                 self.fillTimerList()
22
23                 self["timerlist"] = TimerList(list)
24                 
25                 self["key_red"] = Button(_("Delete"))
26                 self["key_green"] = Button(_("Add"))
27                 self["key_yellow"] = Button("")
28                 self["key_blue"] = Button(_("Cleanup"))
29
30                 self["actions"] = ActionMap(["OkCancelActions", "DirectionActions", "ShortcutActions", "TimerEditActions"], 
31                         {
32                                 "ok": self.openEdit,
33                                 "cancel": self.leave,
34                                 "red": self.removeTimerQuestion,
35                                 "green": self.addCurrentTimer,
36                                 "blue": self.cleanupQuestion,
37                                 "yellow": self.toggleDisabledState,
38                                 "log": self.showLog,
39                                 "left": self.left,
40                                 "right": self.right,
41                                 "up": self.up,
42                                 "down": self.down
43                         }, -1)
44                 self.session.nav.RecordTimer.on_state_change.append(self.onStateChange)
45                 self.onShown.append(self.updateState)
46
47         def up(self):
48                 self["timerlist"].instance.moveSelection(self["timerlist"].instance.moveUp)
49                 self.updateState()
50                 
51         def down(self):
52                 self["timerlist"].instance.moveSelection(self["timerlist"].instance.moveDown)
53                 self.updateState()
54
55         def left(self):
56                 self["timerlist"].instance.moveSelection(self["timerlist"].instance.pageUp)
57                 self.updateState()
58                 
59         def right(self):
60                 self["timerlist"].instance.moveSelection(self["timerlist"].instance.pageDown)
61                 self.updateState()
62                 
63         def toggleDisabledState(self):
64                 cur=self["timerlist"].getCurrent()
65                 if cur:
66                         t = cur
67                 
68                         if t.disabled:
69                                 t.enable()
70                                 self.session.nav.RecordTimer.timeChanged(t)
71
72                         else:
73                                 if t.isRunning() and t.repeated:
74                                         list = []
75                                         list.append((_("Stop current event but not coming events"), "stoponlycurrent"))
76                                         list.append((_("Stop current event and disable coming events"), "stopall"))
77                                         list.append((_("Don't stop current event but disable coming events"), "stoponlycoming"))
78                                         self.session.openWithCallback(boundFunction(self.runningEventCallback, t), ChoiceBox, title=_("Repeating event currently recording... What do you want to do?"), list = list)
79                                 else:
80                                         t.disable()
81                                         self.session.nav.RecordTimer.timeChanged(t)
82                         self.updateState()
83                         self.refill()
84
85         def runningEventCallback(self, t, result):
86                 if result is not None:
87                         if result[1] == "stoponlycurrent" or result[1] == "stopall":
88                                 t.enable()
89                                 t.processRepeated(findRunningEvent = False)
90                                 self.session.nav.RecordTimer.doActivate(t)
91                         if result[1] == "stoponlycoming" or result[1] == "stopall":
92                                 t.disable()
93                         self.session.nav.RecordTimer.timeChanged(t)
94
95                         self.updateState()
96                         self.refill()
97                 
98         def updateState(self):
99                 if len(self.list) > 0:
100                         if self["timerlist"].getCurrent().disabled:
101                                 self["key_yellow"].setText(_("Enable"))
102                         else:
103                                 self["key_yellow"].setText(_("Disable"))
104                         self["key_yellow"].instance.invalidate()
105
106         def fillTimerList(self):
107                 del self.list[:]
108                 
109                 for timer in self.session.nav.RecordTimer.timer_list:
110                         self.list.append((timer, False))
111                 
112                 for timer in self.session.nav.RecordTimer.processed_timers:
113                         self.list.append((timer, True))
114                 self.list.sort(cmp = lambda x, y: x[0].begin < y[0].begin)
115
116         def showLog(self):
117                 cur=self["timerlist"].getCurrent()
118                 if cur:
119                         self.session.openWithCallback(self.finishedEdit, TimerLog, cur)
120
121         def openEdit(self):
122                 cur=self["timerlist"].getCurrent()
123                 if cur:
124                         self.session.openWithCallback(self.finishedEdit, TimerEntry, cur)
125
126         def cleanupQuestion(self):
127                 self.session.openWithCallback(self.cleanupTimer, MessageBox, _("Really delete done timers?"))
128         
129         def cleanupTimer(self, delete):
130                 if delete:
131                         self.session.nav.RecordTimer.cleanup()
132                         self.refill()
133
134         def removeTimerQuestion(self):
135                 if not self["timerlist"].getCurrent():
136                         return
137                 self.session.openWithCallback(self.removeTimer, MessageBox, _("Really delete this timer?"))
138
139         def removeTimer(self, result):
140                 if not result:
141                         return
142                 list = self["timerlist"]
143                 cur = list.getCurrent()
144                 if cur:
145                         timer = cur
146                         timer.afterEvent = AFTEREVENT.NONE
147                         self.session.nav.RecordTimer.removeEntry(timer)
148                         self.refill()
149         
150         def refill(self):
151                 oldsize = len(self.list)
152                 self.fillTimerList()
153                 lst = self["timerlist"]
154                 newsize = len(self.list)
155                 if oldsize and oldsize != newsize:
156                         idx = lst.getCurrentIndex()
157                         lst.entryRemoved(idx)
158                 else:
159                         lst.invalidate()
160         
161         def addCurrentTimer(self):
162                 event = None
163                 service = self.session.nav.getCurrentService()
164                 if service is not None:
165                         info = service.info()
166                         if info is not None:
167                                 event = info.getEvent(0)
168
169                 # FIXME only works if already playing a service
170                 serviceref = ServiceReference(self.session.nav.getCurrentlyPlayingServiceReference())
171                 
172                 if event is None:       
173                         data = (int(time()), int(time() + 60), "", "", None)
174                 else:
175                         data = parseEvent(event, description = False)
176
177                 self.addTimer(RecordTimerEntry(serviceref, checkOldTimers = True, *data))
178                 
179         def addTimer(self, timer):
180                 self.session.openWithCallback(self.finishedAdd, TimerEntry, timer)
181                 
182         def finishedEdit(self, answer):
183                 print "finished edit"
184                 
185                 if answer[0]:
186                         print "Edited timer"
187                         timersanitycheck = TimerSanityCheck(self.session.nav.RecordTimer.timer_list, answer[1])
188                         if not timersanitycheck.check():
189                                 print "Sanity check failed"
190                         else:
191                                 print "Sanity check passed"
192                         self.session.nav.RecordTimer.timeChanged(answer[1])
193                         self.fillTimerList()
194                 else:
195                         print "Timeredit aborted"
196
197         def finishedAdd(self, answer):
198                 print "finished add"
199                 if answer[0]:
200                         timersanitycheck = TimerSanityCheck(self.session.nav.RecordTimer.timer_list, answer[1])
201                         if not timersanitycheck.check():
202                                 print "Sanity check failed"
203                                 self.session.openWithCallback(self.finishSanityCorrection, TimerSanityConflict, timersanitycheck.getSimulTimerList())
204                         else:
205                                 print "Sanity check passed"
206                                 entry = answer[1]
207                                 self.session.nav.RecordTimer.record(entry)
208                                 self.fillTimerList()
209                 else:
210                         print "Timeredit aborted"               
211
212         def finishSanityCorrection(self, answer):
213                 self.finishedAdd(answer)
214
215         def leave(self):
216                 self.session.nav.RecordTimer.on_state_change.remove(self.onStateChange)
217                 self.close()
218
219         def onStateChange(self, entry):
220                 self.refill()
221
222 class TimerSanityConflict(Screen):
223         def __init__(self, session, timer):
224                 Screen.__init__(self, session)
225                 self.timer = timer
226                 print "TimerSanityConflict", timer
227                         
228                 self["timer1"] = TimerList(self.getTimerList(timer[0]))
229                 if len(timer) > 1:
230                         self["timer2"] = TimerList(self.getTimerList(timer[1]))
231                 else:
232                         self["timer2"] = TimerList([])
233                 
234                 self.list = []
235                 count = 0
236                 for x in timer:
237                         if count != 0:
238                                 self.list.append((_("Conflicting timer") + " " + str(count), x))
239                         count += 1
240
241                 self["list"] = MenuList(self.list)
242                 
243                 self["key_red"] = Button("Edit")
244                 self["key_green"] = Button("Disable")
245                 self["key_yellow"] = Button("Edit")
246                 self["key_blue"] = Button("Disable")
247
248                 self["actions"] = ActionMap(["OkCancelActions", "DirectionActions", "ShortcutActions", "TimerEditActions"], 
249                         {
250                                 "ok": self.close,
251                                 #"cancel": self.leave,
252                                 "red": self.editTimer1,
253                                 "green": self.disableTimer1,
254 #                               "yellow": self.editTimer2,
255 #                               "blue": self.disableTimer2,
256                                 #"log": self.showLog,
257                                 #"left": self.left,
258                                 #"right": self.right,
259                                 "up": self.up,
260                                 "down": self.down
261                         }, -1)
262
263         def getTimerList(self, timer):
264                 return [(timer, False)]
265
266         def editTimer1(self):
267                 self.session.openWithCallback(self.finishedEdit, TimerEntry, self["timer1"].getCurrent())
268
269         def disableTimer1(self):
270                 self.timer[0].disabled = True
271                 self.finishedEdit((True, self.timer[0]))
272
273         def finishedEdit(self, answer):
274                 self.close((True, self.timer[0]))
275
276         def up(self):
277                 self["list"].instance.moveSelection(self["list"].instance.moveUp)
278                 self["timer2"].l.setList(self.getTimerList(self["list"].getCurrent()))
279                 
280         def down(self):
281                 self["list"].instance.moveSelection(self["list"].instance.moveDown)
282                 self["timer2"].l.setList(self.getTimerList(self["list"].getCurrent()))