1 from Screen import Screen
2 from Components.Button import Button
3 from Components.ServiceList import ServiceList
4 from Components.ActionMap import ActionMap
5 from EpgSelection import EPGSelection
6 from enigma import eServiceReference, eEPGCache, eEPGCachePtr, eServiceCenter, eServiceCenterPtr, iMutableServiceListPtr, iStaticServiceInformationPtr, eTimer
7 from Components.config import config, configElement, ConfigSubsection, configText
8 from Screens.FixedMenu import FixedMenu
10 import xml.dom.minidom
12 class BouquetSelector(FixedMenu):
13 def __init__(self, session, bouquets, parent):
17 entrys.append((x[0], self.bouquetSelected, x[1]))
18 FixedMenu.__init__(self, session, "Bouquetlist", entrys)
19 self.skinName = "Menu"
21 def bouquetSelected(self):
22 self.parent.addCurrentServiceToBouquet(self["menu"].getCurrent()[2])
25 class ChannelContextMenu(FixedMenu):
26 def __init__(self, session, csel):
31 inBouquetRootList = csel.servicelist.getRoot().toString().find('FROM BOUQUET "bouquets.') != -1 #FIXME HACK
32 inBouquet = csel.getMutableList() is not None
34 if not csel.bouquet_mark_edit and not csel.movemode and not inBouquetRootList:
35 menu.append(("add service to bouquet", self.addServiceToBouquetSelected))
37 menu.append(("remove service", self.removeCurrentService))
39 if inBouquet: # current list is editable?
40 if not csel.bouquet_mark_edit:
42 menu.append(("enable move mode", self.toggleMoveMode))
43 if not inBouquetRootList:
44 menu.append(("enable bouquet edit", self.bouquetMarkStart))
46 menu.append(("disable move mode", self.toggleMoveMode))
47 elif not inBouquetRootList:
48 menu.append(("end bouquet edit", self.bouquetMarkEnd))
49 menu.append(("abort bouquet edit", self.bouquetMarkAbort))
51 menu.append(("back", self.close))
53 FixedMenu.__init__(self, session, "Channel Selection", menu)
54 self.skinName = "Menu"
56 def addServiceToBouquetSelected(self):
58 serviceHandler = eServiceCenter.getInstance()
59 list = serviceHandler.list(self.csel.bouquet_root)
65 if ((s.flags & eServiceReference.flagDirectory) == eServiceReference.flagDirectory):
66 info = serviceHandler.info(s)
68 bouquets.append((info.getName(s), s))
70 if cnt > 1: # show bouquet list
71 self.session.open(BouquetSelector, bouquets, self)
72 elif cnt == 1: # add to only one existing bouquet
73 self.addCurrentServiceToBouquet(bouquet[0][1])
74 else: #no bouquets in root.. so assume only one favourite list is used
75 self.addCurrentServiceToBouquet(self.csel.bouquet_root)
77 def addCurrentServiceToBouquet(self, dest):
78 self.csel.addCurrentServiceToBouquet(dest)
81 def removeCurrentService(self):
82 self.csel.removeCurrentService()
85 def toggleMoveMode(self):
86 self.csel.toggleMoveMode()
89 def bouquetMarkStart(self):
90 self.csel.startMarkedEdit()
93 def bouquetMarkEnd(self):
94 self.csel.endMarkedEdit(abort=False)
97 def bouquetMarkAbort(self):
98 self.csel.endMarkedEdit(abort=True)
101 class ChannelSelection(Screen):
102 def lastService(self):
103 self.lastServiceTimer.stop()
104 #zap to last running tv service
105 self.setRoot(eServiceReference(config.tv.lastroot.value))
106 self.session.nav.playService(eServiceReference(config.tv.lastservice.value))
108 def __init__(self, session):
109 Screen.__init__(self, session)
111 #config for lastservice
112 config.tv = ConfigSubsection();
113 config.tv.lastservice = configElement("config.tv.lastservice", configText, "", 0);
114 config.tv.lastroot = configElement("config.tv.lastroot", configText, "", 0);
116 self.entry_marked = False
117 self.movemode = False
118 self.bouquet_mark_edit = False
119 self.bouquet_root = eServiceReference('1:7:1:0:0:0:0:0:0:0:(type == 1) FROM BOUQUET "bouquets.tv" ORDER BY bouquet')
120 self.mutableList = None
124 self["key_red"] = Button("All")
125 self["key_green"] = Button("Provider")
126 self["key_yellow"] = Button("Satellite")
127 self["key_blue"] = Button("Favourites")
129 self["list"] = ServiceList()
130 self.servicelist = self["list"]
132 if config.tv.lastroot.value == "":
133 self.servicelist.setRoot(eServiceReference("""1:0:1:0:0:0:0:0:0:0:(type == 1)"""))
134 #self["okbutton"] = Button("ok", [self.channelSelected])
136 self.lastServiceTimer = eTimer()
137 self.lastServiceTimer.timeout.get().append(self.lastService)
138 self.lastServiceTimer.start(100)
140 class ChannelActionMap(ActionMap):
141 def action(self, contexts, action):
142 if action[:7] == "bouquet":
144 if action.find("FROM BOUQUET") != -1:
145 l.setRoot(eServiceReference("1:7:1:0:0:0:0:0:0:0:" + action[8:]))
147 l.setRoot(eServiceReference("1:0:1:0:0:0:0:0:0:0:" + action[8:]))
149 if action == "cancel":
151 if l.movemode: #movemode active?
152 l.channelSelected() # unmark
153 l.toggleMoveMode() # disable move mode
154 elif l.bouquet_mark_edit:
155 l.endMarkedEdit(True) # abort edit mode
156 ActionMap.action(self, contexts, action)
158 self["actions"] = ChannelActionMap(["ChannelSelectActions", "OkCancelActions", "ContextMenuActions"],
160 "cancel": self.close,
161 "ok": self.channelSelected,
163 "contextMenu": self.doContext,
164 "showFavourites": self.showFavourites,
165 "showEPGList": self.showEPGList
167 self["actions"].csel = self
169 def showEPGList(self):
170 ref=self.servicelist.getCurrent()
171 ptr=eEPGCache.getInstance()
172 if ptr.startTimeQuery(ref) != -1:
173 self.session.open(EPGSelection, ref)
175 print 'no epg for service', ref.toString()
177 def getMutableList(self, root=eServiceReference()):
178 if not self.mutableList is None:
179 return self.mutableList
180 serviceHandler = eServiceCenter.getInstance()
182 root=self.servicelist.getRoot()
183 list = serviceHandler.list(root)
185 return list.startEdit()
188 # multiple marked entry stuff ( edit mode, later multiepg selection )
189 def startMarkedEdit(self):
190 self.mutableList = self.getMutableList()
191 # add all services from the current list to internal marked set in listboxservicecontent
192 self.bouquetRoot = self.servicelist.getRoot()
193 self.clearMarks() # this clears the internal marked set in the listboxservicecontent
194 self.bouquet_mark_edit = True
195 self.__marked = self.servicelist.getRootServices()
196 for x in self.__marked:
197 self.servicelist.addMarked(eServiceReference(x))
199 def endMarkedEdit(self, abort):
200 if not abort and self.mutableList is not None:
201 new_marked = set(self.servicelist.getMarked())
202 old_marked = set(self.__marked)
203 removed = old_marked - new_marked
204 added = new_marked - old_marked
208 self.mutableList.removeService(eServiceReference(x))
211 self.mutableList.addService(eServiceReference(x))
213 self.mutableList.flushChanges()
214 self.setRoot(self.bouquetRoot)
215 #self.showFavourites()
218 self.bouquet_mark_edit = False
219 self.bouquetRoot = None
220 self.mutableList = None
222 def clearMarks(self):
223 self.servicelist.clearMarks()
226 ref = self.servicelist.getCurrent()
227 if self.servicelist.isMarked(ref):
228 self.servicelist.removeMarked(ref)
230 self.servicelist.addMarked(ref)
232 def removeCurrentService(self):
233 ref = self.servicelist.getCurrent()
234 mutableList = self.getMutableList()
235 if ref.valid() and mutableList is not None:
236 if not mutableList.removeService(ref):
237 mutableList.flushChanges() #FIXME dont flush on each single removed service
238 self.setRoot(self.servicelist.getRoot())
240 def addCurrentServiceToBouquet(self, dest):
241 mutableList = self.getMutableList(dest)
242 if not mutableList is None:
243 if not mutableList.addService(self.servicelist.getCurrent()):
244 mutableList.flushChanges()
247 def searchNumberHelper(self, serviceHandler, num, bouquet):
248 servicelist = serviceHandler.list(bouquet)
249 if not servicelist is None:
251 serviceIterator = servicelist.getNext()
252 if not serviceIterator.valid(): #check end of list
254 if serviceIterator.flags: #assume normal dvb service have no flags set
257 if not num: #found service with searched number ?
258 return serviceIterator, 0
261 def zapToNumber(self, number):
262 bouquet = self.bouquet_root
264 serviceHandler = eServiceCenter.getInstance()
265 if bouquet.toString().find('FROM BOUQUET "bouquets.') == -1: #FIXME HACK
266 service, number = self.searchNumberHelper(serviceHandler, number, bouquet)
268 bouquetlist = serviceHandler.list(bouquet)
269 if not bouquetlist is None:
271 bouquet = bouquetlist.getNext()
272 if not bouquet.valid(): #check end of list
274 if ((bouquet.flags & eServiceReference.flagDirectory) != eServiceReference.flagDirectory):
276 service, number = self.searchNumberHelper(serviceHandler, number, bouquet)
277 if not service is None:
278 self.session.nav.playService(service) #play service
279 if self.servicelist.getRoot() != bouquet: #already in correct bouquet?
280 self.setRoot(bouquet)
281 self.servicelist.setCurrent(service) #select the service in servicelist
283 def getBouquetNumOffset(self, bouquet):
284 if self.bouquet_root.toString().find('FROM BOUQUET "bouquets.') == -1: #FIXME HACK
287 serviceHandler = eServiceCenter.getInstance()
288 bouquetlist = serviceHandler.list(self.bouquet_root)
289 if not bouquetlist is None:
291 bouquetIterator = bouquetlist.getNext()
292 if not bouquetIterator.valid() or bouquetIterator == bouquet: #end of list or bouquet found
294 if ((bouquetIterator.flags & eServiceReference.flagDirectory) != eServiceReference.flagDirectory):
296 servicelist = serviceHandler.list(bouquetIterator)
297 if not servicelist is None:
299 serviceIterator = servicelist.getNext()
300 if not serviceIterator.valid(): #check if end of list
302 if serviceIterator.flags: #playable services have no flags
307 def setRoot(self, root):
308 if not self.movemode: # dont change root when movemode is enabled
309 inBouquetRootList = root.toString().find('FROM BOUQUET "bouquets.') != -1 #FIXME HACK
310 if not inBouquetRootList and ((root.flags & eServiceReference.flagDirectory) == eServiceReference.flagDirectory):
311 self.servicelist.setMode(ServiceList.MODE_FAVOURITES)
312 self.servicelist.setNumberOffset(self.getBouquetNumOffset(root))
314 self.servicelist.setMode(ServiceList.MODE_NORMAL)
315 self.servicelist.setRoot(root)
317 def channelSelected(self):
318 ref = self.servicelist.getCurrent()
320 if self.entry_marked:
321 self.servicelist.setCurrentMarked(False)
322 self.entry_marked = False
324 self.servicelist.setCurrentMarked(True)
325 self.entry_marked = True
326 elif (ref.flags & 7) == 7:
328 elif self.bouquet_mark_edit:
331 self.session.nav.playService(ref)
337 self.session.nav.playService(self.servicelist.getCurrent())
341 self.servicelist.moveUp()
344 self.servicelist.moveDown()
347 self.session.open(ChannelContextMenu, self)
349 def toggleMoveMode(self):
351 if self.entry_marked:
352 self.channelSelected() # unmark current entry
353 self.movemode = False
354 self.mutableList.flushChanges() # FIXME add check if changes was made
355 self.mutableList = None
357 self.mutableList = self.getMutableList()
360 def showFavourites(self):
361 self.setRoot(self.bouquet_root)
363 def saveRoot(self, root):
365 config.tv.lastroot.value = root.toString()
366 config.tv.lastroot.save()
368 def saveChannel(self):
369 ref = self.session.nav.getCurrentlyPlayingServiceReference()
371 refstr = ref.toString()
374 config.tv.lastservice.value = refstr
375 config.tv.lastservice.save()