add graphical multiepg plugin (like e1)

[enigma2.git] / lib / python / Plugins / Extensions / GraphMultiEPG / plugin.py

diff --git a/lib/python/Plugins/Extensions/GraphMultiEPG/plugin.py b/lib/python/Plugins/Extensions/GraphMultiEPG/plugin.py
new file mode 100644 (file)
index 0000000..a247a61
--- /dev/null
+++ b/lib/python/Plugins/Extensions/GraphMultiEPG/plugin.py
@@ -0,0 +1,96 @@
+from Plugins.Plugin import PluginDescriptor
+from GraphMultiEpg import GraphMultiEPG
+from Screens.ChannelSelection import BouquetSelector
+from enigma import eServiceCenter, eServiceReference
+from ServiceReference import ServiceReference
+
+Session = None
+Servicelist = None
+
+bouquetSel = None
+epg_bouquet = None
+dlg_stack = [ ]
+
+def zapToService(service):
+       if not service is None:
+               if Servicelist.getRoot() != epg_bouquet: #already in correct bouquet?
+                       Servicelist.clearPath()
+                       if Servicelist.bouquet_root != epg_bouquet:
+                               Servicelist.enterPath(Servicelist.bouquet_root)
+                       Servicelist.enterPath(epg_bouquet)
+               Servicelist.setCurrentSelection(service) #select the service in Servicelist
+               Servicelist.zap()
+
+def getBouquetServices(bouquet):
+       services = [ ]
+       Servicelist = eServiceCenter.getInstance().list(bouquet)
+       if not Servicelist is None:
+               while True:
+                       service = Servicelist.getNext()
+                       if not service.valid(): #check if end of list
+                               break
+                       if service.flags & (eServiceReference.isDirectory | eServiceReference.isMarker): #ignore non playable services
+                               continue
+                       services.append(ServiceReference(service))
+       return services
+
+def cleanup():
+       global Session
+       Session = None
+       global Servicelist
+       Servicelist = None
+
+def closed(ret=False):
+       closedScreen = dlg_stack.pop()
+       if bouquetSel and closedScreen == bouquetSel:
+               global bouquetSel
+               bouquetSel = None
+       dlgs=len(dlg_stack)
+       if ret and dlgs > 0: # recursive close wished
+               dlg_stack[dlgs-1].close(dlgs > 1)
+       if dlgs <= 0:
+               cleanup()
+
+def openBouquetEPG(bouquet):
+       services = getBouquetServices(bouquet)
+       if len(services):
+               global epg_bouquet
+               epg_bouquet = bouquet
+               dlg_stack.append(Session.openWithCallback(closed, GraphMultiEPG, services, zapToService, changeBouquetCB))
+               return True
+       return False
+
+def changeBouquetCB(direction, epg):
+       if bouquetSel:
+               if direction > 0:
+                       bouquetSel.down()
+               else:
+                       bouquetSel.up()
+               bouquet = bouquetSel.getCurrent()
+               services = getBouquetServices(bouquet)
+               if len(services):
+                       epg_bouquet = bouquet
+                       epg.setServices(services)
+
+def main(session, servicelist, **kwargs):
+       global Session
+       Session = session
+       global Servicelist
+       Servicelist = servicelist
+       bouquets = Servicelist.getBouquetList()
+       if bouquets is None:
+               cnt = 0
+       else:
+               cnt = len(bouquets)
+       if cnt > 1: # show bouquet list
+               global bouquetSel
+               bouquetSel = Session.openWithCallback(closed, BouquetSelector, bouquets, openBouquetEPG, enableWrapAround=True)
+               dlg_stack.append(bouquetSel)
+       elif cnt == 1:
+               if not openBouquetEPG(bouquets[0][1]):
+                       cleanup()
+
+def Plugins(**kwargs):
+       name = _("Graphical Multi EPG")
+       descr = _("A graphical EPG for all services of an specific bouquet")
+       return [ PluginDescriptor(name=name, description=descr, where = PluginDescriptor.WHERE_EXTENSIONSMENU, fnc=main) ]