wizard: cleanup circular reference on close
[enigma2.git] / lib / python / Screens / Wizard.py
index 56ab56f41101749fa48449c7bedc8c28bc663c6b..fd9e9c3c44ab1fd042a13578de8986f5b51580db 100644 (file)
 from Screen import Screen
 
+import string
+
 from Screens.HelpMenu import HelpableScreen
 from Components.Label import Label
-from Components.ActionMap import HelpableActionMap
+from Components.Slider import Slider
+from Components.ActionMap import HelpableActionMap, NumberActionMap
 from Components.config import config, configElementBoolean
+from Components.Pixmap import *
+from Components.MenuList import MenuList
+from Components.ConfigList import ConfigList
 
-config.misc.firstrun = configElementBoolean("config.misc.firstrun", 1);
+from xml.sax import make_parser
+from xml.sax.handler import ContentHandler
 
-class WelcomeWizard(Screen, HelpableScreen):
+config.misc.firstrun = configElementBoolean("config.misc.firstrun", 1);
 
-       skin = """
-               <screen position="140,125" size="460,350" title="Welcome...">
-                       <widget name="text" position="20,20" size="440,300" font="Arial;30" />
-               </screen>"""
+class Wizard(Screen, HelpableScreen):
 
+       class parseWizard(ContentHandler):
+               def __init__(self, wizard):
+                       self.isPointsElement, self.isReboundsElement = 0, 0
+                       self.wizard = wizard
+                       self.currContent = ""
+               
+               def startElement(self, name, attrs):
+                       print name
+                       self.currContent = name
+                       if (name == "step"):
+                               self.lastStep = int(attrs.get('number'))
+                               self.wizard[self.lastStep] = {"text": "", "list": [], "config": {"screen": None, "args": None, "type": "" }, "code": ""}
+                       elif (name == "text"):
+                               self.wizard[self.lastStep]["text"] = string.replace(str(attrs.get('value')), "\\n", "\n")
+                       elif (name == "listentry"):
+                               self.wizard[self.lastStep]["list"].append((str(attrs.get('caption')), str(attrs.get('step'))))
+                       elif (name == "config"):
+                               exec "from Screens." + str(attrs.get('module')) + " import *"
+                               self.wizard[self.lastStep]["config"]["screen"] = eval(str(attrs.get('screen')))
+                               if (attrs.has_key('args')):
+                                       print "has args"
+                                       self.wizard[self.lastStep]["config"]["args"] = str(attrs.get('args'))
+                               self.wizard[self.lastStep]["config"]["type"] = str(attrs.get('type'))
+               def endElement(self, name):
+                       self.currContent = ""
+                       if name == 'code':
+                               self.wizard[self.lastStep]["code"] = self.wizard[self.lastStep]["code"].strip()
+                               
+               def characters(self, ch):
+                       if self.currContent == "code":
+                                self.wizard[self.lastStep]["code"] = self.wizard[self.lastStep]["code"] + ch
+                               
        def __init__(self, session):
-               self.skin = WelcomeWizard.skin
-
                Screen.__init__(self, session)
                HelpableScreen.__init__(self)
 
+               self.wizard = {}
+               parser = make_parser()
+               print "Reading " + self.xmlfile
+               wizardHandler = self.parseWizard(self.wizard)
+               parser.setContentHandler(wizardHandler)
+               parser.parse('/usr/share/enigma2/' + self.xmlfile)
+               
+               self.numSteps = len(self.wizard)
+               self.currStep = 1
+
+               self["text"] = Label()
 
-               self["text"] = Label(_("Welcome!\n\nYou can always press the help key!\n\nPlease Note: Make a service search first!"));
+               self["config"] = ConfigList([])
+
+               self["step"] = Label()
+                               
+               self["stepslider"] = Slider(1, self.numSteps)
+               
+               self.list = []
+               self["list"] = MenuList(self.list)
+
+               self.onShown.append(self.updateValues)
+               self.onClose.append(self.delReferences)
                
-               self["actions"] = HelpableActionMap(self, "OkCancelActions",
-                       {
-                               "ok": (self.ok, _("Close this Screen...")),
-                       })
+               self["actions"] = NumberActionMap(["WizardActions", "NumberActions"],
+               {
+                       "ok": self.ok,
+                       "back": self.back,
+                       "left": self.left,
+                       "right": self.right,
+                       "up": self.up,
+                       "down": self.down,
+                       "1": self.keyNumberGlobal,
+                       "2": self.keyNumberGlobal,
+                       "3": self.keyNumberGlobal,
+                       "4": self.keyNumberGlobal,
+                       "5": self.keyNumberGlobal,
+                       "6": self.keyNumberGlobal,
+                       "7": self.keyNumberGlobal,
+                       "8": self.keyNumberGlobal,
+                       "9": self.keyNumberGlobal,
+                       "0": self.keyNumberGlobal
+               }, -1)
+               
+               #self["actions"] = HelpableActionMap(self, "OkCancelActions",
+                       #{
+                               #"ok": (self.ok, _("Close this Screen...")),
+                       #})
 
+       def back(self):
+               self.currStep -= 1
+               if self.currStep < 1:
+                       self.currStep = 1
+               self.updateValues()
+               
        def ok(self):
-               config.misc.firstrun.value = 0;
-               config.misc.firstrun.save()
-               self.session.close()
+               print "OK"
+               if (self.wizard[self.currStep]["config"]["screen"] != None):
+                       try: # don't die, if no run() is available
+                               self.configInstance.run()
+                       except:
+                               print "Failed to run configInstance"
+               
+               if (len(self.wizard[self.currStep]["list"]) > 0):
+                       nextStep = self.wizard[self.currStep]["list"][self["list"].l.getCurrentSelectionIndex()][1]
+                       if nextStep == "end":
+                               self.currStep = self.numSteps
+                       elif nextStep == "next":
+                               pass
+                       else:
+                               self.currStep = int(nextStep) - 1
 
-def listActiveWizards():
-       wizards = [ ]
+               if (self.currStep == self.numSteps): # wizard finished
+                       config.misc.firstrun.value = 0;
+                       config.misc.firstrun.save()
+                       self.session.close()
+               else:
+                       self.currStep += 1
+                       self.updateValues()
+                       
+               print "Now: " + str(self.currStep)
 
-       if config.misc.firstrun.value:
-               wizards.append(WelcomeWizard)
+       def keyNumberGlobal(self, number):
+               if (self.wizard[self.currStep]["config"]["screen"] != None):
+                       self.configInstance.keyNumberGlobal(number)
+               
+       def left(self):
+               if (self.wizard[self.currStep]["config"]["screen"] != None):
+                       self.configInstance.keyLeft()
+               print "left"
        
-       return wizards
+       def right(self):
+               if (self.wizard[self.currStep]["config"]["screen"] != None):
+                       self.configInstance.keyRight()
+               print "right"
+
+       def up(self):
+               if (self.wizard[self.currStep]["config"]["screen"] != None):
+                       self["config"].instance.moveSelection(self["config"].instance.moveUp)
+               elif (len(self.wizard[self.currStep]["list"]) > 0):
+                       self["list"].instance.moveSelection(self["list"].instance.moveUp)
+               print "up"
+               
+       def down(self):
+               if (self.wizard[self.currStep]["config"]["screen"] != None):
+                       self["config"].instance.moveSelection(self["config"].instance.moveDown)
+               elif (len(self.wizard[self.currStep]["list"]) > 0):
+                       self["list"].instance.moveSelection(self["list"].instance.moveDown)
+               print "down"
+               
+       def updateValues(self):
+               print "Updating values in step " + str(self.currStep)
+               self["step"].setText(_("Step ") + str(self.currStep) + "/" + str(self.numSteps))
+               self["stepslider"].setValue(self.currStep)
+
+               print _(self.wizard[self.currStep]["text"])
+               self["text"].setText(_(self.wizard[self.currStep]["text"]))
+
+               if self.wizard[self.currStep]["code"] != "":
+                       print self.wizard[self.currStep]["code"]
+                       exec(self.wizard[self.currStep]["code"])
+               
+               self["list"].instance.setZPosition(1)
+               self.list = []
+               if (len(self.wizard[self.currStep]["list"]) > 0):
+                       self["list"].instance.setZPosition(2)
+                       for x in self.wizard[self.currStep]["list"]:
+                               self.list.append((_(x[0]), None))
+               self["list"].l.setList(self.list)
+
+               self["config"].instance.setZPosition(1)
+               if (self.wizard[self.currStep]["config"]["screen"] != None):
+                       if self.wizard[self.currStep]["config"]["type"] == "standalone":
+                               print "Type is standalone"
+                               self.session.openWithCallback(self.ok, self.wizard[self.currStep]["config"]["screen"])
+                       else:
+                               self["config"].instance.setZPosition(2)
+                               print self.wizard[self.currStep]["config"]["screen"]
+                               if self.wizard[self.currStep]["config"]["args"] == None:
+                                       self.configInstance = self.session.instantiateDialog(self.wizard[self.currStep]["config"]["screen"])
+                               else:
+                                       self.configInstance = self.session.instantiateDialog(self.wizard[self.currStep]["config"]["screen"], eval(self.wizard[self.currStep]["config"]["args"]))
+                               self["config"].l.setList(self.configInstance["config"].list)
+                               self.configInstance["config"] = self["config"]
+               else:
+                       self["config"].l.setList([])
+
+       def delReferences(self):
+               del self.configInstance
+
+class WizardManager:
+       def __init__(self):
+               self.wizards = []
+       
+       def registerWizard(self, wizard):
+               self.wizards.append(wizard)
+       
+       def getWizards(self):
+               if config.misc.firstrun.value:
+                       return self.wizards
+               else:
+                       return []
+
+wizardManager = WizardManager()