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