start timer after message box is actually visible
[vuplus_dvbapp] / lib / python / Screens / MessageBox.py
1 from Screen import Screen
2 from Components.ActionMap import ActionMap
3 from Components.Label import Label
4 from Components.Button import Button
5 from Components.Pixmap import Pixmap
6 from Components.MenuList import MenuList
7 from enigma import eSize, ePoint, eTimer
8
9 class MessageBox(Screen):
10         TYPE_YESNO = 0
11         TYPE_INFO = 1
12         TYPE_WARNING = 2
13         TYPE_ERROR = 3
14         
15         def __init__(self, session, text, type = TYPE_YESNO, timeout = -1, close_on_any_key = False):
16                 self.type = type
17                 Screen.__init__(self, session)
18                 
19                 self["text"] = Label(text)
20                 
21                 self.text = text
22                 self.close_on_any_key = close_on_any_key
23                 
24                 self["ErrorPixmap"] = Pixmap()
25                 self["QuestionPixmap"] = Pixmap()
26                 self["InfoPixmap"] = Pixmap()
27                 self.timerRunning = False
28                 if timeout > 0:
29                         self.timer = eTimer()
30                         self.timer.timeout.get().append(self.timerTick)
31                         self.onExecBegin.append(self.startTimer)
32                         self.origTitle = None
33                         self.onShown.append(self.timerTick)
34                         self.timerRunning = True
35                 self.timeout = timeout
36                 
37                 self.list = []
38                 if type != self.TYPE_ERROR:
39                         self["ErrorPixmap"].hide()
40                 if type != self.TYPE_YESNO:
41                         self["QuestionPixmap"].hide()
42                 if type != self.TYPE_INFO:
43                         self["InfoPixmap"].hide()
44                         
45                 if type == self.TYPE_YESNO:
46                         self.list = [ (_("yes"), 0), (_("no"), 1) ]
47
48
49                 self["list"] = MenuList(self.list)
50                 
51                 self["actions"] = ActionMap(["MsgBoxActions", "DirectionActions"], 
52                         {
53                                 "cancel": self.cancel,
54                                 "ok": self.ok,
55                                 "alwaysOK": self.alwaysOK,
56                                 "up": self.up,
57                                 "down": self.down,
58                                 "left": self.left,
59                                 "right": self.right,
60                                 "upRepeated": self.up,
61                                 "downRepeated": self.down,
62                                 "leftRepeated": self.left,
63                                 "rightRepeated": self.right
64                         }, -1)
65
66         def startTimer(self):
67                 self.timer.start(1000)
68
69         def timerTick(self):
70                 self.timeout -= 1
71                 if self.origTitle is None:
72                         self.origTitle = self.instance.getTitle()
73                 self.setTitle(self.origTitle + " (" + str(self.timeout) + ")")
74                 if self.timeout == 0:
75                         self.timer.stop()
76                         self.timerRunning = False
77                         self.timeoutCallback()
78                         
79         def timeoutCallback(self):
80                 print "Timeout!"
81                 self.ok()
82         
83         def cancel(self):
84                 self.close(False)
85         
86         def ok(self):
87                 if self.type == self.TYPE_YESNO:
88                         self.close(self["list"].getCurrent()[1] == 0)
89                 else:
90                         self.close(True)
91
92         def alwaysOK(self):
93                 self.close(True)
94
95         def up(self):
96                 self.move(self["list"].instance.moveUp)
97                 
98         def down(self):
99                 self.move(self["list"].instance.moveDown)
100
101         def left(self):
102                 self.move(self["list"].instance.pageUp)
103                 
104         def right(self):
105                 self.move(self["list"].instance.pageDown)
106
107         def move(self, direction):
108                 if self.close_on_any_key:
109                         self.close(True)
110
111                 self["list"].instance.moveSelection(direction)
112                 if self.timerRunning:
113                         self.timer.stop()
114                         self.setTitle(self.origTitle)
115                         self.timerRunning = False
116
117         def __repr__(self):
118                 return str(type(self)) + "(" + self.text + ")"