add ugly buttons to the TimerEntry and change behaviour on pressing OK
[enigma2.git] / lib / python / Screens / TimerEntry.py
1 from Screen import Screen
2 from Components.config import *
3 from Components.ActionMap import NumberActionMap
4 from Components.ConfigList import ConfigList
5 from Components.NimManager import nimmanager
6 from Components.Label import Label
7 import time
8 import datetime
9 from math import log
10
11 class TimerEntry(Screen):
12         def __init__(self, session, timer):
13                 Screen.__init__(self, session)
14                 self.timer = timer;
15                 
16                 self["ok"] = Label("OK")
17                 self["cancel"] = Label("Cancel")
18
19                 self.createConfig()
20
21                 self["actions"] = NumberActionMap(["SetupActions"],
22                 {
23                         "ok": self.keySelect,
24                         "save": self.keyGo,
25                         "cancel": self.keyCancel,
26                         "left": self.keyLeft,
27                         "right": self.keyRight,
28                         "1": self.keyNumberGlobal,
29                         "2": self.keyNumberGlobal,
30                         "3": self.keyNumberGlobal,
31                         "4": self.keyNumberGlobal,
32                         "5": self.keyNumberGlobal,
33                         "6": self.keyNumberGlobal,
34                         "7": self.keyNumberGlobal,
35                         "8": self.keyNumberGlobal,
36                         "9": self.keyNumberGlobal,
37                         "0": self.keyNumberGlobal
38                 }, -1)
39
40                 self.list = []
41                 self["config"] = ConfigList(self.list)
42                 self.createSetup()
43
44         def createConfig(self):
45                         config.timerentry = ConfigSubsection()
46                         
47                         # calculate default values
48                         day = []
49                         weekday = 0
50                         for x in range(0,7):
51                                 day.append(1)
52                         if (self.timer.repeated != 0): # repeated
53                                 type = 1 # repeated
54                                 if (self.timer.repeated == 31): # Mon-Fri
55                                         repeated = 2 # Mon - Fri
56                                 elif (self.timer.repeated == 127): # daily
57                                         repeated = 0 # daily
58                                 else:
59                                         flags = self.timer.repeated
60                                         repeated = 3 # user-defined
61                                         count = 0
62                                         for x in range(0, 7):
63                                                 if (flags == 1): # weekly
64                                                         print "Set to weekday " + str(x)
65                                                         weekday = x
66                                                 if (flags & 1 == 1): # set user-defined flags
67                                                         day[x] = 0
68                                                         count += 1
69                                                 else:
70                                                         day[x] = 1
71                                                 flags = flags >> 1
72                                         if (count == 1):
73                                                 repeated = 1 # weekly
74                         else: # once
75                                 type = 0
76                                 repeated = 0
77                         
78                         config.timerentry.type = configElement_nonSave("config.timerentry.type", configSelection, type, ("once", "repeated"))
79                         config.timerentry.description = configElement_nonSave("config.timerentry.description", configText, self.timer.description, (configText.extendableSize, self.keyRightCallback))
80
81                         config.timerentry.repeated = configElement_nonSave("config.timerentry.repeated", configSelection, repeated, ("daily", "weekly", "Mon-Fri", "user-defined"))
82
83                         config.timerentry.startdate = configElement_nonSave("config.timerentry.startdate", configDateTime, self.timer.begin, ("%d.%B %Y", 86400))
84                         config.timerentry.starttime = configElement_nonSave("config.timerentry.starttime", configSequence, [int(time.strftime("%H", time.localtime(self.timer.begin))), int(time.strftime("%M", time.localtime(self.timer.begin)))], configsequencearg.get("CLOCK"))
85
86                         config.timerentry.enddate = configElement_nonSave("config.timerentry.enddate", configDateTime, self.timer.end, ("%d.%B %Y", 86400))
87                         config.timerentry.endtime = configElement_nonSave("config.timerentry.endtime", configSequence, [int(time.strftime("%H", time.localtime(self.timer.end))), int(time.strftime("%M", time.localtime(self.timer.end)))], configsequencearg.get("CLOCK"))
88
89                         config.timerentry.weekday = configElement_nonSave("config.timerentry.weekday", configSelection, weekday, ("Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"))
90
91                         config.timerentry.day = []
92                         for x in range(0,7):
93                                 config.timerentry.day.append(configElement_nonSave("config.timerentry.day[" + str(x) + "]", configSelection, day[x], ("yes", "no")))
94
95
96                         # FIXME some service-chooser needed here
97                         config.timerentry.service = configElement_nonSave("config.timerentry.service", configSelection, 0, ((str(self.timer.service_ref.getServiceName())),))
98                         
99                         config.timerentry.startdate.addNotifier(self.checkDate)
100                         config.timerentry.enddate.addNotifier(self.checkDate)
101
102         def checkDate(self, configElement):
103                 if (configElement.getConfigPath() == "config.timerentry.startdate"):
104                         if (config.timerentry.enddate.value < config.timerentry.startdate.value):
105                                 config.timerentry.enddate.value = config.timerentry.startdate.value
106                                 config.timerentry.enddate.change()
107                                 self["config"].invalidate(config.timerentry.enddate)
108                 if (configElement.getConfigPath() == "config.timerentry.enddate"):
109                         if (config.timerentry.enddate.value < config.timerentry.startdate.value):
110                                 config.timerentry.startdate.value = config.timerentry.enddate.value
111                                 config.timerentry.startdate.change()
112                                 self["config"].invalidate(config.timerentry.startdate)
113
114         def createSetup(self):
115                 self.list = []
116                 self.list.append(getConfigListEntry("Description", config.timerentry.description))
117                 self.list.append(getConfigListEntry("TimerType", config.timerentry.type))
118
119                 if (config.timerentry.type.value == 0): # once
120                         pass
121                 else: # repeated
122                         self.list.append(getConfigListEntry("Frequency", config.timerentry.repeated))
123                         if (config.timerentry.repeated.value == 0): # daily
124                                 pass
125                         if (config.timerentry.repeated.value == 2): # Mon-Fri
126                                 pass
127                         if (config.timerentry.repeated.value == 1): # weekly
128                                 self.list.append(getConfigListEntry("Weekday", config.timerentry.weekday))
129
130                         if (config.timerentry.repeated.value == 3): # user-defined
131                                 self.list.append(getConfigListEntry("Monday", config.timerentry.day[0]))
132                                 self.list.append(getConfigListEntry("Tuesday", config.timerentry.day[1]))
133                                 self.list.append(getConfigListEntry("Wednesday", config.timerentry.day[2]))
134                                 self.list.append(getConfigListEntry("Thursday", config.timerentry.day[3]))
135                                 self.list.append(getConfigListEntry("Friday", config.timerentry.day[4]))
136                                 self.list.append(getConfigListEntry("Saturday", config.timerentry.day[5]))
137                                 self.list.append(getConfigListEntry("Sunday", config.timerentry.day[6]))
138
139                         #self.list.append(getConfigListEntry("StartDate", config.timerentry.startdate))
140 #               self.list.append(getConfigListEntry("Weekday", config.timerentry.weekday))
141
142                 if (config.timerentry.type.value == 0): # once
143                         self.list.append(getConfigListEntry("StartDate", config.timerentry.startdate))
144                 self.list.append(getConfigListEntry("StartTime", config.timerentry.starttime))
145                 if (config.timerentry.type.value == 0): # once
146                         self.list.append(getConfigListEntry("EndDate", config.timerentry.enddate))
147                 self.list.append(getConfigListEntry("EndTime", config.timerentry.endtime))
148
149                 self.list.append(getConfigListEntry("Channel", config.timerentry.service))
150
151                 self["config"].list = self.list
152                 self["config"].l.setList(self.list)
153
154         def newConfig(self):
155                 print self["config"].getCurrent()
156                 if self["config"].getCurrent()[0] == "TimerType":
157                         self.createSetup()
158                 if self["config"].getCurrent()[0] == "Frequency":
159                         self.createSetup()
160
161         def keyLeft(self):
162                 self["config"].handleKey(config.key["prevElement"])
163                 self.newConfig()
164
165         def keyRightCallback(self, configPath):
166                 currentConfigPath = self["config"].getCurrent()[1].parent.getConfigPath()
167                 # check if we are still on the same config entry
168                 if (currentConfigPath == configPath):
169                         self.keyRight()
170
171         def keyRight(self):
172                 self["config"].handleKey(config.key["nextElement"])
173                 self.newConfig()
174                 
175         def keySelect(self):
176                 pass
177
178         def keyNumberGlobal(self, number):
179                 print "You pressed number " + str(number)
180                 if (self["config"].getCurrent()[1].parent.enabled == True):
181                         self["config"].handleKey(config.key[str(number)])
182
183         def getTimestamp(self, date, mytime):
184                 d = time.localtime(date)
185                 dt = datetime.datetime(d.tm_year, d.tm_mon, d.tm_mday, mytime[0], mytime[1])
186                 print dt
187                 return int(mktime(dt.timetuple()))
188
189         def keyGo(self):
190                 self.timer.resetRepeated()
191                 
192                 if (config.timerentry.type.value == 0): # once
193                         self.timer.begin = self.getTimestamp(config.timerentry.startdate.value, config.timerentry.starttime.value)
194                         self.timer.end = self.getTimestamp(config.timerentry.enddate.value, config.timerentry.endtime.value)
195                 if (config.timerentry.type.value == 1): # repeated
196                         if (config.timerentry.repeated.value == 0): # daily
197                                 for x in range(0,7):
198                                         self.timer.setRepeated(x)
199
200                         if (config.timerentry.repeated.value == 1): # weekly
201                                 self.timer.setRepeated(config.timerentry.weekday.value)
202                                 
203                         if (config.timerentry.repeated.value == 2): # Mon-Fri
204                                 for x in range(0,5):
205                                         self.timer.setRepeated(x)
206                                 
207                         if (config.timerentry.repeated.value == 3): # user-defined
208                                 for x in range(0,7):
209                                         if (config.timerentry.day[x].value == 0): self.timer.setRepeated(x)
210
211                         self.timer.begin = self.getTimestamp(time.time(), config.timerentry.starttime.value)
212                         self.timer.end = self.getTimestamp(time.time(), config.timerentry.endtime.value)                                
213
214                 self.close((True, self.timer))
215
216         def keyCancel(self):
217                 self.close((False,))