code.vuplus.com Git - vuplus_dvbapp/blob - lib/python/Screens/Menu.py

   1 from Screen import Screen
   2 from Components.Sources.List import List
   3 from Components.ActionMap import NumberActionMap
   4 from Components.Sources.StaticText import StaticText
   5 from Components.config import configfile
   6 from Components.PluginComponent import plugins
   7 from Components.config import config
   8 from Components.SystemInfo import SystemInfo
   9
  10 from Tools.Directories import resolveFilename, SCOPE_SKIN
  11
  12 import xml.etree.cElementTree
  13
  14 from Screens.Setup import Setup, getSetupTitle
  15
  16 #               <item text="TV-Mode">self.setModeTV()</item>
  17 #               <item text="Radio-Mode">self.setModeRadio()</item>
  18 #               <item text="File-Mode">self.setModeFile()</item>
  19 #                       <item text="Sleep Timer"></item>
  20
  21
  22 # read the menu
  23 mdom = xml.etree.cElementTree.parse(resolveFilename(SCOPE_SKIN, 'menu.xml'))
  24
  25 class boundFunction:
  26         def __init__(self, fnc, *args):
  27                 self.fnc = fnc
  28                 self.args = args
  29         def __call__(self):
  30                 self.fnc(*self.args)
  31
  32 class MenuUpdater:
  33         def __init__(self):
  34                 self.updatedMenuItems = {}
  35
  36         def addMenuItem(self, id, pos, text, module, screen, weight):
  37                 if not self.updatedMenuAvailable(id):
  38                         self.updatedMenuItems[id] = []
  39                 self.updatedMenuItems[id].append([text, pos, module, screen, weight])
  40
  41         def delMenuItem(self, id, pos, text, module, screen, weight):
  42                 self.updatedMenuItems[id].remove([text, pos, module, screen, weight])
  43
  44         def updatedMenuAvailable(self, id):
  45                 return self.updatedMenuItems.has_key(id)
  46
  47         def getUpdatedMenu(self, id):
  48                 return self.updatedMenuItems[id]
  49
  50 menuupdater = MenuUpdater()
  51
  52 class MenuSummary(Screen):
  53         skin = """
  54         <screen position="0,0" size="132,64">
  55                 <widget source="parent.title" render="Label" position="6,4" size="120,21" font="Regular;18" />
  56                 <widget source="parent.menu" render="Label" position="6,25" size="120,21" font="Regular;16">
  57                         <convert type="StringListSelection" />
  58                 </widget>
  59                 <widget source="global.CurrentTime" render="Label" position="56,46" size="82,18" font="Regular;16" >
  60                         <convert type="ClockToText">WithSeconds</convert>
  61                 </widget>
  62         </screen>"""
  63
  64         def __init__(self, session, parent):
  65                 Screen.__init__(self, session, parent)
  66
  67 class Menu(Screen):
  68
  69         ALLOW_SUSPEND = True
  70
  71         def okbuttonClick(self):
  72                 print "okbuttonClick"
  73                 selection = self["menu"].getCurrent()
  74                 if selection is not None:
  75                         selection[1]()
  76
  77         def execText(self, text):
  78                 exec text
  79
  80         def runScreen(self, arg):
  81                 # arg[0] is the module (as string)
  82                 # arg[1] is Screen inside this module
  83                 #        plus possible arguments, as
  84                 #        string (as we want to reference
  85                 #        stuff which is just imported)
  86                 # FIXME. somehow
  87                 if arg[0] != "":
  88                         exec "from " + arg[0] + " import *"
  89
  90                 self.openDialog(*eval(arg[1]))
  91
  92         def nothing(self): #dummy
  93                 pass
  94
  95         def openDialog(self, *dialog):                          # in every layer needed
  96                 self.session.openWithCallback(self.menuClosed, *dialog)
  97
  98         def openSetup(self, dialog):
  99                 self.session.openWithCallback(self.menuClosed, Setup, dialog)
 100
 101         def addMenu(self, destList, node):
 102                 requires = node.get("requires")
 103                 if requires and not SystemInfo.get(requires, False):
 104                         return
 105                 MenuTitle = _(node.get("text", "??").encode("UTF-8"))
 106                 entryID = node.get("entryID", "undefined")
 107                 weight = node.get("weight", 50)
 108                 x = node.get("flushConfigOnClose")
 109                 if x:
 110                         a = boundFunction(self.session.openWithCallback, self.menuClosedWithConfigFlush, Menu, node)
 111                 else:
 112                         a = boundFunction(self.session.openWithCallback, self.menuClosed, Menu, node)
 113                 #TODO add check if !empty(node.childNodes)
 114                 destList.append((MenuTitle, a, entryID, weight))
 115
 116         def menuClosedWithConfigFlush(self, *res):
 117                 configfile.save()
 118                 self.menuClosed(*res)
 119
 120         def menuClosed(self, *res):
 121                 if len(res) and res[0]:
 122                         self.close(True)
 123
 124         def addItem(self, destList, node):
 125                 requires = node.get("requires")
 126                 if requires and not SystemInfo.get(requires, False):
 127                         return
 128                 item_text = node.get("text", "").encode("UTF-8")
 129                 entryID = node.get("entryID", "undefined")
 130                 weight = node.get("weight", 50)
 131                 for x in node:
 132                         if x.tag == 'screen':
 133                                 module = x.get("module")
 134                                 screen = x.get("screen")
 135
 136                                 if screen is None:
 137                                         screen = module
 138
 139                                 print module, screen
 140                                 if module:
 141                                         module = "Screens." + module
 142                                 else:
 143                                         module = ""
 144
 145                                 # check for arguments. they will be appended to the
 146                                 # openDialog call
 147                                 args = x.text or ""
 148                                 screen += ", " + args
 149
 150                                 destList.append((_(item_text or "??"), boundFunction(self.runScreen, (module, screen)), entryID, weight))
 151                                 return
 152                         elif x.tag == 'code':
 153                                 destList.append((_(item_text or "??"), boundFunction(self.execText, x.text), entryID, weight))
 154                                 return
 155                         elif x.tag == 'setup':
 156                                 id = x.get("id")
 157                                 if item_text == "":
 158                                         item_text = _(getSetupTitle(id)) + "..."
 159                                 else:
 160                                         item_text = _(item_text)
 161                                 destList.append((item_text, boundFunction(self.openSetup, id), entryID, weight))
 162                                 return
 163                 destList.append((item_text, self.nothing, entryID, weight))
 164
 165
 166         def __init__(self, session, parent):
 167                 Screen.__init__(self, session)
 168
 169                 list = []
 170
 171                 menuID = None
 172                 for x in parent:                                                #walk through the actual nodelist
 173                         if x.tag == 'item':
 174                                 item_level = int(x.get("level", 0))
 175                                 if item_level <= config.usage.setup_level.index:
 176                                         self.addItem(list, x)
 177                                         count += 1
 178                         elif x.tag == 'menu':
 179                                 self.addMenu(list, x)
 180                                 count += 1
 181                         elif x.tag == "id":
 182                                 menuID = x.get("val")
 183                                 count = 0
 184
 185                         if menuID is not None:
 186                                 # menuupdater?
 187                                 if menuupdater.updatedMenuAvailable(menuID):
 188                                         for x in menuupdater.getUpdatedMenu(menuID):
 189                                                 if x[1] == count:
 190                                                         list.append((x[0], boundFunction(self.runScreen, (x[2], x[3] + ", ")), x[4]))
 191                                                         count += 1
 192
 193                 if menuID is not None:
 194                         # plugins
 195                         for l in plugins.getPluginsForMenu(menuID):
 196                                 # check if a plugin overrides an existing menu
 197                                 plugin_menuid = l[2]
 198                                 for x in list:
 199                                         if x[2] == plugin_menuid:
 200                                                 list.remove(x)
 201                                                 break
 202                                 list.append((l[0], boundFunction(l[1], self.session), l[2], l[3] or 50))
 203
 204                 # for the skin: first try a menu_<menuID>, then Menu
 205                 self.skinName = [ ]
 206                 if menuID is not None:
 207                         self.skinName.append("menu_" + menuID)
 208                 self.skinName.append("Menu")
 209
 210                 # Sort by Weight
 211                 list.sort(key=lambda x: int(x[3]))
 212
 213                 self["menu"] = List(list)
 214
 215                 self["actions"] = NumberActionMap(["OkCancelActions", "MenuActions", "NumberActions"],
 216                         {
 217                                 "ok": self.okbuttonClick,
 218                                 "cancel": self.closeNonRecursive,
 219                                 "menu": self.closeRecursive,
 220                                 "1": self.keyNumberGlobal,
 221                                 "2": self.keyNumberGlobal,
 222                                 "3": self.keyNumberGlobal,
 223                                 "4": self.keyNumberGlobal,
 224                                 "5": self.keyNumberGlobal,
 225                                 "6": self.keyNumberGlobal,
 226                                 "7": self.keyNumberGlobal,
 227                                 "8": self.keyNumberGlobal,
 228                                 "9": self.keyNumberGlobal
 229                         })
 230
 231                 a = parent.get("title", "").encode("UTF-8") or None
 232                 a = a and _(a)
 233                 if a is None:
 234                         a = _(parent.get("text", "").encode("UTF-8"))
 235                 self["title"] = StaticText(a)
 236                 self.menu_title = a
 237
 238         def keyNumberGlobal(self, number):
 239                 print "menu keyNumber:", number
 240                 # Calculate index
 241                 number -= 1
 242
 243                 if len(self["menu"].list) > number:
 244                         self["menu"].setIndex(number)
 245                         self.okbuttonClick()
 246
 247         def closeNonRecursive(self):
 248                 self.close(False)
 249
 250         def closeRecursive(self):
 251                 self.close(True)
 252
 253         def createSummary(self):
 254                 return MenuSummary
 255
 256 class MainMenu(Menu):
 257         #add file load functions for the xml-file
 258
 259         def __init__(self, *x):
 260                 self.skinName = "Menu"
 261                 Menu.__init__(self, *x)