lib/python/Screens/ChoiceBox.py

   1 from enigma import *
   2 from Screens.Screen import Screen
   3 from Screens.MessageBox import MessageBox
   4 from Components.ActionMap import NumberActionMap
   5 from Components.Label import Label
   6 from Components.MenuList import MenuList
   7 from Components.GUIComponent import *
   8 from Components.ChoiceList import ChoiceEntryComponent, ChoiceList
   9
  10 import os
  11
  12 class ChoiceBox(Screen):
  13         def __init__(self, session, title = "", list = [], keys = None, selection = 0):
  14                 Screen.__init__(self, session)
  15
  16                 self["text"] = Label(title)
  17                 self.list = []
  18                 if keys is None:
  19                         self.keys = [ "1", "2", "3", "4", "5", "6", "7", "8", "9", "0", "red", "green", "yellow", "blue" ] + (len(list) - 10) * [""]
  20                 else:
  21                         self.keys = keys + (len(list) - len(keys)) * [""]
  22
  23                 self.keymap = {}
  24                 pos = 0
  25                 for x in list:
  26                         strpos = str(self.keys[pos])
  27                         self.list.append(ChoiceEntryComponent(key = strpos, text = x))
  28                         if self.keys[pos] != "":
  29                                 self.keymap[self.keys[pos]] = list[pos]
  30                         pos += 1
  31                 self["list"] = ChoiceList(list = self.list, selection = selection)
  32
  33                 self["actions"] = NumberActionMap(["WizardActions", "InputActions", "ColorActions", "DirectionActions"],
  34                 {
  35                         "ok": self.go,
  36                         "back": self.cancel,
  37                         "1": self.keyNumberGlobal,
  38                         "2": self.keyNumberGlobal,
  39                         "3": self.keyNumberGlobal,
  40                         "4": self.keyNumberGlobal,
  41                         "5": self.keyNumberGlobal,
  42                         "6": self.keyNumberGlobal,
  43                         "7": self.keyNumberGlobal,
  44                         "8": self.keyNumberGlobal,
  45                         "9": self.keyNumberGlobal,
  46                         "0": self.keyNumberGlobal,
  47                         "red": self.keyRed,
  48                         "green": self.keyGreen,
  49                         "yellow": self.keyYellow,
  50                         "blue": self.keyBlue,
  51                         "up": self.up,
  52                         "down": self.down
  53                 }, -1)
  54
  55         def keyLeft(self):
  56                 pass
  57
  58         def keyRight(self):
  59                 pass
  60
  61         def up(self):
  62                 if len(self["list"].list) > 0:
  63                         while 1:
  64                                 self["list"].instance.moveSelection(self["list"].instance.moveUp)
  65                                 if self["list"].l.getCurrentSelection()[0][0] != "--" or self["list"].l.getCurrentSelectionIndex() == 0:
  66                                         break
  67
  68         def down(self):
  69                 if len(self["list"].list) > 0:
  70                         while 1:
  71                                 self["list"].instance.moveSelection(self["list"].instance.moveDown)
  72                                 if self["list"].l.getCurrentSelection()[0][0] != "--" or self["list"].l.getCurrentSelectionIndex() == len(self["list"].list) - 1:
  73                                         break
  74
  75
  76         def keyNumberGlobal(self, number):
  77                 print "pressed", number
  78                 if self.keymap.has_key(str(number)):
  79                         self.close(self.keymap[str(number)])
  80
  81         def go(self):
  82                 if len(self["list"].list) > 0:
  83                         self.close(self["list"].l.getCurrentSelection()[0])
  84                 else:
  85                         self.close(None)
  86
  87         def keyRed(self):
  88                 if self.keymap.has_key("red"):
  89                         self.close(self.keymap["red"])
  90
  91         def keyGreen(self):
  92                 if self.keymap.has_key("green"):
  93                         self.close(self.keymap["green"])
  94
  95         def keyYellow(self):
  96                 if self.keymap.has_key("yellow"):
  97                         self.close(self.keymap["yellow"])
  98
  99         def keyBlue(self):
 100                 if self.keymap.has_key("blue"):
 101                         self.close(self.keymap["blue"])
 102
 103         def cancel(self):
 104                 self.close(None)