1 from Screen import Screen
2 from Components.Button import Button
3 from Components.ServiceList import ServiceList
4 from Components.ActionMap import NumberActionMap
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
9 from Tools.NumericalTextInput import NumericalTextInput
10 from Components.NimManager import nimmanager
12 import xml.dom.minidom
14 class BouquetSelector(FixedMenu):
15 def __init__(self, session, bouquets, selectedFunc):
16 self.selectedFunc=selectedFunc
19 entrys.append((x[0], self.bouquetSelected, x[1]))
20 FixedMenu.__init__(self, session, "Bouquetlist", entrys)
21 self.skinName = "Menu"
23 def bouquetSelected(self):
24 self.selectedFunc(self["menu"].getCurrent()[2])
26 class ChannelContextMenu(FixedMenu):
27 def __init__(self, session, csel):
32 inBouquetRootList = csel.servicelist.getRoot().getPath().find('FROM BOUQUET "bouquets.') != -1 #FIXME HACK
33 inBouquet = csel.getMutableList() is not None
34 haveBouquets = csel.bouquet_root.getPath().find('FROM BOUQUET "bouquets.') != -1
36 if not csel.bouquet_mark_edit and not csel.movemode and not inBouquetRootList:
37 if (csel.getCurrentSelection().flags & eServiceReference.flagDirectory) != eServiceReference.flagDirectory:
39 menu.append((_("add service to bouquet"), self.addServiceToBouquetSelected))
41 menu.append((_("add service to favourites"), self.addServiceToBouquetSelected))
43 menu.append((_("remove service"), self.removeCurrentService))
45 if inBouquet: # current list is editable?
46 if not csel.bouquet_mark_edit:
48 menu.append((_("enable move mode"), self.toggleMoveMode))
49 if not inBouquetRootList:
51 menu.append((_("enable bouquet edit"), self.bouquetMarkStart))
53 menu.append((_("enable favourite edit"), self.bouquetMarkStart))
55 menu.append((_("disable move mode"), self.toggleMoveMode))
56 elif not inBouquetRootList:
58 menu.append((_("end bouquet edit"), self.bouquetMarkEnd))
59 menu.append((_("abort bouquet edit"), self.bouquetMarkAbort))
61 menu.append((_("end favourites edit"), self.bouquetMarkEnd))
62 menu.append((_("abort favourites edit"), self.bouquetMarkAbort))
64 menu.append((_("back"), self.close))
66 FixedMenu.__init__(self, session, _("Channel Selection"), menu)
67 self.skinName = "Menu"
69 def addServiceToBouquetSelected(self):
70 bouquets = self.csel.getBouquetList()
75 if cnt > 1: # show bouquet list
76 self.session.open(BouquetSelector, bouquets, self.addCurrentServiceToBouquet)
77 elif cnt == 1: # add to only one existing bouquet
78 self.addCurrentServiceToBouquet(bouquets[0][1])
79 else: #no bouquets in root.. so assume only one favourite list is used
80 self.addCurrentServiceToBouquet(self.csel.bouquet_root)
82 def addCurrentServiceToBouquet(self, dest):
83 self.csel.addCurrentServiceToBouquet(dest)
86 def removeCurrentService(self):
87 self.csel.removeCurrentService()
90 def toggleMoveMode(self):
91 self.csel.toggleMoveMode()
94 def bouquetMarkStart(self):
95 self.csel.startMarkedEdit()
98 def bouquetMarkEnd(self):
99 self.csel.endMarkedEdit(abort=False)
102 def bouquetMarkAbort(self):
103 self.csel.endMarkedEdit(abort=True)
106 class ChannelSelectionEdit:
108 self.entry_marked = False
109 self.movemode = False
110 self.bouquet_mark_edit = False
111 self.mutableList = None
113 self.saved_title = None
114 self.saved_root = None
116 def getMutableList(self, root=eServiceReference()):
117 if not self.mutableList is None:
118 return self.mutableList
119 serviceHandler = eServiceCenter.getInstance()
121 root=self.servicelist.getRoot()
122 list = serviceHandler.list(root)
124 return list.startEdit()
127 # multiple marked entry stuff ( edit mode, later multiepg selection )
128 def startMarkedEdit(self):
129 self.mutableList = self.getMutableList()
130 # add all services from the current list to internal marked set in listboxservicecontent
131 self.bouquetRoot = self.servicelist.getRoot()
132 self.clearMarks() # this clears the internal marked set in the listboxservicecontent
133 self.saved_title = self.instance.getTitle()
134 new_title = self.saved_title
135 if self.bouquet_root.getPath().find('FROM BOUQUET "bouquets.') != -1:
136 new_title += ' ' + _("[bouquet edit]")
138 new_title += ' ' + _("[favourite edit]")
139 self.instance.setTitle(new_title)
140 self.bouquet_mark_edit = True
141 self.__marked = self.servicelist.getRootServices()
142 for x in self.__marked:
143 self.servicelist.addMarked(eServiceReference(x))
144 self.saved_root = self.getRoot()
145 self.showAllServices()
147 def endMarkedEdit(self, abort):
148 if not abort and self.mutableList is not None:
149 new_marked = set(self.servicelist.getMarked())
150 old_marked = set(self.__marked)
151 removed = old_marked - new_marked
152 added = new_marked - old_marked
156 self.mutableList.removeService(eServiceReference(x))
159 self.mutableList.addService(eServiceReference(x))
161 self.mutableList.flushChanges()
164 self.bouquet_mark_edit = False
165 self.bouquetRoot = None
166 self.mutableList = None
167 self.instance.setTitle(self.saved_title)
168 self.saved_title = None
169 self.setRoot(self.saved_root)
171 def clearMarks(self):
172 self.servicelist.clearMarks()
175 ref = self.servicelist.getCurrent()
176 if self.servicelist.isMarked(ref):
177 self.servicelist.removeMarked(ref)
179 self.servicelist.addMarked(ref)
181 def removeCurrentService(self):
182 ref = self.servicelist.getCurrent()
183 mutableList = self.getMutableList()
184 if ref.valid() and mutableList is not None:
185 if not mutableList.removeService(ref):
186 currentIndex = self.servicelist.getCurrentIndex()
187 self.servicelist.moveDown()
188 if self.servicelist.getCurrentIndex() == currentIndex:
190 mutableList.flushChanges() #FIXME dont flush on each single removed service
191 self.setRoot(self.servicelist.getRoot())
192 self.servicelist.moveToIndex(currentIndex)
194 def addCurrentServiceToBouquet(self, dest):
195 mutableList = self.getMutableList(dest)
196 if not mutableList is None:
197 if not mutableList.addService(self.servicelist.getCurrent()):
198 mutableList.flushChanges()
201 def toggleMoveMode(self):
203 if self.entry_marked:
204 self.toggleMoveMarked() # unmark current entry
205 self.movemode = False
206 self.mutableList.flushChanges() # FIXME add check if changes was made
207 self.mutableList = None
208 self.instance.setTitle(self.saved_title)
209 self.saved_title = None
211 self.mutableList = self.getMutableList()
213 self.saved_title = self.instance.getTitle()
214 new_title = self.saved_title
215 new_title += ' ' + _("[move mode]");
216 self.instance.setTitle(new_title);
218 def handleEditCancel(self):
219 if self.movemode: #movemode active?
220 self.channelSelected() # unmark
221 self.toggleMoveMode() # disable move mode
222 elif self.bouquet_mark_edit:
223 self.endMarkedEdit(True) # abort edit mode
225 def toggleMoveMarked(self):
226 if self.entry_marked:
227 self.servicelist.setCurrentMarked(False)
228 self.entry_marked = False
230 self.servicelist.setCurrentMarked(True)
231 self.entry_marked = True
234 self.session.open(ChannelContextMenu, self)
236 USE_MULTIBOUQUETS = False
238 class ChannelSelectionBase(Screen):
239 def __init__(self, session):
240 Screen.__init__(self, session)
242 # this makes it much simple to implement a selectable radio or tv mode :)
243 self.service_types_tv = '1:7:1:0:0:0:0:0:0:0:(type == 1) || (type == 17)'
244 # self.service_types_radio = '1:7:1:0:0:0:0:0:0:0:(type == 2)'
246 self.service_types = self.service_types_tv
247 if USE_MULTIBOUQUETS:
248 self.bouquet_root = eServiceReference('1:7:1:0:0:0:0:0:0:0:(type == 1) FROM BOUQUET "bouquets.tv" ORDER BY bouquet')
250 self.bouquet_root = eServiceReference('%s FROM BOUQUET "userbouquet.favourites.tv" ORDER BY bouquet'%(self.service_types))
252 self["key_red"] = Button(_("All"))
253 self["key_green"] = Button(_("Satellites"))
254 self["key_yellow"] = Button(_("Provider"))
255 self["key_blue"] = Button(_("Favourites"))
257 self["list"] = ServiceList()
258 self.servicelist = self["list"]
260 #self["okbutton"] = Button("ok", [self.channelSelected])
262 self.numericalTextInput = NumericalTextInput()
264 def getBouquetNumOffset(self, bouquet):
265 if self.bouquet_root.getPath().find('FROM BOUQUET "bouquets.') == -1: #FIXME HACK
268 serviceHandler = eServiceCenter.getInstance()
269 bouquetlist = serviceHandler.list(self.bouquet_root)
270 if not bouquetlist is None:
272 bouquetIterator = bouquetlist.getNext()
273 if not bouquetIterator.valid() or bouquetIterator == bouquet: #end of list or bouquet found
275 if ((bouquetIterator.flags & eServiceReference.flagDirectory) != eServiceReference.flagDirectory):
277 servicelist = serviceHandler.list(bouquetIterator)
278 if not servicelist is None:
280 serviceIterator = servicelist.getNext()
281 if not serviceIterator.valid(): #check if end of list
283 if serviceIterator.flags: #playable services have no flags
288 def setRootBase(self, root, justSet=False):
289 path = root.getPath()
290 inBouquetRootList = path.find('FROM BOUQUET "bouquets.') != -1 #FIXME HACK
291 pos = path.find(' FROM BOUQUET')
292 isBouquet = pos != -1
293 if not inBouquetRootList and isBouquet:
294 self.servicelist.setMode(ServiceList.MODE_FAVOURITES)
295 self.servicelist.setNumberOffset(self.getBouquetNumOffset(root))
296 refstr = self.service_types + path[pos:]
297 root = eServiceReference(refstr)
299 self.servicelist.setMode(ServiceList.MODE_NORMAL)
300 self.servicelist.setRoot(root, justSet)
303 self.servicelist.moveUp()
306 self.servicelist.moveDown()
308 def showAllServices(self):
309 ref = eServiceReference('%s ORDER BY name'%(self.service_types))
312 def showSatellites(self):
313 ref = eServiceReference('%s FROM SATELLITES ORDER BY satellitePosition'%(self.service_types))
314 self.setRoot(ref, True)
315 serviceHandler = eServiceCenter.getInstance()
316 servicelist = serviceHandler.list(ref)
317 if not servicelist is None:
319 service = servicelist.getNext()
320 if not service.valid(): #check if end of list
322 orbpos = service.getData(4) >> 16
323 if service.getPath().find("FROM PROVIDER") != -1:
324 service_name = _("Providers")
326 service_name = _("Services")
328 service_name += str(' - %s'%(nimmanager.getSatDescription(orbpos)))
329 service.setName(service_name) # why we need this cast?
331 if orbpos > 1800: # west
332 service.setName("%s (%3.1f" + _("W") + ")" %(str, (0 - (orbpos - 3600)) / 10.0))
334 service.setName("%s (%3.1f" + _("E") + ")" % (str, orbpos / 10.0))
335 self.servicelist.addService(service)
336 self.servicelist.finishFill()
338 def showProviders(self):
339 ref = eServiceReference('%s FROM PROVIDERS ORDER BY name'%(self.service_types))
342 def showFavourites(self):
343 self.setRoot(self.bouquet_root)
345 def keyNumberGlobal(self, number):
346 char = self.numericalTextInput.getKey(number)
347 print "You pressed number " + str(number)
348 print "You would go to character " + str(char)
349 self.servicelist.moveToChar(char)
351 def enterBouquet(self, action):
352 if action[:7] == "bouquet":
353 if action.find("FROM BOUQUET") != -1:
354 self.setRoot(eServiceReference("1:7:1:0:0:0:0:0:0:0:" + action[8:]))
356 self.setRoot(eServiceReference("1:0:1:0:0:0:0:0:0:0:" + action[8:]))
361 return self.servicelist.getRoot()
363 def getCurrentSelection(self):
364 return self.servicelist.getCurrent()
366 def setCurrentSelection(self, service):
367 self.servicelist.setCurrent(service)
372 def getBouquetList(self):
375 serviceHandler = eServiceCenter.getInstance()
376 list = serviceHandler.list(self.bouquet_root)
382 if ((s.flags & eServiceReference.flagDirectory) == eServiceReference.flagDirectory):
383 info = serviceHandler.info(s)
385 bouquets.append((info.getName(s), s))
388 if len(bouquets) == 0 and serviceCount > 0:
389 info = serviceHandler.info(self.bouquet_root)
391 bouquets.append((info.getName(self.bouquet_root), self.bouquet_root))
395 class ChannelSelection(ChannelSelectionBase, ChannelSelectionEdit):
396 def __init__(self, session):
397 ChannelSelectionBase.__init__(self,session)
398 ChannelSelectionEdit.__init__(self)
400 #config for lastservice
401 config.tv = ConfigSubsection();
402 config.tv.lastservice = configElement("config.tv.lastservice", configText, "", 0);
403 config.tv.lastroot = configElement("config.tv.lastroot", configText, "", 0);
405 class ChannelActionMap(NumberActionMap):
406 def action(self, contexts, action):
407 if not self.csel.enterBouquet(action):
408 if action == "cancel":
409 self.csel.handleEditCancel()
410 NumberActionMap.action(self, contexts, action)
411 self["actions"] = ChannelActionMap(["ChannelSelectActions", "OkCancelActions", "ContextMenuActions"],
413 "cancel": self.cancel,
414 "ok": self.channelSelected,
416 "contextMenu": self.doContext,
417 "showFavourites": self.showFavourites,
418 "showAllServices": self.showAllServices,
419 "showProviders": self.showProviders,
420 "showSatellites": self.showSatellites,
421 "showEPGList": self.showEPGList,
422 "1": self.keyNumberGlobal,
423 "2": self.keyNumberGlobal,
424 "3": self.keyNumberGlobal,
425 "4": self.keyNumberGlobal,
426 "5": self.keyNumberGlobal,
427 "6": self.keyNumberGlobal,
428 "7": self.keyNumberGlobal,
429 "8": self.keyNumberGlobal,
430 "9": self.keyNumberGlobal,
431 "0": self.keyNumberGlobal
433 self["actions"].csel = self
434 self.onShown.append(self.onShow)
436 # self.onLayoutFinish.append(self.onCreate)
437 self.lastChannelRootTimer = eTimer()
438 self.lastChannelRootTimer.timeout.get().append(self.onCreate)
439 self.lastChannelRootTimer.start(100,True)
442 lastroot=eServiceReference(config.tv.lastroot.value)
444 self.setRoot(lastroot)
446 self.showFavourites()
447 self.saveRoot(self.getRoot())
448 lastservice=eServiceReference(config.tv.lastservice.value)
449 if lastservice.valid():
450 self.session.nav.playService(lastservice)
451 self.servicelist.setCurrent(lastservice)
454 ref = self.session.nav.getCurrentlyPlayingServiceReference()
455 if ref is not None and ref.valid() and ref.getPath() == "":
456 self.servicelist.setPlayableIgnoreService(ref)
458 self.servicelist.setPlayableIgnoreService(eServiceReference())
460 def showEPGList(self):
461 ref=self.getCurrentSelection()
462 ptr=eEPGCache.getInstance()
463 if ptr.startTimeQuery(ref) != -1:
464 self.session.open(EPGSelection, ref)
466 print 'no epg for service', ref.toString()
468 def channelSelected(self):
469 ref = self.getCurrentSelection()
471 self.toggleMoveMarked()
472 elif (ref.flags & 7) == 7:
474 elif self.bouquet_mark_edit:
480 def setRoot(self, root, justSet=False):
481 if not self.movemode:
482 self.setRootBase(root, justSet)
484 #called from infoBar and channelSelected
486 self.session.nav.playService(self.getCurrentSelection())
487 self.saveRoot(self.getRoot())
490 def saveRoot(self, root):
492 config.tv.lastroot.value = root.toString()
493 config.tv.lastroot.save()
495 def saveChannel(self):
496 ref = self.session.nav.getCurrentlyPlayingServiceReference()
498 refstr = ref.toString()
501 config.tv.lastservice.value = refstr
502 config.tv.lastservice.save()
506 lastroot=eServiceReference(config.tv.lastroot.value)
507 lastservice=eServiceReference(config.tv.lastservice.value)
508 if lastroot.valid() and self.getRoot() != lastroot:
509 self.setRoot(lastroot)
510 if lastservice.valid() and self.getCurrentSelection() != lastservice:
511 self.servicelist.setCurrent(lastservice)
513 class SimpleChannelSelection(ChannelSelectionBase):
514 def __init__(self, session, title):
515 ChannelSelectionBase.__init__(self, session)
517 self.onShown.append(self.onExecCallback)
519 class ChannelActionMap(NumberActionMap):
520 def action(self, contexts, action):
521 if not self.csel.enterBouquet(action):
522 NumberActionMap.action(self, contexts, action)
523 self["actions"] = ChannelActionMap(["ChannelSelectActions", "OkCancelActions", "ContextMenuActions"],
525 "cancel": self.cancel,
526 "ok": self.channelSelected,
527 "showFavourites": self.showFavourites,
528 "showAllServices": self.showAllServices,
529 "showProviders": self.showProviders,
530 "showSatellites": self.showSatellites,
531 "1": self.keyNumberGlobal,
532 "2": self.keyNumberGlobal,
533 "3": self.keyNumberGlobal,
534 "4": self.keyNumberGlobal,
535 "5": self.keyNumberGlobal,
536 "6": self.keyNumberGlobal,
537 "7": self.keyNumberGlobal,
538 "8": self.keyNumberGlobal,
539 "9": self.keyNumberGlobal,
540 "0": self.keyNumberGlobal
542 self["actions"].csel = self
544 def onExecCallback(self):
545 print "onExecCallback"
546 self.showFavourites()
547 self.session.currentDialog.instance.setTitle(self.title)
549 def channelSelected(self): # just return selected service
550 ref = self.getCurrentSelection()
553 def setRoot(self, root):
554 self.setRootBase(root)