small optimizations and cleanups by Moritz Venn
[enigma2.git] / lib / python / Screens / Wizard.py
old mode 100644 (file)
new mode 100755 (executable)
index 17a8150..555110a
@@ -1,9 +1,8 @@
 from Screen import Screen
 
-import string
-
 from Screens.HelpMenu import HelpableScreen
-from Components.config import config, KEY_LEFT, KEY_RIGHT, KEY_DELETE, KEY_BACKSPACE
+from Components.config import config, KEY_LEFT, KEY_RIGHT, KEY_HOME, KEY_END, KEY_0, KEY_DELETE, KEY_BACKSPACE, KEY_OK, KEY_TOGGLEOW, KEY_ASCII, KEY_TIMEOUT, KEY_NUMBERS
+
 from Components.Label import Label
 from Components.Slider import Slider
 from Components.ActionMap import NumberActionMap
@@ -86,9 +85,9 @@ class Wizard(Screen):
                                        timeoutstep = ''
                                self.wizard[self.lastStep] = {"id": id, "condition": "", "text": "", "timeout": timeout, "timeoutaction": timeoutaction, "timeoutstep": timeoutstep, "list": [], "config": {"screen": None, "args": None, "type": "" }, "code": "", "codeafter": "", "code_async": "", "codeafter_async": "", "nextstep": nextstep}
                        elif (name == "text"):
-                               self.wizard[self.lastStep]["text"] = string.replace(str(attrs.get('value')), "\\n", "\n")
+                               self.wizard[self.lastStep]["text"] = str(attrs.get('value')).replace("\\n", "\n")
                        elif (name == "displaytext"):
-                               self.wizard[self.lastStep]["displaytext"] = string.replace(str(attrs.get('value')), "\\n", "\n")
+                               self.wizard[self.lastStep]["displaytext"] = str(attrs.get('value')).replace("\\n", "\n")
                        elif (name == "list"):
                                if (attrs.has_key('type')):
                                        if attrs["type"] == "dynamic":
@@ -215,8 +214,9 @@ class Wizard(Screen):
                
                self.disableKeys = False
                
-               self["actions"] = NumberActionMap(["WizardActions", "NumberActions", "ColorActions", "SetupActions"],
+               self["actions"] = NumberActionMap(["WizardActions", "NumberActions", "ColorActions", "SetupActions", "InputAsciiActions"],
                {
+                       "gotAsciiCode": self.keyGotAscii,
                        "ok": self.ok,
                        "back": self.back,
                        "left": self.left,
@@ -307,37 +307,42 @@ class Wizard(Screen):
 
        def finished(self, gotoStep = None, *args, **kwargs):
                print "finished"
-               self.prevStep = self.currStep
-               self.gotoStep = gotoStep
+               currStep = self.currStep
 
                if self.updateValues not in self.onShown:
                        self.onShown.append(self.updateValues)
-
+                       
                if self.showConfig:
-                       if self.wizard[self.prevStep]["config"]["type"] == "dynamic":
-                               eval("self." + self.wizard[self.prevStep]["config"]["evaluation"])()
+                       if self.wizard[currStep]["config"]["type"] == "dynamic":
+                               eval("self." + self.wizard[currStep]["config"]["evaluation"])()
 
                if self.showList:
-                       if (len(self.wizard[self.prevStep]["evaluatedlist"]) > 0):
+                       if (len(self.wizard[currStep]["evaluatedlist"]) > 0):
                                print "current:", self["list"].current
                                nextStep = self["list"].current[1]
-                               if (self.wizard[self.prevStep].has_key("listevaluation")):
+                               if (self.wizard[currStep].has_key("listevaluation")):
                                        exec("self." + self.wizard[self.currStep]["listevaluation"] + "('" + nextStep + "')")
                                else:
                                        self.currStep = self.getStepWithID(nextStep)
 
-               if ((self.prevStep == self.numSteps and self.wizard[self.prevStep]["nextstep"] is None) or self.wizard[self.prevStep]["id"] == "end"): # wizard finished
+               print_now = True
+               if ((currStep == self.numSteps and self.wizard[currStep]["nextstep"] is None) or self.wizard[currStep]["id"] == "end"): # wizard finished
                        print "wizard finished"
                        self.markDone()
                        self.close()
                else:
-                       self.codeafterA=True
-                       self.runCode(self.wizard[self.prevStep]["codeafter"])
-                       if self.runCode(self.wizard[self.prevStep]["codeafter_async"]):
+                       self.codeafter = True
+                       self.runCode(self.wizard[currStep]["codeafter"])
+                       self.prevStep = currStep
+                       self.gotoStep = gotoStep
+                       if not self.runCode(self.wizard[currStep]["codeafter_async"]):
+                               self.afterAsyncCode()
+                       else:
                                if self.updateValues in self.onShown:
                                        self.onShown.remove(self.updateValues)
-                       else:
-                               self.afterAsyncCode()
+
+               if print_now:
+                       print "Now: " + str(self.currStep)
 
        def ok(self):
                print "OK"
@@ -363,6 +368,10 @@ class Wizard(Screen):
        def keyNumberGlobal(self, number):
                if (self.wizard[self.currStep]["config"]["screen"] != None):
                        self.configInstance.keyNumberGlobal(number)
+
+       def keyGotAscii(self):
+               if (self.wizard[self.currStep]["config"]["screen"] != None):
+                       self["config"].handleKey(KEY_ASCII)
                
        def left(self):
                self.resetCounter()
@@ -460,41 +469,53 @@ class Wizard(Screen):
                        self.configInstance.doClose()
                        self.configInstance = None
 
+               self.condition = True
                exec (self.wizard[self.currStep]["condition"])
-
-               if len(self.stepHistory) == 0 or self.stepHistory[-1] != self.currStep:
-                       self.stepHistory.append(self.currStep)
-               print "wizard step:", self.wizard[self.currStep]
+               if not self.condition:
+                       self.currStep += 1
+                       self.updateValues()
+               else:
+                       if self.wizard[self.currStep].has_key("displaytext"):
+                               displaytext = self.wizard[self.currStep]["displaytext"]
+                               print "set LCD text"
+                               for x in self.lcdCallbacks:
+                                       x(displaytext)
+                       if len(self.stepHistory) == 0 or self.stepHistory[-1] != self.currStep:
+                               self.stepHistory.append(self.currStep)
+                       print "wizard step:", self.wizard[self.currStep]
                        
-               if self.showSteps:
-                       self["step"].setText(self.getTranslation("Step ") + str(self.currStep) + "/" + str(self.numSteps))
-               if self.showStepSlider:
-                       self["stepslider"].setValue(self.currStep)
+                       if self.showSteps:
+                               self["step"].setText(self.getTranslation("Step ") + str(self.currStep) + "/" + str(self.numSteps))
+                       if self.showStepSlider:
+                               self["stepslider"].setValue(self.currStep)
                
-               if self.wizard[self.currStep]["timeout"] is not None:
-                       self.resetCounter() 
-                       self.timeoutTimer.start(1000)
+                       if self.wizard[self.currStep]["timeout"] is not None:
+                               self.resetCounter() 
+                               self.timeoutTimer.start(1000)
                        
-               print "wizard text", self.getTranslation(self.wizard[self.currStep]["text"])
-               self.updateText(firstset = True)
-               if self.wizard[self.currStep].has_key("displaytext"):
-                       displaytext = self.wizard[self.currStep]["displaytext"]
-                       print "set LCD text"
-                       for x in self.lcdCallbacks:
-                               x(displaytext)
+                       print "wizard text", self.getTranslation(self.wizard[self.currStep]["text"])
+                       self.updateText(firstset = True)
+                       if self.wizard[self.currStep].has_key("displaytext"):
+                               displaytext = self.wizard[self.currStep]["displaytext"]
+                               print "set LCD text"
+                               for x in self.lcdCallbacks:
+                                       x(displaytext)
                                
-               self.codeafterA=False
-               self.runCode(self.wizard[self.currStep]["code"])
-               if self.runCode(self.wizard[self.currStep]["code_async"]):
-                       if self.updateValues in self.onShown:
-                               self.onShown.remove(self.updateValues)
-               else:
-                       self.afterAsyncCode()
+                       self.codeafter=False
+                       self.runCode(self.wizard[self.currStep]["code"])
+                       if self.runCode(self.wizard[self.currStep]["code_async"]):
+                               if self.updateValues in self.onShown:
+                                       self.onShown.remove(self.updateValues)
+                       else:
+                               self.afterAsyncCode()
 
        def afterAsyncCode(self):
-               if self.codeafterA:
+               if not self.updateValues in self.onShown:
+                       self.onShown.append(self.updateValues)
+
+               if self.codeafter:
                        if self.wizard[self.prevStep]["nextstep"] is not None:
-                               self.currStep = self.getStepWithID(self.wizard[self.currStep]["nextstep"])
+                               self.currStep = self.getStepWithID(self.wizard[self.prevStep]["nextstep"])
                        if self.gotoStep is not None:
                                self.currStep = self.getStepWithID(self.gotoStep)
                        self.currStep += 1
@@ -583,10 +604,7 @@ class WizardManager:
                self.wizards.append((wizard, precondition, priority))
        
        def getWizards(self):
-               list = []
-               for x in self.wizards:
-                       if x[1] == 1: # precondition
-                               list.append((x[2], x[0]))
-               return list
+               # x[1] is precondition
+               return [(x[2], x[0]) for x in self.wizards if x[1] == 1]
 
 wizardManager = WizardManager()