0483f98b381cc4bf4c2f4b515e9b825fa6ce0d26
[enigma2.git] / lib / python / Screens / EpgSelection.py
1 from Screen import Screen
2 from Components.Button import Button
3 from Components.Pixmap import Pixmap
4 from Components.Label import Label
5 from Components.EpgList import *
6 from Components.ActionMap import ActionMap
7 from Components.ScrollLabel import ScrollLabel
8 from Screens.EventView import EventViewSimple
9 from TimeDateInput import TimeDateInput
10 from enigma import eServiceReference
11 from Screens.FixedMenu import FixedMenu
12 from RecordTimer import RecordTimerEntry, parseEvent
13 from TimerEdit import TimerEditList
14 from TimerEntry import TimerEntry
15 from ServiceReference import ServiceReference
16 from time import localtime, time
17
18 import xml.dom.minidom
19
20 class EPGSelection(Screen):
21         def __init__(self, session, service, zapFunc=None, eventid=None, bouquetChangeCB=None):
22                 Screen.__init__(self, session)
23                 self.bouquetChangeCB = bouquetChangeCB
24                 self.ask_time = -1 #now
25                 self["key_red"] = Button("")
26                 self.closeRecursive = False
27                 if isinstance(service, str) and eventid != None:
28                         self.type = EPG_TYPE_SIMILAR
29                         self["key_yellow"] = Button()
30                         self["key_blue"] = Button()
31                         self["key_red"] = Button()
32                         self.currentService=service
33                         self.eventid = eventid
34                         self.zapFunc = None
35                 elif isinstance(service, eServiceReference) or isinstance(service, str):
36                         self.type = EPG_TYPE_SINGLE
37                         self["key_yellow"] = Button()
38                         self["key_blue"] = Button()
39                         self.currentService=ServiceReference(service)
40                         self.zapFunc = None
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                 class ChannelActionMap(ActionMap):
63                         def action(self, contexts, action):
64                                 return ActionMap.action(self, contexts, action)
65
66                 self["actions"] = ChannelActionMap(["EPGSelectActions", "OkCancelActions"],
67                         {
68                                 "cancel": self.closeScreen,
69                                 "ok": self.eventSelected,
70                                 "timerAdd": self.timerAdd,
71                                 "yellow": self.yellowButtonPressed,
72                                 "blue": self.blueButtonPressed,
73                                 "info": self.infoKeyPressed,
74                                 "red": self.zapTo,
75                                 "input_date_time": self.enterDateTime,
76                                 "nextBouquet": self.nextBouquet,
77                                 "prevBouquet": self.prevBouquet
78                         })
79                 self["actions"].csel = self
80
81                 self.onLayoutFinish.append(self.onCreate)
82
83         def nextBouquet(self):
84                 if self.bouquetChangeCB:
85                         self.bouquetChangeCB(1, self)
86
87         def prevBouquet(self):
88                 if self.bouquetChangeCB:
89                         self.bouquetChangeCB(-1, self)
90
91         def enterDateTime(self):
92                 if self.type == EPG_TYPE_MULTI:
93                         self.session.openWithCallback(self.onDateTimeInputClosed, TimeDateInput)
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
162         def blueButtonPressed(self):
163                 if self.type == EPG_TYPE_MULTI:
164                         self["list"].updateMultiEPG(1)
165
166         def timerAdd(self):
167                 cur = self["list"].getCurrent()
168                 event = cur[0]
169                 serviceref = cur[1]
170                 if event is None:
171                         return
172                 newEntry = RecordTimerEntry(serviceref, checkOldTimers = True, *parseEvent(event))
173                 self.session.openWithCallback(self.timerEditFinished, TimerEntry, newEntry)
174
175         def timerEditFinished(self, answer):
176                 if answer[0]:
177                         self.session.nav.RecordTimer.record(answer[1])
178                 else:
179                         print "Timeredit aborted"       
180
181         def moveUp(self):
182                 self["list"].moveUp()
183
184         def moveDown(self):
185                 self["list"].moveDown()
186
187         def applyButtonState(self, state):
188                 if state == 0:
189                         self["now_button"].hide()
190                         self["now_button_sel"].hide()
191                         self["next_button"].hide()
192                         self["next_button_sel"].hide()
193                         self["more_button"].hide()
194                         self["more_button_sel"].hide()
195                         self["now_text"].hide()
196                         self["next_text"].hide()
197                         self["more_text"].hide()
198                         self["key_red"].setText("")
199                 else:
200                         if state == 1:
201                                 self["key_red"].setText("Zap")
202                                 self["now_button_sel"].show()
203                                 self["now_button"].hide()
204                         else:
205                                 self["now_button"].show()
206                                 self["now_button_sel"].hide()
207                                 self["key_red"].setText("")
208
209                         if state == 2:
210                                 self["next_button_sel"].show()
211                                 self["next_button"].hide()
212                         else:
213                                 self["next_button"].show()
214                                 self["next_button_sel"].hide()
215
216                         if state == 3:
217                                 self["more_button_sel"].show()
218                                 self["more_button"].hide()
219                         else:
220                                 self["more_button"].show()
221                                 self["more_button_sel"].hide()
222
223         def onSelectionChanged(self):
224                 if self.type == EPG_TYPE_MULTI:
225                         count = self["list"].getCurrentChangeCount()
226                         if self.ask_time != -1:
227                                 self.applyButtonState(0)
228                         elif count > 1:
229                                 self.applyButtonState(3)
230                         elif count > 0:
231                                 self.applyButtonState(2)
232                         else:
233                                 self.applyButtonState(1)
234                         days = [ _("Mon"), _("Tue"), _("Wed"), _("Thu"), _("Fri"), _("Sat"), _("Sun") ]
235                         datastr = ""
236                         event = self["list"].getCurrent()[0]
237                         if event is not None:
238                                 now = time()
239                                 beg = event.getBeginTime()
240                                 nowTime = localtime(now)
241                                 begTime = localtime(beg)
242                                 if nowTime[2] != begTime[2]:
243                                         datestr = '%s %d.%d.'%(days[begTime[6]], begTime[2], begTime[1])
244                                 else:
245                                         datestr = '%s %d.%d.'%(_("Today"), begTime[2], begTime[1])
246                         self["date"].setText(datestr)