delete correct entry in timerlist
[vuplus_dvbapp] / 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 from Components.TimerSanityCheck import TimerSanityCheck
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.removeTimer,
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                 self["timerlist"].getCurrent()[0].disabled = not self["timerlist"].getCurrent()[0].disabled
65                 self.updateState()
66                 self.refill()
67                 
68         def updateState(self):
69                 if len(self.list) > 0:
70                         if self["timerlist"].getCurrent()[0].disabled:
71                                 self["key_yellow"].setText(_("Enable"))
72                         else:
73                                 self["key_yellow"].setText(_("Disable"))
74                         self["key_yellow"].instance.invalidate()
75
76         def fillTimerList(self):
77                 del self.list[:]
78                 
79                 for timer in self.session.nav.RecordTimer.timer_list:
80                         self.list.append(TimerEntryComponent(timer, processed=False))
81                 
82                 for timer in self.session.nav.RecordTimer.processed_timers:
83                         self.list.append(TimerEntryComponent(timer, processed=True))
84
85         def showLog(self):
86                 self.session.openWithCallback(self.finishedEdit, TimerLog, self["timerlist"].getCurrent()[0])
87
88         def openEdit(self):
89                 self.session.openWithCallback(self.finishedEdit, TimerEntry, self["timerlist"].getCurrent()[0])
90                 #self.session.open(TimerEdit, self["timerlist"].getCurrent()[0])
91                 
92         def cleanupQuestion(self):
93                 self.session.openWithCallback(self.cleanupTimer, MessageBox, _("Really delete done timers?"))
94         
95         def cleanupTimer(self, delete):
96                 if delete:
97                         self.session.nav.RecordTimer.cleanup()
98                         self.refill()
99                 
100         def removeTimer(self):
101                 list = self["timerlist"]
102                 timer = list.getCurrent()[0]
103                 currentIndex = list.getCurrentIndex()
104                 self.session.nav.RecordTimer.removeEntry(timer)
105                 list.moveDown()
106                 if list.getCurrentIndex() == currentIndex:
107                         currentIndex -= 1
108                 self.refill()
109                 list.moveToIndex(currentIndex)
110         
111         def refill(self):
112                 self.fillTimerList()
113                 self["timerlist"].invalidate()
114         
115         def addCurrentTimer(self):
116                 event = None
117                 service = self.session.nav.getCurrentService()
118                 if service is not None:
119                         info = service.info()
120                         if info is not None:
121                                 event = info.getEvent(0)
122
123                 # FIXME only works if already playing a service
124                 serviceref = ServiceReference(self.session.nav.getCurrentlyPlayingServiceReference())
125                 
126                 if event is None:       
127                         data = (int(time()), int(time() + 60), "unknown event", "", None)
128                 else:
129                         data = parseEvent(event)
130
131                 self.addTimer(RecordTimerEntry(serviceref, *data))
132                 
133         def addTimer(self, timer):
134                 self.session.openWithCallback(self.finishedAdd, TimerEntry, timer)
135                 
136         def finishedEdit(self, answer):
137                 print "finished edit"
138                 
139                 if answer[0]:
140                         print "Edited timer"
141                         if not TimerSanityCheck(self.session.nav.RecordTimer.timer_list, answer[1]).check():
142                                 print "Sanity check failed"
143                         else:
144                                 print "Sanity check passed"
145                         self.session.nav.RecordTimer.timeChanged(answer[1])
146                         self.fillTimerList()
147                 else:
148                         print "Timeredit aborted"
149
150         def finishedAdd(self, answer):
151                 print "finished add"
152                 if answer[0]:
153                         if not TimerSanityCheck(self.session.nav.RecordTimer.timer_list, answer[1]).check():
154                                 print "Sanity check failed"
155                         else:
156                                 print "Sanity check passed"
157                         entry = answer[1]
158                         self.session.nav.RecordTimer.record(entry)
159                         self.fillTimerList()
160                 else:
161                         print "Timeredit aborted"               
162
163         def leave(self):
164                 self.session.nav.RecordTimer.saveTimer()
165                 self.session.nav.RecordTimer.on_state_change.remove(self.onStateChange)
166                 self.close()
167
168         def onStateChange(self, entry):
169                 self.refill()