3aa56800d0329909c1eac1cc6c65d14f2daeefe4
[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 ActionMap, NumberActionMap
6 from Components.ConfigList import ConfigList
7 from Components.MenuList import MenuList
8 from Components.Button import Button
9 from Components.NimManager import nimmanager
10 from Components.Label import Label
11 from Components.Pixmap import Pixmap
12 from Screens.SubserviceSelection import SubserviceSelection
13 from Screens.MessageBox import MessageBox
14 from enigma import eEPGCache
15 import time
16 import datetime
17
18 class TimerEntry(Screen):
19         def __init__(self, session, timer):
20                 Screen.__init__(self, session)
21                 self.timer = timer
22                 
23                 self["oktext"] = Label(_("OK"))
24                 self["canceltext"] = Label(_("Cancel"))
25                 self["ok"] = Pixmap()
26                 self["cancel"] = Pixmap()
27
28                 self.createConfig()
29
30                 self["actions"] = NumberActionMap(["SetupActions", "TextEntryActions"],
31                 {
32                         "ok": self.keySelect,
33                         "save": self.keyGo,
34                         "cancel": self.keyCancel,
35                         "left": self.keyLeft,
36                         "right": self.keyRight,
37                         "delete": self.keyDelete,
38                         "1": self.keyNumberGlobal,
39                         "2": self.keyNumberGlobal,
40                         "3": self.keyNumberGlobal,
41                         "4": self.keyNumberGlobal,
42                         "5": self.keyNumberGlobal,
43                         "6": self.keyNumberGlobal,
44                         "7": self.keyNumberGlobal,
45                         "8": self.keyNumberGlobal,
46                         "9": self.keyNumberGlobal,
47                         "0": self.keyNumberGlobal
48                 }, -1)
49
50                 self.list = []
51                 self["config"] = ConfigList(self.list)
52                 self.createSetup("config")
53
54         def createConfig(self):
55                         config.timerentry = ConfigSubsection()
56                         
57                         # calculate default values
58                         day = []
59                         weekday = 0
60                         for x in range(0,7):
61                                 day.append(1)
62                         if (self.timer.repeated != 0): # repeated
63                                 type = 1 # repeated
64                                 if (self.timer.repeated == 31): # Mon-Fri
65                                         repeated = 2 # Mon - Fri
66                                 elif (self.timer.repeated == 127): # daily
67                                         repeated = 0 # daily
68                                 else:
69                                         flags = self.timer.repeated
70                                         repeated = 3 # user defined
71                                         count = 0
72                                         for x in range(0, 7):
73                                                 if (flags == 1): # weekly
74                                                         print "Set to weekday " + str(x)
75                                                         weekday = x
76                                                 if (flags & 1 == 1): # set user defined flags
77                                                         day[x] = 0
78                                                         count += 1
79                                                 else:
80                                                         day[x] = 1
81                                                 flags = flags >> 1
82                                         if (count == 1):
83                                                 repeated = 1 # weekly
84                         else: # once
85                                 type = 0
86                                 repeated = 0
87                                 weekday = (int(strftime("%w", time.localtime(self.timer.begin))) - 1) % 7
88                                 day[weekday] = 0
89                         
90                         config.timerentry.type = configElement_nonSave("config.timerentry.type", configSelection, type, (_("once"), _("repeated")))
91                         config.timerentry.name = configElement_nonSave("config.timerentry.name", configText, self.timer.name, (configText.extendableSize, self.keyRightCallback))
92                         config.timerentry.description = configElement_nonSave("config.timerentry.description", configText, self.timer.description, (configText.extendableSize, self.keyRightCallback))
93
94                         config.timerentry.repeated = configElement_nonSave("config.timerentry.repeated", configSelection, repeated, (_("daily"), _("weekly"), _("Mon-Fri"), _("user defined")))
95
96                         config.timerentry.startdate = configElement_nonSave("config.timerentry.startdate", configDateTime, self.timer.begin, (_("%d.%B %Y"), 86400))
97                         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"))
98
99                         config.timerentry.enddate = configElement_nonSave("config.timerentry.enddate", configDateTime, self.timer.end, (_("%d.%B %Y"), 86400))
100                         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"))
101
102                         config.timerentry.weekday = configElement_nonSave("config.timerentry.weekday", configSelection, weekday, (_("Monday"), _("Tuesday"), _("Wednesday"), _("Thursday"), _("Friday"), _("Saturday"), _("Sunday")))
103
104                         config.timerentry.day = []
105                         for x in range(0,7):
106                                 config.timerentry.day.append(configElement_nonSave("config.timerentry.day[" + str(x) + "]", configSelection, day[x], (_("yes"), _("no"))))
107
108
109                         # FIXME some service-chooser needed here
110                         servicename = "N/A"
111                         try: # no current service available?
112                                 servicename = str(self.timer.service_ref.getServiceName())
113                         except:
114                                 pass
115                         config.timerentry.service = configElement_nonSave("config.timerentry.service", configSelection, 0, ((servicename),))
116                         
117                         config.timerentry.startdate.addNotifier(self.checkDate)
118                         config.timerentry.enddate.addNotifier(self.checkDate)
119
120         def checkDate(self, configElement):
121                 if (configElement.getConfigPath() == "config.timerentry.startdate"):
122                         if (config.timerentry.enddate.value < config.timerentry.startdate.value):
123                                 config.timerentry.enddate.value = config.timerentry.startdate.value
124                                 config.timerentry.enddate.change()
125                                 try:
126                                         self["config"].invalidate(config.timerentry.enddate)
127                                 except:
128                                         pass
129                 if (configElement.getConfigPath() == "config.timerentry.enddate"):
130                         if (config.timerentry.enddate.value < config.timerentry.startdate.value):
131                                 config.timerentry.startdate.value = config.timerentry.enddate.value
132                                 config.timerentry.startdate.change()
133                                 try:
134                                         self["config"].invalidate(config.timerentry.startdate)
135                                 except:
136                                         pass
137
138         def createSetup(self, widget):
139                 self.list = []
140                 self.list.append(getConfigListEntry(_("Name"), config.timerentry.name))
141                 self.list.append(getConfigListEntry(_("Description"), config.timerentry.description))
142                 self.timerTypeEntry = getConfigListEntry(_("Timer Type"), config.timerentry.type)
143                 self.list.append(self.timerTypeEntry)
144
145                 if (config.timerentry.type.value == 0): # once
146                         self.frequencyEntry = None
147                 else: # repeated
148                         self.frequencyEntry = getConfigListEntry(_("Frequency"), config.timerentry.repeated)
149                         self.list.append(self.frequencyEntry)
150                         if (config.timerentry.repeated.value == 0): # daily
151                                 pass
152                         if (config.timerentry.repeated.value == 2): # Mon-Fri
153                                 pass
154                         if (config.timerentry.repeated.value == 1): # weekly
155                                 self.list.append(getConfigListEntry(_("Weekday"), config.timerentry.weekday))
156
157                         if (config.timerentry.repeated.value == 3): # user defined
158                                 self.list.append(getConfigListEntry(_("Monday"), config.timerentry.day[0]))
159                                 self.list.append(getConfigListEntry(_("Tuesday"), config.timerentry.day[1]))
160                                 self.list.append(getConfigListEntry(_("Wednesday"), config.timerentry.day[2]))
161                                 self.list.append(getConfigListEntry(_("Thursday"), config.timerentry.day[3]))
162                                 self.list.append(getConfigListEntry(_("Friday"), config.timerentry.day[4]))
163                                 self.list.append(getConfigListEntry(_("Saturday"), config.timerentry.day[5]))
164                                 self.list.append(getConfigListEntry(_("Sunday"), config.timerentry.day[6]))
165
166                         #self.list.append(getConfigListEntry("StartDate", config.timerentry.startdate))
167 #               self.list.append(getConfigListEntry("Weekday", config.timerentry.weekday))
168
169                 if (config.timerentry.type.value == 0): # once
170                         self.list.append(getConfigListEntry(_("Start"), config.timerentry.startdate))
171                         self.list.append(getConfigListEntry(" ", config.timerentry.starttime))
172                 else:
173                         self.list.append(getConfigListEntry(_("StartTime"), config.timerentry.starttime))
174                 if (config.timerentry.type.value == 0): # once
175                         self.list.append(getConfigListEntry(_("End"), config.timerentry.enddate))
176                         self.list.append(getConfigListEntry(" ", config.timerentry.endtime))
177                 else:
178                         self.list.append(getConfigListEntry(_("EndTime"), config.timerentry.endtime))
179
180                 self.channelEntry = getConfigListEntry(_("Channel"), config.timerentry.service)
181                 self.list.append(self.channelEntry)
182
183                 self[widget].list = self.list
184                 self[widget].l.setList(self.list)
185
186         def newConfig(self):
187                 print "newConfig", self["config"].getCurrent()
188                 if self["config"].getCurrent() == self.timerTypeEntry:
189                         self.createSetup("config")
190                 if self["config"].getCurrent() == self.frequencyEntry:
191                         self.createSetup("config")
192
193         def keyLeft(self):
194                 if self["config"].getCurrent() == self.channelEntry:
195                         self.keySelect()
196                 else:
197                         self["config"].handleKey(config.key["prevElement"])
198                         self.newConfig()
199
200         def keyDelete(self):
201                 self["config"].handleKey(config.key["delete"])
202                         
203         def keyRightCallback(self, configPath):
204                 currentConfigPath = self["config"].getCurrent()[1].parent.getConfigPath()
205                 # check if we are still on the same config entry
206                 if (currentConfigPath == configPath):
207                         self.keyRight()
208
209         def keyRight(self):
210                 if self["config"].getCurrent() == self.channelEntry:
211                         self.keySelect()
212                 else:
213                         self["config"].handleKey(config.key["nextElement"])
214                         self.newConfig()
215                 
216         def keySelect(self):
217                 if self["config"].getCurrent() == self.channelEntry:
218                         self.session.openWithCallback(self.finishedChannelSelection, ChannelSelection.SimpleChannelSelection, _("Select channel to record from"))
219                 else:
220                         self.keyGo()
221
222         def finishedChannelSelection(self, args):
223                 oldref = self.timer.service_ref
224                 try:
225                         self.timer.service_ref = ServiceReference(args)
226                         config.timerentry.service.vals = (str(self.timer.service_ref.getServiceName()),)
227                         self["config"].invalidate(config.timerentry.service)
228                 except:
229                         print "you pressed cancel"
230                         self.timer.service_ref = oldref
231
232         def keyNumberGlobal(self, number):
233                 print "You pressed number " + str(number)
234                 if (self["config"].getCurrent()[1].parent.enabled == True):
235                         self["config"].handleKey(config.key[str(number)])
236
237         def getTimestamp(self, date, mytime):
238                 d = time.localtime(date)
239                 dt = datetime.datetime(d.tm_year, d.tm_mon, d.tm_mday, mytime[0], mytime[1])
240                 return int(mktime(dt.timetuple()))
241
242         def keyGo(self):
243                 self.timer.name = config.timerentry.name.value
244                 self.timer.description = config.timerentry.description.value
245                 self.timer.resetRepeated()
246                 
247                 if (config.timerentry.type.value == 0): # once
248                         self.timer.begin = self.getTimestamp(config.timerentry.startdate.value, config.timerentry.starttime.value)
249                         self.timer.end = self.getTimestamp(config.timerentry.enddate.value, config.timerentry.endtime.value)
250                 if (config.timerentry.type.value == 1): # repeated
251                         if (config.timerentry.repeated.value == 0): # daily
252                                 for x in range(0,7):
253                                         self.timer.setRepeated(x)
254
255                         if (config.timerentry.repeated.value == 1): # weekly
256                                 self.timer.setRepeated(config.timerentry.weekday.value)
257                                 
258                         if (config.timerentry.repeated.value == 2): # Mon-Fri
259                                 for x in range(0,5):
260                                         self.timer.setRepeated(x)
261                                 
262                         if (config.timerentry.repeated.value == 3): # user defined
263                                 for x in range(0,7):
264                                         if (config.timerentry.day[x].value == 0): self.timer.setRepeated(x)
265
266                         self.timer.begin = self.getTimestamp(time.time(), config.timerentry.starttime.value)
267                         self.timer.end = self.getTimestamp(time.time(), config.timerentry.endtime.value)
268
269                 if self.timer.eit is not None:
270                         event = eEPGCache.getInstance().lookupEventId(self.timer.service_ref.ref, self.timer.eit)
271                         if event is not None:
272                                 if event.getNumOfLinkageServices() > 0:
273                                         self.session.openWithCallback(self.subserviceSelected, SubserviceSelection, event, self.timer.service_ref.ref)
274                                         return
275                 self.close((True, self.timer))
276
277         def subserviceSelected(self, service):
278                 if not service is None:
279                         self.timer.service_ref = ServiceReference(service)
280                 self.close((True, self.timer))
281
282         def keyCancel(self):
283                 self.close((False,))
284                 
285 class TimerLog(Screen):
286         def __init__(self, session, timer):
287                 Screen.__init__(self, session)
288                 self.timer = timer;
289                 self.log_entries = self.timer.log_entries[:]
290                 
291                 self.fillLogList()
292                 
293                 self["loglist"] = MenuList(self.list)
294                 self["logentry"] = Label()
295                 
296                 self["key_red"] = Button(_("Delete entry"))
297                 self["key_green"] = Button()
298                 self["key_yellow"] = Button("")
299                 self["key_blue"] = Button(_("Clear log"))
300                 
301                 self.onShown.append(self.updateText)
302
303                 self["actions"] = NumberActionMap(["OkCancelActions", "DirectionActions", "ColorActions"],
304                 {
305                         "ok": self.keyClose,
306                         "cancel": self.keyClose,
307                         "up": self.up,
308                         "down": self.down,
309                         "left": self.left,
310                         "right": self.right,
311                         "red": self.deleteEntry,
312                         "blue": self.clearLog
313                 }, -1)
314
315         def deleteEntry(self):
316                 self.log_entries.remove(self["loglist"].getCurrent()[1])
317                 self.fillLogList()
318                 self["loglist"].l.setList(self.list)
319                 self.updateText()
320
321         def fillLogList(self):
322                 self.list = [ ]
323                 for x in self.log_entries:
324                         self.list.append((str(strftime("%Y-%m-%d %H-%M", localtime(x[0])) + " - " + x[2]), x))
325         
326         def clearLog(self):             
327                 self.log_entries = []
328                 self.fillLogList()
329                 self["loglist"].l.setList(self.list)
330                 self.updateText()
331                 
332         def keyClose(self):
333                 if self.timer.log_entries != self.log_entries:
334                         self.timer.log_entries = self.log_entries
335                         self.close((True, self.timer))
336                 else:
337                         self.close((False,))
338                 
339         def up(self):
340                 self["loglist"].instance.moveSelection(self["loglist"].instance.moveUp)
341                 self.updateText()
342                 
343         def down(self):
344                 self["loglist"].instance.moveSelection(self["loglist"].instance.moveDown)
345                 self.updateText()
346
347         def left(self):
348                 self["loglist"].instance.moveSelection(self["loglist"].instance.pageUp)
349                 self.updateText()
350                 
351         def right(self):
352                 self["loglist"].instance.moveSelection(self["loglist"].instance.pageDown)
353                 self.updateText()
354
355         def updateText(self):
356                 if len(self.list) > 0:
357                         self["logentry"].setText(str(self["loglist"].getCurrent()[1][2]))
358                 else:
359                         self["logentry"].setText("")