lib/python/Screens/EventView.py

   1 from Screen import Screen
   2 from Screens.TimerEdit import TimerSanityConflict
   3 from Screens.MessageBox import MessageBox
   4 from Components.ActionMap import ActionMap
   5 from Components.Button import Button
   6 from Components.Label import Label
   7 from Components.ScrollLabel import ScrollLabel
   8 from Components.TimerList import TimerList
   9 from Components.UsageConfig import preferredTimerPath
  10 from enigma import eEPGCache, eTimer, eServiceReference
  11 from RecordTimer import RecordTimerEntry, parseEvent, AFTEREVENT
  12 from TimerEntry import TimerEntry
  13 from time import localtime
  14 from Components.config import config
  15
  16 class EventViewBase:
  17         ADD_TIMER = 0
  18         REMOVE_TIMER = 1
  19
  20         def __init__(self, Event, Ref, callback=None, similarEPGCB=None):
  21                 self.similarEPGCB = similarEPGCB
  22                 self.cbFunc = callback
  23                 self.currentService=Ref
  24                 self.isRecording = (not Ref.ref.flags & eServiceReference.isGroup) and Ref.ref.getPath()
  25                 self.event = Event
  26                 self["epg_description"] = ScrollLabel()
  27                 self["datetime"] = Label()
  28                 self["channel"] = Label()
  29                 self["duration"] = Label()
  30                 self["key_red"] = Button("")
  31                 if similarEPGCB is not None:
  32                         self.SimilarBroadcastTimer = eTimer()
  33                         self.SimilarBroadcastTimer.callback.append(self.getSimilarEvents)
  34                 else:
  35                         self.SimilarBroadcastTimer = None
  36                 self.key_green_choice = self.ADD_TIMER
  37                 if self.isRecording:
  38                         self["key_green"] = Button("")
  39                 else:
  40                         self["key_green"] = Button(_("Add timer"))
  41                 self["key_yellow"] = Button("")
  42                 self["key_blue"] = Button("")
  43                 self["actions"] = ActionMap(["OkCancelActions", "EventViewActions"],
  44                         {
  45                                 "cancel": self.close,
  46                                 "ok": self.close,
  47                                 "pageUp": self.pageUp,
  48                                 "pageDown": self.pageDown,
  49                                 "prevEvent": self.prevEvent,
  50                                 "nextEvent": self.nextEvent,
  51                                 "timerAdd": self.timerAdd,
  52                                 "openSimilarList": self.openSimilarList
  53                         })
  54                 self.onShown.append(self.onCreate)
  55
  56         def onCreate(self):
  57                 self.setService(self.currentService)
  58                 self.setEvent(self.event)
  59
  60         def prevEvent(self):
  61                 if self.cbFunc is not None:
  62                         self.cbFunc(self.setEvent, self.setService, -1)
  63
  64         def nextEvent(self):
  65                 if self.cbFunc is not None:
  66                         self.cbFunc(self.setEvent, self.setService, +1)
  67
  68         def removeTimer(self, timer):
  69                 timer.afterEvent = AFTEREVENT.NONE
  70                 self.session.nav.RecordTimer.removeEntry(timer)
  71                 self["key_green"].setText(_("Add timer"))
  72                 self.key_green_choice = self.ADD_TIMER
  73
  74         def timerAdd(self):
  75                 if self.isRecording:
  76                         return
  77                 event = self.event
  78                 serviceref = self.currentService
  79                 if event is None:
  80                         return
  81                 eventid = event.getEventId()
  82                 refstr = serviceref.ref.toString()
  83                 for timer in self.session.nav.RecordTimer.timer_list:
  84                         if timer.eit == eventid and timer.service_ref.ref.toString() == refstr:
  85                                 cb_func = lambda ret : not ret or self.removeTimer(timer)
  86                                 self.session.openWithCallback(cb_func, MessageBox, _("Do you really want to delete %s?") % event.getEventName())
  87                                 break
  88                 else:
  89                         newEntry = RecordTimerEntry(self.currentService, checkOldTimers = True, dirname = preferredTimerPath(), *parseEvent(self.event))
  90                         self.session.openWithCallback(self.finishedAdd, TimerEntry, newEntry)
  91
  92         def finishedAdd(self, answer):
  93                 print "finished add"
  94                 if answer[0]:
  95                         entry = answer[1]
  96                         simulTimerList = self.session.nav.RecordTimer.record(entry)
  97                         if simulTimerList is not None:
  98                                 for x in simulTimerList:
  99                                         if x.setAutoincreaseEnd(entry):
 100                                                 self.session.nav.RecordTimer.timeChanged(x)
 101                                 simulTimerList = self.session.nav.RecordTimer.record(entry)
 102                                 if simulTimerList is not None:
 103                                         self.session.openWithCallback(self.finishSanityCorrection, TimerSanityConflict, simulTimerList)
 104                         self["key_green"].setText(_("Remove timer"))
 105                         self.key_green_choice = self.REMOVE_TIMER
 106                 else:
 107                         self["key_green"].setText(_("Add timer"))
 108                         self.key_green_choice = self.ADD_TIMER
 109                         print "Timeredit aborted"
 110
 111         def finishSanityCorrection(self, answer):
 112                 self.finishedAdd(answer)
 113
 114         def setService(self, service):
 115                 self.currentService=service
 116                 if self.isRecording:
 117                         self["channel"].setText(_("Recording"))
 118                 else:
 119                         name = self.currentService.getServiceName()
 120                         if name is not None:
 121                                 self["channel"].setText(name)
 122                         else:
 123                                 self["channel"].setText(_("unknown service"))
 124
 125         def sort_func(self,x,y):
 126                 if x[1] < y[1]:
 127                         return -1
 128                 elif x[1] == y[1]:
 129                         return 0
 130                 else:
 131                         return 1
 132
 133         def setEvent(self, event):
 134                 self.event = event
 135                 if event is None:
 136                         return
 137                 text = event.getEventName()
 138                 short = event.getShortDescription()
 139                 ext = event.getExtendedDescription()
 140                 if short and short != text:
 141                         text += '\n\n' + short
 142                 if ext:
 143                         if text:
 144                                 text += '\n\n'
 145                         text += ext
 146
 147                 self.setTitle(event.getEventName())
 148                 self["epg_description"].setText(text)
 149                 self["datetime"].setText(event.getBeginTimeString())
 150                 self["duration"].setText(_("%d min")%(event.getDuration()/60))
 151                 self["key_red"].setText("")
 152                 if self.SimilarBroadcastTimer is not None:
 153                         self.SimilarBroadcastTimer.start(400,True)
 154
 155                 serviceref = self.currentService
 156                 eventid = self.event.getEventId()
 157                 refstr = serviceref.ref.toString()
 158                 isRecordEvent = False
 159                 for timer in self.session.nav.RecordTimer.timer_list:
 160                         if timer.eit == eventid and timer.service_ref.ref.toString() == refstr:
 161                                 isRecordEvent = True
 162                                 break
 163                 if isRecordEvent and self.key_green_choice != self.REMOVE_TIMER:
 164                         self["key_green"].setText(_("Remove timer"))
 165                         self.key_green_choice = self.REMOVE_TIMER
 166                 elif not isRecordEvent and self.key_green_choice != self.ADD_TIMER:
 167                         self["key_green"].setText(_("Add timer"))
 168                         self.key_green_choice = self.ADD_TIMER
 169
 170
 171         def pageUp(self):
 172                 self["epg_description"].pageUp()
 173
 174         def pageDown(self):
 175                 self["epg_description"].pageDown()
 176
 177         def getSimilarEvents(self):
 178          # search similar broadcastings
 179                 refstr = str(self.currentService)
 180                 id = self.event.getEventId()
 181                 epgcache = eEPGCache.getInstance()
 182                 ret = epgcache.search(('NB', 100, eEPGCache.SIMILAR_BROADCASTINGS_SEARCH, refstr, id))
 183                 if ret is not None:
 184                         descr = self["epg_description"]
 185                         text = descr.getText()
 186                         text += '\n\n' + _('Similar broadcasts:')
 187                         ret.sort(self.sort_func)
 188                         for x in ret:
 189                                 t = localtime(x[1])
 190                                 text += '\n%d.%d.%d, %02d:%02d  -  %s'%(t[2], t[1], t[0], t[3], t[4], x[0])
 191                         descr.setText(text)
 192                         self["key_red"].setText(_("Similar"))
 193
 194         def openSimilarList(self):
 195                 if self.similarEPGCB is not None and self["key_red"].getText():
 196                         id = self.event and self.event.getEventId()
 197                         refstr = str(self.currentService)
 198                         if id is not None:
 199                                 self.similarEPGCB(id, refstr)
 200
 201 class EventViewSimple(Screen, EventViewBase):
 202         def __init__(self, session, Event, Ref, callback=None, similarEPGCB=None):
 203                 Screen.__init__(self, session)
 204                 self.skinName = "EventView"
 205                 EventViewBase.__init__(self, Event, Ref, callback, similarEPGCB)
 206
 207 class EventViewEPGSelect(Screen, EventViewBase):
 208         def __init__(self, session, Event, Ref, callback=None, singleEPGCB=None, multiEPGCB=None, similarEPGCB=None):
 209                 Screen.__init__(self, session)
 210                 self.skinName = "EventView"
 211                 EventViewBase.__init__(self, Event, Ref, callback, similarEPGCB)
 212                 self["key_yellow"].setText(_("Single EPG"))
 213                 self["key_blue"].setText(_("Multi EPG"))
 214                 self["epgactions"] = ActionMap(["EventViewEPGActions"],
 215                         {
 216                                 "openSingleServiceEPG": singleEPGCB,
 217                                 "openMultiServiceEPG": multiEPGCB,
 218                         })