lib/python/Screens/TimerEntry.py

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