lib/python/Screens/EpgSelection.py

   1 from Screen import Screen
   2 from Components.config import config, ConfigClock
   3 from Components.Button import Button
   4 from Components.Pixmap import Pixmap
   5 from Components.Label import Label
   6 from Components.EpgList import EPGList, EPG_TYPE_SINGLE, EPG_TYPE_SIMILAR, EPG_TYPE_MULTI
   7 from Components.ActionMap import ActionMap
   8 from Screens.EventView import EventViewSimple
   9 from TimeDateInput import TimeDateInput
  10 from enigma import eServiceReference
  11 from RecordTimer import RecordTimerEntry, parseEvent
  12 from TimerEntry import TimerEntry
  13 from ServiceReference import ServiceReference
  14 from time import localtime, time
  15
  16 mepg_config_initialized = False
  17
  18 class EPGSelection(Screen):
  19         def __init__(self, session, service, zapFunc=None, eventid=None, bouquetChangeCB=None):
  20                 Screen.__init__(self, session)
  21                 self.bouquetChangeCB = bouquetChangeCB
  22                 self.ask_time = -1 #now
  23                 self["key_red"] = Button("")
  24                 self.closeRecursive = False
  25                 if isinstance(service, str) and eventid != None:
  26                         self.type = EPG_TYPE_SIMILAR
  27                         self["key_yellow"] = Button()
  28                         self["key_blue"] = Button()
  29                         self["key_red"] = Button()
  30                         self.currentService=service
  31                         self.eventid = eventid
  32                         self.zapFunc = None
  33                 elif isinstance(service, eServiceReference) or isinstance(service, str):
  34                         self.type = EPG_TYPE_SINGLE
  35                         self["key_yellow"] = Button()
  36                         self["key_blue"] = Button()
  37                         self.currentService=ServiceReference(service)
  38                         self.zapFunc = None
  39                         self.sort_type = 0
  40                         self.setSortDescription()
  41                 else:
  42                         self.skinName = "EPGSelectionMulti"
  43                         self.type = EPG_TYPE_MULTI
  44                         self["key_yellow"] = Button(_("Prev"))
  45                         self["key_blue"] = Button(_("Next"))
  46                         self["now_button"] = Pixmap()
  47                         self["next_button"] = Pixmap()
  48                         self["more_button"] = Pixmap()
  49                         self["now_button_sel"] = Pixmap()
  50                         self["next_button_sel"] = Pixmap()
  51                         self["more_button_sel"] = Pixmap()
  52                         self["now_text"] = Label()
  53                         self["next_text"] = Label()
  54                         self["more_text"] = Label()
  55                         self["date"] = Label()
  56                         self.services = service
  57                         self.zapFunc = zapFunc
  58
  59                 self["key_green"] = Button(_("Add timer"))
  60                 self["list"] = EPGList(type = self.type, selChangedCB = self.onSelectionChanged, timer = self.session.nav.RecordTimer)
  61
  62                 self["actions"] = ActionMap(["EPGSelectActions", "OkCancelActions"],
  63                         {
  64                                 "cancel": self.closeScreen,
  65                                 "ok": self.eventSelected,
  66                                 "timerAdd": self.timerAdd,
  67                                 "yellow": self.yellowButtonPressed,
  68                                 "blue": self.blueButtonPressed,
  69                                 "info": self.infoKeyPressed,
  70                                 "red": self.zapTo,
  71                                 "input_date_time": self.enterDateTime,
  72                                 "nextBouquet": self.nextBouquet,
  73                                 "prevBouquet": self.prevBouquet
  74                         })
  75                 self["actions"].csel = self
  76
  77                 self.onLayoutFinish.append(self.onCreate)
  78
  79         def nextBouquet(self):
  80                 if self.bouquetChangeCB:
  81                         self.bouquetChangeCB(1, self)
  82
  83         def prevBouquet(self):
  84                 if self.bouquetChangeCB:
  85                         self.bouquetChangeCB(-1, self)
  86
  87         def enterDateTime(self):
  88                 if self.type == EPG_TYPE_MULTI:
  89                         global mepg_config_initialized
  90                         if not mepg_config_initialized:
  91                                 config.misc.prev_mepg_time=ConfigClock(default = time())
  92                                 mepg_config_initialized = True
  93                         self.session.openWithCallback(self.onDateTimeInputClosed, TimeDateInput, config.misc.prev_mepg_time )
  94
  95         def onDateTimeInputClosed(self, ret):
  96                 if len(ret) > 1:
  97                         if ret[0]:
  98                                 self.ask_time=ret[1]
  99                                 self["list"].fillMultiEPG(self.services, ret[1])
 100
 101         def closeScreen(self):
 102                 self.close(self.closeRecursive)
 103
 104         def infoKeyPressed(self):
 105                 cur = self["list"].getCurrent()
 106                 event = cur[0]
 107                 service = cur[1]
 108                 if event is not None:
 109                         if self.type != EPG_TYPE_SIMILAR:
 110                                 self.session.open(EventViewSimple, event, service, self.eventViewCallback, self.openSimilarList)
 111                         else:
 112                                 self.session.open(EventViewSimple, event, service, self.eventViewCallback)
 113
 114         def openSimilarList(self, eventid, refstr):
 115                 self.session.open(EPGSelection, refstr, None, eventid)
 116
 117         def setServices(self, services):
 118                 self.services = services
 119                 self.onCreate()
 120
 121         #just used in multipeg
 122         def onCreate(self):
 123                 l = self["list"]
 124                 l.recalcEntrySize()
 125                 if self.type == EPG_TYPE_MULTI:
 126                         l.fillMultiEPG(self.services, self.ask_time)
 127                 elif self.type == EPG_TYPE_SINGLE:
 128                         l.fillSingleEPG(self.currentService)
 129                 else:
 130                         l.fillSimilarList(self.currentService, self.eventid)
 131
 132         def eventViewCallback(self, setEvent, setService, val):
 133                 l = self["list"]
 134                 old = l.getCurrent()
 135                 if val == -1:
 136                         self.moveUp()
 137                 elif val == +1:
 138                         self.moveDown()
 139                 cur = l.getCurrent()
 140                 if self.type == EPG_TYPE_MULTI and cur[0] is None and cur[1].ref != old[1].ref:
 141                         self.eventViewCallback(setEvent, setService, val)
 142                 else:
 143                         setService(cur[1])
 144                         setEvent(cur[0])
 145
 146         def zapTo(self): # just used in multiepg
 147                 if self.zapFunc and self["key_red"].getText() == "Zap":
 148                         lst = self["list"]
 149                         count = lst.getCurrentChangeCount()
 150                         if count == 0:
 151                                 self.closeRecursive = True
 152                                 ref = lst.getCurrent()[1]
 153                                 self.zapFunc(ref.ref)
 154
 155         def eventSelected(self):
 156                 self.infoKeyPressed()
 157
 158         def yellowButtonPressed(self):
 159                 if self.type == EPG_TYPE_MULTI:
 160                         self["list"].updateMultiEPG(-1)
 161                 elif self.type == EPG_TYPE_SINGLE:
 162                         if self.sort_type == 0:
 163                                 self.sort_type = 1
 164                         else:
 165                                 self.sort_type = 0
 166                         self["list"].sortSingleEPG(self.sort_type)
 167                         self.setSortDescription()
 168
 169         def setSortDescription(self):
 170                 if self.sort_type == 1:
 171                         # TRANSLATORS: This must fit into the header button in the EPG-List
 172                         self["key_yellow"].setText(_("Sort Time"))
 173                 else:
 174                         # TRANSLATORS: This must fit into the header button in the EPG-List
 175                         self["key_yellow"].setText(_("Sort A-Z"))
 176
 177         def blueButtonPressed(self):
 178                 if self.type == EPG_TYPE_MULTI:
 179                         self["list"].updateMultiEPG(1)
 180
 181         def timerAdd(self):
 182                 cur = self["list"].getCurrent()
 183                 event = cur[0]
 184                 serviceref = cur[1]
 185                 if event is None:
 186                         return
 187                 newEntry = RecordTimerEntry(serviceref, checkOldTimers = True, dirname = config.movielist.last_timer_videodir.value, *parseEvent(event))
 188                 self.session.openWithCallback(self.timerEditFinished, TimerEntry, newEntry)
 189
 190         def timerEditFinished(self, answer):
 191                 if answer[0]:
 192                         self.session.nav.RecordTimer.record(answer[1])
 193                 else:
 194                         print "Timeredit aborted"
 195
 196         def moveUp(self):
 197                 self["list"].moveUp()
 198
 199         def moveDown(self):
 200                 self["list"].moveDown()
 201
 202         def applyButtonState(self, state):
 203                 if state == 0:
 204                         self["now_button"].hide()
 205                         self["now_button_sel"].hide()
 206                         self["next_button"].hide()
 207                         self["next_button_sel"].hide()
 208                         self["more_button"].hide()
 209                         self["more_button_sel"].hide()
 210                         self["now_text"].hide()
 211                         self["next_text"].hide()
 212                         self["more_text"].hide()
 213                         self["key_red"].setText("")
 214                 else:
 215                         if state == 1:
 216                                 self["key_red"].setText("Zap")
 217                                 self["now_button_sel"].show()
 218                                 self["now_button"].hide()
 219                         else:
 220                                 self["now_button"].show()
 221                                 self["now_button_sel"].hide()
 222                                 self["key_red"].setText("")
 223
 224                         if state == 2:
 225                                 self["next_button_sel"].show()
 226                                 self["next_button"].hide()
 227                         else:
 228                                 self["next_button"].show()
 229                                 self["next_button_sel"].hide()
 230
 231                         if state == 3:
 232                                 self["more_button_sel"].show()
 233                                 self["more_button"].hide()
 234                         else:
 235                                 self["more_button"].show()
 236                                 self["more_button_sel"].hide()
 237
 238         def onSelectionChanged(self):
 239                 if self.type == EPG_TYPE_MULTI:
 240                         count = self["list"].getCurrentChangeCount()
 241                         if self.ask_time != -1:
 242                                 self.applyButtonState(0)
 243                         elif count > 1:
 244                                 self.applyButtonState(3)
 245                         elif count > 0:
 246                                 self.applyButtonState(2)
 247                         else:
 248                                 self.applyButtonState(1)
 249                         days = [ _("Mon"), _("Tue"), _("Wed"), _("Thu"), _("Fri"), _("Sat"), _("Sun") ]
 250                         datastr = ""
 251                         event = self["list"].getCurrent()[0]
 252                         if event is not None:
 253                                 now = time()
 254                                 beg = event.getBeginTime()
 255                                 nowTime = localtime(now)
 256                                 begTime = localtime(beg)
 257                                 if nowTime[2] != begTime[2]:
 258                                         datestr = '%s %d.%d.'%(days[begTime[6]], begTime[2], begTime[1])
 259                                 else:
 260                                         datestr = '%s %d.%d.'%(_("Today"), begTime[2], begTime[1])
 261                         self["date"].setText(datestr)