code.vuplus.com Git - vuplus_dvbapp/blob - lib/python/Screens/TimerEntry.py

   1 from Screen import Screen
   2 import ChannelSelection
   3 from ServiceReference import ServiceReference
   4 from Components.config import ConfigSelection, ConfigText, ConfigSubList, ConfigDateTime, ConfigClock, ConfigYesNo, getConfigListEntry
   5 from Components.ActionMap import NumberActionMap
   6 from Components.ConfigList import ConfigListScreen
   7 from Components.MenuList import MenuList
   8 from Components.Button import Button
   9 from Components.Label import Label
  10 from Components.Pixmap import Pixmap
  11 from Screens.ChoiceBox import ChoiceBox
  12 from RecordTimer import AFTEREVENT
  13 from enigma import eEPGCache
  14 import time
  15 import datetime
  16
  17 class TimerEntry(Screen, ConfigListScreen):
  18         def __init__(self, session, timer):
  19                 Screen.__init__(self, session)
  20                 self.timer = timer
  21
  22                 self.entryStartDate = None
  23                 self.entryEndDate = None
  24                 self.entryService = None
  25
  26                 self["oktext"] = Label(_("OK"))
  27                 self["canceltext"] = Label(_("Cancel"))
  28                 self["ok"] = Pixmap()
  29                 self["cancel"] = Pixmap()
  30
  31                 self.createConfig()
  32
  33                 self["actions"] = NumberActionMap(["SetupActions"],
  34                 {
  35                         "ok": self.keySelect,
  36                         "save": self.keyGo,
  37                         "cancel": self.keyCancel,
  38                 }, -2)
  39
  40                 self.list = []
  41                 ConfigListScreen.__init__(self, self.list, session = session)
  42                 self.createSetup("config")
  43
  44         def createConfig(self):
  45                         justplay = self.timer.justplay
  46
  47                         afterevent = { AFTEREVENT.NONE: "nothing", AFTEREVENT.DEEPSTANDBY: "deepstandby", AFTEREVENT.STANDBY: "standby"}[self.timer.afterEvent]
  48
  49                         weekday_table = ["mon", "tue", "wed", "thu", "fri", "sat", "sun"]
  50
  51                         # calculate default values
  52                         day = []
  53                         weekday = 0
  54                         for x in range(0,7):
  55                                 day.append(0)
  56                         if self.timer.repeated: # repeated
  57                                 type = "repeated"
  58                                 if (self.timer.repeated == 31): # Mon-Fri
  59                                         repeated = "weekdays"
  60                                 elif (self.timer.repeated == 127): # daily
  61                                         repeated = "daily"
  62                                 else:
  63                                         flags = self.timer.repeated
  64                                         repeated = "user"
  65                                         count = 0
  66                                         for x in range(0, 7):
  67                                                 if flags == 1: # weekly
  68                                                         print "Set to weekday " + str(x)
  69                                                         weekday = x
  70                                                 if flags & 1 == 1: # set user defined flags
  71                                                         day[x] = 1
  72                                                         count += 1
  73                                                 else:
  74                                                         day[x] = 0
  75                                                 flags = flags >> 1
  76                                         if count == 1:
  77                                                 repeated = "weekly"
  78                         else: # once
  79                                 type = "once"
  80                                 repeated = None
  81                                 weekday = (int(time.strftime("%w", time.localtime(self.timer.begin))) - 1) % 7
  82                                 day[weekday] = 1
  83
  84                         self.timerentry_justplay = ConfigSelection(choices = [("zap", _("zap")), ("record", _("record"))], default = {0: "record", 1: "zap"}[justplay])
  85                         self.timerentry_afterevent = ConfigSelection(choices = [("nothing", _("do nothing")), ("standby", _("go to standby")), ("deepstandby", _("go to deep standby"))], default = afterevent)
  86                         self.timerentry_type = ConfigSelection(choices = [("once",_("once")), ("repeated", _("repeated"))], default = type)
  87                         self.timerentry_name = ConfigText(default = self.timer.name, fixed_size = False)
  88                         self.timerentry_description = ConfigText(default = self.timer.description, fixed_size = False)
  89
  90                         self.timerentry_repeated = ConfigSelection(default = repeated, choices = [("daily", _("daily")), ("weekly", _("weekly")), ("weekdays", _("Mon-Fri")), ("user", _("user defined"))])
  91
  92                         self.timerentry_startdate = ConfigDateTime(default = self.timer.begin, formatstring = _("%d.%B %Y"), increment = 86400)
  93                         self.timerentry_starttime = ConfigClock(default = self.timer.begin)
  94
  95                         self.timerentry_enddate = ConfigDateTime(default = self.timer.end, formatstring =  _("%d.%B %Y"), increment = 86400)
  96                         self.timerentry_endtime = ConfigClock(default = self.timer.end)
  97
  98                         self.timerentry_repeatedbegindate = ConfigDateTime(default = self.timer.repeatedbegindate, formatstring = _("%d.%B %Y"), increment = 86400)
  99
 100                         self.timerentry_weekday = ConfigSelection(default = weekday_table[weekday], choices = [("mon",_("Monday")), ("tue", _("Tuesday")), ("wed",_("Wednesday")), ("thu", _("Thursday")), ("fri", _("Friday")), ("sat", _("Saturday")), ("sun", _("Sunday"))])
 101
 102                         self.timerentry_day = ConfigSubList()
 103                         for x in range(0,7):
 104                                 self.timerentry_day.append(ConfigYesNo(default = day[x]))
 105
 106                         # FIXME some service-chooser needed here
 107                         servicename = "N/A"
 108                         try: # no current service available?
 109                                 servicename = str(self.timer.service_ref.getServiceName())
 110                         except:
 111                                 pass
 112                         self.timerentry_service = ConfigSelection([servicename])
 113
 114                         self.timerentry_startdate.addNotifier(self.checkDate)
 115                         self.timerentry_enddate.addNotifier(self.checkDate)
 116
 117         def checkDate(self, configElement):
 118                 if configElement is self.timerentry_startdate:
 119                         if self.timerentry_enddate.value < self.timerentry_startdate.value:
 120                                 self.timerentry_enddate.value = self.timerentry_startdate.value
 121                                 self["config"].invalidate(self.entryEndDate)
 122                 if configElement is self.timerentry_enddate:
 123                         if (self.timerentry_enddate.value < self.timerentry_startdate.value):
 124                                 self.timerentry_startdate.value = self.timerentry_enddate.value
 125                                 self["config"].invalidate(self.entryStartDate)
 126
 127         def createSetup(self, widget):
 128                 self.list = []
 129                 self.list.append(getConfigListEntry(_("Name"), self.timerentry_name))
 130                 self.list.append(getConfigListEntry(_("Description"), self.timerentry_description))
 131                 self.timerJustplayEntry = getConfigListEntry(_("Timer Type"), self.timerentry_justplay)
 132                 self.list.append(self.timerJustplayEntry)
 133                 self.timerTypeEntry = getConfigListEntry(_("Repeat Type"), self.timerentry_type)
 134                 self.list.append(self.timerTypeEntry)
 135
 136                 if self.timerentry_type.value == "once":
 137                         self.frequencyEntry = None
 138                 else: # repeated
 139                         self.frequencyEntry = getConfigListEntry(_("Frequency"), self.timerentry_repeated)
 140                         self.list.append(self.frequencyEntry)
 141                         self.repeatedbegindateEntry = getConfigListEntry(_("Starting on"), self.timerentry_repeatedbegindate)
 142                         self.list.append(self.repeatedbegindateEntry)
 143                         if self.timerentry_repeated.value == "daily":
 144                                 pass
 145                         if self.timerentry_repeated.value == "weekdays":
 146                                 pass
 147                         if self.timerentry_repeated.value == "weekly":
 148                                 self.list.append(getConfigListEntry(_("Weekday"), self.timerentry_weekday))
 149
 150                         if self.timerentry_repeated.value == "user":
 151                                 self.list.append(getConfigListEntry(_("Monday"), self.timerentry_day[0]))
 152                                 self.list.append(getConfigListEntry(_("Tuesday"), self.timerentry_day[1]))
 153                                 self.list.append(getConfigListEntry(_("Wednesday"), self.timerentry_day[2]))
 154                                 self.list.append(getConfigListEntry(_("Thursday"), self.timerentry_day[3]))
 155                                 self.list.append(getConfigListEntry(_("Friday"), self.timerentry_day[4]))
 156                                 self.list.append(getConfigListEntry(_("Saturday"), self.timerentry_day[5]))
 157                                 self.list.append(getConfigListEntry(_("Sunday"), self.timerentry_day[6]))
 158
 159                         #self.list.append(getConfigListEntry("StartDate", self.timerentry_startdate))
 160 #               self.list.append(getConfigListEntry("Weekday", self.timerentry_weekday))
 161
 162                 self.entryStartDate = getConfigListEntry(_("Start"), self.timerentry_startdate)
 163                 if self.timerentry_type.value == "once":
 164                         self.list.append(self.entryStartDate)
 165                         self.list.append(getConfigListEntry(" ", self.timerentry_starttime))
 166                 else:
 167                         self.list.append(getConfigListEntry(_("StartTime"), self.timerentry_starttime))
 168
 169                 self.entryEndDate = getConfigListEntry(_("End"), self.timerentry_enddate)
 170                 if self.timerentry_type.value == "once":
 171                         if self.timerentry_justplay.value != "zap":
 172                                 self.list.append(self.entryEndDate)
 173                                 self.list.append(getConfigListEntry(" ", self.timerentry_endtime))
 174                 else:
 175                         if self.timerentry_justplay.value != "zap":
 176                                 self.list.append(getConfigListEntry(_("EndTime"), self.timerentry_endtime))
 177
 178                 if self.timerentry_justplay.value != "zap":
 179                         self.list.append(getConfigListEntry(_("After event"), self.timerentry_afterevent))
 180
 181                 self.channelEntry = getConfigListEntry(_("Channel"), self.timerentry_service)
 182                 self.list.append(self.channelEntry)
 183
 184                 self[widget].list = self.list
 185                 self[widget].l.setList(self.list)
 186
 187         def newConfig(self):
 188                 print "newConfig", self["config"].getCurrent()
 189                 if self["config"].getCurrent() == self.timerTypeEntry:
 190                         self.createSetup("config")
 191                 if self["config"].getCurrent() == self.timerJustplayEntry:
 192                         self.createSetup("config")
 193                 if self["config"].getCurrent() == self.frequencyEntry:
 194                         self.createSetup("config")
 195
 196         def keyLeft(self):
 197                 if self["config"].getCurrent() is self.channelEntry:
 198                         self.keySelect()
 199                 else:
 200                         ConfigListScreen.keyLeft(self)
 201                         self.newConfig()
 202
 203         def keyRight(self):
 204                 if self["config"].getCurrent() is self.channelEntry:
 205                         self.keySelect()
 206                 else:
 207                         ConfigListScreen.keyRight(self)
 208                         self.newConfig()
 209
 210         def keySelect(self):
 211                 if self["config"].getCurrent() == self.channelEntry:
 212                         self.session.openWithCallback(self.finishedChannelSelection, ChannelSelection.SimpleChannelSelection, _("Select channel to record from"))
 213                 else:
 214                         self.keyGo()
 215
 216         def finishedChannelSelection(self, *args):
 217                 if len(args):
 218                         self.timer.service_ref = ServiceReference(args[0])
 219                         self.timerentry_service.setCurrentText(self.timer.service_ref.getServiceName())
 220                         self["config"].invalidate(self.channelEntry)
 221
 222         def getTimestamp(self, date, mytime):
 223                 d = time.localtime(date)
 224                 dt = datetime.datetime(d.tm_year, d.tm_mon, d.tm_mday, mytime[0], mytime[1])
 225                 return int(time.mktime(dt.timetuple()))
 226
 227         def getBeginEnd(self):
 228                 enddate = self.timerentry_enddate.value
 229                 endtime = self.timerentry_endtime.value
 230
 231                 startdate = self.timerentry_startdate.value
 232                 starttime = self.timerentry_starttime.value
 233
 234                 begin = self.getTimestamp(startdate, starttime)
 235                 end = self.getTimestamp(enddate, endtime)
 236
 237                 # because of the dateChecks, startdate can't be < enddate.
 238                 # however, the endtime can be less than the starttime.
 239                 # in this case, add 1 day.
 240                 if end < begin:
 241                         end += 86400
 242                 return begin, end
 243
 244         def keyGo(self):
 245                 self.timer.name = self.timerentry_name.value
 246                 self.timer.description = self.timerentry_description.value
 247                 self.timer.justplay = self.timerentry_justplay.value == "zap"
 248                 self.timer.resetRepeated()
 249                 self.timer.afterEvent = {"nothing": AFTEREVENT.NONE, "deepstandby": AFTEREVENT.DEEPSTANDBY, "standby": AFTEREVENT.STANDBY}[self.timerentry_afterevent.value]
 250
 251                 if self.timerentry_type.value == "once":
 252                         self.timer.begin, self.timer.end = self.getBeginEnd()
 253                 if self.timerentry_type.value == "repeated":
 254                         self.timer.repeatedbegindate = self.timerentry_repeatedbegindate.value
 255                         if self.timerentry_repeated.value == "daily":
 256                                 for x in range(0,7):
 257                                         self.timer.setRepeated(x)
 258
 259                         if self.timerentry_repeated.value == "weekly":
 260                                 self.timer.setRepeated(self.timerentry_weekday.index)
 261
 262                         if self.timerentry_repeated.value == "weekdays":
 263                                 for x in range(0,5):
 264                                         self.timer.setRepeated(x)
 265
 266                         if self.timerentry_repeated.value == "user":
 267                                 for x in range(0,7):
 268                                         if self.timerentry_day[x].value:
 269                                                 self.timer.setRepeated(x)
 270
 271                         self.timer.begin = self.getTimestamp(time.time(), self.timerentry_starttime.value)
 272                         self.timer.end = self.getTimestamp(time.time(), self.timerentry_endtime.value)
 273
 274                         # when a timer end is set before the start, add 1 day
 275                         if self.timer.end < self.timer.begin:
 276                                 self.timer.end += 86400
 277
 278                 if self.timer.eit is not None:
 279                         event = eEPGCache.getInstance().lookupEventId(self.timer.service_ref.ref, self.timer.eit)
 280                         if event is not None:
 281                                 n = event.getNumOfLinkageServices()
 282                                 if n > 0:
 283                                         tlist = []
 284                                         ref = self.session.nav.getCurrentlyPlayingServiceReference()
 285                                         parent = self.timer.service_ref.ref
 286                                         selection = 0
 287                                         for x in range(n):
 288                                                 i = event.getLinkageService(parent, x)
 289                                                 if i.toString() == ref.toString():
 290                                                         selection = x
 291                                                 tlist.append((i.getName(), i))
 292                                         self.session.openWithCallback(self.subserviceSelected, ChoiceBox, title=_("Please select a subservice to record..."), list = tlist, selection = selection)
 293                                         return
 294                 self.close((True, self.timer))
 295
 296         def subserviceSelected(self, service):
 297                 if not service is None:
 298                         self.timer.service_ref = ServiceReference(service[1])
 299                 self.close((True, self.timer))
 300
 301         def keyCancel(self):
 302                 self.close((False,))
 303
 304 class TimerLog(Screen):
 305         def __init__(self, session, timer):
 306                 Screen.__init__(self, session)
 307                 self.timer = timer;
 308                 self.log_entries = self.timer.log_entries[:]
 309
 310                 self.fillLogList()
 311
 312                 self["loglist"] = MenuList(self.list)
 313                 self["logentry"] = Label()
 314
 315                 self["key_red"] = Button(_("Delete entry"))
 316                 self["key_green"] = Button()
 317                 self["key_yellow"] = Button("")
 318                 self["key_blue"] = Button(_("Clear log"))
 319
 320                 self.onShown.append(self.updateText)
 321
 322                 self["actions"] = NumberActionMap(["OkCancelActions", "DirectionActions", "ColorActions"],
 323                 {
 324                         "ok": self.keyClose,
 325                         "cancel": self.keyClose,
 326                         "up": self.up,
 327                         "down": self.down,
 328                         "left": self.left,
 329                         "right": self.right,
 330                         "red": self.deleteEntry,
 331                         "blue": self.clearLog
 332                 }, -1)
 333
 334         def deleteEntry(self):
 335                 self.log_entries.remove(self["loglist"].getCurrent()[1])
 336                 self.fillLogList()
 337                 self["loglist"].l.setList(self.list)
 338                 self.updateText()
 339
 340         def fillLogList(self):
 341                 self.list = [ ]
 342                 for x in self.log_entries:
 343                         self.list.append((str(time.strftime("%Y-%m-%d %H-%M", time.localtime(x[0])) + " - " + x[2]), x))
 344
 345         def clearLog(self):
 346                 self.log_entries = []
 347                 self.fillLogList()
 348                 self["loglist"].l.setList(self.list)
 349                 self.updateText()
 350
 351         def keyClose(self):
 352                 if self.timer.log_entries != self.log_entries:
 353                         self.timer.log_entries = self.log_entries
 354                         self.close((True, self.timer))
 355                 else:
 356                         self.close((False,))
 357
 358         def up(self):
 359                 self["loglist"].instance.moveSelection(self["loglist"].instance.moveUp)
 360                 self.updateText()
 361
 362         def down(self):
 363                 self["loglist"].instance.moveSelection(self["loglist"].instance.moveDown)
 364                 self.updateText()
 365
 366         def left(self):
 367                 self["loglist"].instance.moveSelection(self["loglist"].instance.pageUp)
 368                 self.updateText()
 369
 370         def right(self):
 371                 self["loglist"].instance.moveSelection(self["loglist"].instance.pageDown)
 372                 self.updateText()
 373
 374         def updateText(self):
 375                 if len(self.list) > 0:
 376                         self["logentry"].setText(str(self["loglist"].getCurrent()[1][2]))
 377                 else:
 378                         self["logentry"].setText("")