fix possible crash

[enigma2.git] / lib / python / Screens / MessageBox.py

diff --git a/lib/python/Screens/MessageBox.py b/lib/python/Screens/MessageBox.py
index 6c7bdd3c620f5af8cba0b275e6b95deb77b6c342..86bf07d3b69bff1fc2b18799d521d548e2b35419 100644 (file)
--- a/lib/python/Screens/MessageBox.py
+++ b/lib/python/Screens/MessageBox.py
@@ -1,26 +1,28 @@
 from Screen import Screen
 from Components.ActionMap import ActionMap
 from Components.Label import Label
-from Components.Button import Button
 from Components.Pixmap import Pixmap
+from Components.Sources.StaticText import StaticText
 from Components.MenuList import MenuList
-from enigma import eSize, ePoint, eTimer
+from enigma import eTimer
 
 class MessageBox(Screen):
        TYPE_YESNO = 0
        TYPE_INFO = 1
        TYPE_WARNING = 2
        TYPE_ERROR = 3
-       
-       def __init__(self, session, text, type = TYPE_YESNO, timeout = -1, close_on_any_key = False):
+
+       def __init__(self, session, text, type = TYPE_YESNO, timeout = -1, close_on_any_key = False, default = True, enable_input = True):
                self.type = type
                Screen.__init__(self, session)
-               
-               self["text"] = Label(text)
-               
+
+               self["text"] = Label(text)
+               self["Text"] = StaticText(text)
+               self["selectedChoice"] = StaticText()
+
                self.text = text
                self.close_on_any_key = close_on_any_key
-               
+
                self["ErrorPixmap"] = Pixmap()
                self["QuestionPixmap"] = Pixmap()
                self["InfoPixmap"] = Pixmap()
@@ -34,53 +36,61 @@ class MessageBox(Screen):
                        self["QuestionPixmap"].hide()
                if type != self.TYPE_INFO:
                        self["InfoPixmap"].hide()
-                       
-               if type == self.TYPE_YESNO:
-                       self.list = [ (_("yes"), 0), (_("no"), 1) ]
 
-               self["list"] = MenuList(self.list)
+               if type == self.TYPE_YESNO:
+                       if default == True:
+                               self.list = [ (_("yes"), 0), (_("no"), 1) ]
+                       else:
+                               self.list = [ (_("no"), 1), (_("yes"), 0) ]
                
-               self["actions"] = ActionMap(["MsgBoxActions", "DirectionActions"], 
-                       {
-                               "cancel": self.cancel,
-                               "ok": self.ok,
-                               "alwaysOK": self.alwaysOK,
-                               "up": self.up,
-                               "down": self.down,
-                               "left": self.left,
-                               "right": self.right,
-                               "upRepeated": self.up,
-                               "downRepeated": self.down,
-                               "leftRepeated": self.left,
-                               "rightRepeated": self.right
-                       }, -1)
+               if self.list:
+                       self["selectedChoice"].setText(self.list[0][0])
+               self["list"] = MenuList(self.list)
+
+               if enable_input:
+                       self["actions"] = ActionMap(["MsgBoxActions", "DirectionActions"], 
+                               {
+                                       "cancel": self.cancel,
+                                       "ok": self.ok,
+                                       "alwaysOK": self.alwaysOK,
+                                       "up": self.up,
+                                       "down": self.down,
+                                       "left": self.left,
+                                       "right": self.right,
+                                       "upRepeated": self.up,
+                                       "downRepeated": self.down,
+                                       "leftRepeated": self.left,
+                                       "rightRepeated": self.right
+                               }, -1)
 
        def initTimeout(self, timeout):
                self.timeout = timeout
                if timeout > 0:
                        self.timer = eTimer()
-                       self.timer.timeout.get().append(self.timerTick)
+                       self.timer.callback.append(self.timerTick)
                        self.onExecBegin.append(self.startTimer)
                        self.origTitle = None
-                       try:
-                               if self.instance and self.instance.isVisible():
-                                       self.timerTick()
-                               else:
-                                       self.onShown.append(self.timerTick)
-                       except AttributeError:
-                               self.onShown.append(self.timerTick)
+                       if self.execing:
+                               self.timerTick()
+                       else:
+                               self.onShown.append(self.__onShown)
                        self.timerRunning = True
                else:
                        self.timerRunning = False
 
+       def __onShown(self):
+               self.onShown.remove(self.__onShown)
+               self.timerTick()
+
        def startTimer(self):
                self.timer.start(1000)
 
        def stopTimer(self):
                if self.timerRunning:
                        del self.timer
+                       self.onExecBegin.remove(self.startTimer)
                        self.setTitle(self.origTitle)
-                       self.onShown.remove(self.timerTick)
+                       self.timerRunning = False
 
        def timerTick(self):
                if self.execing:
@@ -96,10 +106,10 @@ class MessageBox(Screen):
        def timeoutCallback(self):
                print "Timeout!"
                self.ok()
-       
+
        def cancel(self):
                self.close(False)
-       
+
        def ok(self):
                if self.type == self.TYPE_YESNO:
                        self.close(self["list"].getCurrent()[1] == 0)
@@ -111,13 +121,13 @@ class MessageBox(Screen):
 
        def up(self):
                self.move(self["list"].instance.moveUp)
-               
+
        def down(self):
                self.move(self["list"].instance.moveDown)
 
        def left(self):
                self.move(self["list"].instance.pageUp)
-               
+
        def right(self):
                self.move(self["list"].instance.pageDown)
 
@@ -125,6 +135,8 @@ class MessageBox(Screen):
                if self.close_on_any_key:
                        self.close(True)
                self["list"].instance.moveSelection(direction)
+               if self.list:
+                       self["selectedChoice"].setText(self["list"].getCurrent()[0])
                self.stopTimer()
 
        def __repr__(self):