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