add possibility to select between different list styles via context menu
[enigma2.git] / lib / python / Components / Converter / ServiceName.py
index 9ccbda5..97b81f6 100644 (file)
@@ -1,36 +1,46 @@
 # -*- coding: utf-8 -*-
 from Components.Converter.Converter import Converter
-from enigma import iServiceInformation, iPlayableService
+from enigma import iServiceInformation, iPlayableService, iPlayableServicePtr
 from Components.Element import cached
 
 class ServiceName(Converter, object):
        NAME = 0
        PROVIDER = 1
+       REFERENCE = 2
 
        def __init__(self, type):
                Converter.__init__(self, type)
                if type == "Provider":
                        self.type = self.PROVIDER
+               elif type == "Reference":
+                       self.type = self.REFERENCE
                else:
                        self.type = self.NAME
 
-       def getServiceInfoValue(self, info, what):
-               v = info.getInfo(what)
-               if v != -2:
+       def getServiceInfoValue(self, info, what, ref=None):
+               v = ref and info.getInfo(ref, what) or info.getInfo(what)
+               if v != iServiceInformation.resIsString:
                        return "N/A"
-               return info.getInfoString(what)
+               return ref and info.getInfoString(ref, what) or info.getInfoString(what)
 
        @cached
        def getText(self):
                service = self.source.service
-               info = service and service.info()
+               if isinstance(service, iPlayableServicePtr):
+                       info = service and service.info()
+                       ref = None
+               else: # reference
+                       info = service and self.source.info
+                       ref = service
                if info is None:
                        return ""
-               
                if self.type == self.NAME:
-                       return info.getName().replace('\xc2\x86', '').replace('\xc2\x87', '')
+                       name = ref and info.getName(ref) or info.getName()
+                       return name.replace('\xc2\x86', '').replace('\xc2\x87', '')
                elif self.type == self.PROVIDER:
-                       return self.getServiceInfoValue(info, iServiceInformation.sProvider)
+                       return self.getServiceInfoValue(info, iServiceInformation.sProvider, ref)
+               elif self.type == self.REFERENCE:
+                       return self.getServiceInfoValue(info, iServiceInformation.sServiceref, ref)
 
        text = property(getText)