aboutsummaryrefslogtreecommitdiff
path: root/RecordTimer.py
blob: f1b33592d616964a313983f8752b85d95c012bd4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
import time
import codecs
#from time import datetime
from Tools import Directories, Notifications

import timer
import xml.dom.minidom

from Screens.MessageBox import MessageBox
import NavigationInstance

from Tools.XMLTools import elementsWithTag
from ServiceReference import ServiceReference

# ok, for descriptions etc we have:
# service reference  (to get the service name)
# name               (title)
# description        (description)
# event data         (ONLY for time adjustments etc.)


# parses an event, and gives out a (begin, end, name, duration, eit)-tuple.
def parseEvent(ev):
	name = ev.getEventName()
	description = ev.getShortDescription()
	begin = ev.getBeginTime()
	end = begin + ev.getDuration()
	eit = None
	return (begin, end, name, description, eit)

class RecordTimerEntry(timer.TimerEntry):
	def __init__(self, serviceref, begin, end, name, description, eit):
		timer.TimerEntry.__init__(self, int(begin), int(end))
		
		assert isinstance(serviceref, ServiceReference)
		
		self.service_ref = serviceref
		
		self.eit = eit
		
		self.dontSave = False
		self.name = name
		self.description = description
		self.timer = None
		self.record_service = None
		self.wantStart = False
		self.prepareOK = False
		
	def calculateFilename(self):
		service_name = self.service_ref.getServiceName()
#		begin_date = datetime.fromtimestamp(begin).strf...
		begin_date = ""
		
		print "begin_date: ", begin_date
		print "service_name: ", service_name
		print "name:", self.name
		print "description: ", self.description

		self.Filename = Directories.getRecordingFilename(service_name)
		#begin_date + " - " + service_name + description)
		
	
	def tryPrepare(self):
		self.calculateFilename()
		self.record_service = NavigationInstance.instance.recordService(self.service_ref)
		if self.record_service == None:
			return False
		else:	
			if self.record_service.prepare(self.Filename + ".ts"):
				self.record_service = None
				return False

			f = open(self.Filename + ".ts.meta", "w")
			f.write(str(self.service_ref) + "\n")
			f.write(self.name + "\n")
			f.write(self.description + "\n")
			f.write(str(self.begin) + "\n")
			del f
			return True

	def activate(self, event):
		if event == self.EventPrepare:
			self.prepareOK = False
			if self.tryPrepare():
				self.prepareOK = True
			else:
				# error.
				Notifications.AddNotificationWithCallback(self.failureCB, MessageBox, _("A timer failed to record!\nDisable TV and try again?\n"))
		elif event == self.EventStart:
			if self.prepareOK:
				self.record_service.start()
				print "timer started!"
			else:
				print "prepare failed, thus start failed, too."
				self.wantStart = True
		elif event == self.EventEnd or event == self.EventAbort:
			self.wantStart = False
			if self.prepareOK:
				self.record_service.stop()
				self.record_service = None
				print "Timer successfully ended"
			else:
				print "prepare failed, thus nothing was recorded."

	def abort():
		# fixme
		pass

	def failureCB(self, answer):
		if answer == True:
			NavigationInstance.instance.stopUserServices()
			self.activate(self.EventPrepare)
			if self.wantStart:
				print "post-activating record"
				self.activate(self.EventStart)
		else:
			print "user killed record"

def createTimer(xml):
	begin = int(xml.getAttribute("begin"))
	end = int(xml.getAttribute("end"))
	serviceref = ServiceReference(str(xml.getAttribute("serviceref")))
	description = xml.getAttribute("description").encode("utf-8")
	repeated = xml.getAttribute("repeated").encode("utf-8")
	eit = xml.getAttribute("eit").encode("utf-8")
	name = xml.getAttribute("name").encode("utf-8")
	#filename = xml.getAttribute("filename").encode("utf-8")
	entry = RecordTimerEntry(serviceref, begin, end, name, description, eit)
	entry.repeated = int(repeated)
	return entry

class RecordTimer(timer.Timer):
	def __init__(self):
		timer.Timer.__init__(self)
		
		self.Filename = Directories.resolveFilename(Directories.SCOPE_CONFIG, "timers.xml")
		
		try:
			self.loadTimer()
		except IOError:
			print "unable to load timers from file!"
			
	def isRecording(self):
		isRunning = False
		for timer in self.timer_list:
			if timer.isRunning():
				isRunning = True
		return isRunning
	
	def loadTimer(self):
		# TODO: PATH!
		doc = xml.dom.minidom.parse(self.Filename)
		
		root = doc.childNodes[0]
		for timer in elementsWithTag(root.childNodes, "timer"):
			self.record(createTimer(timer))
	
	def saveTimer(self):
		doc = xml.dom.minidom.Document()
		root_element = doc.createElement('timers')
		doc.appendChild(root_element)
		root_element.appendChild(doc.createTextNode("\n"))
		
		for timer in self.timer_list + self.processed_timers:
			# some timers (instant records) don't want to be saved.
			# skip them
			if timer.dontSave:
				continue
			t = doc.createTextNode("\t")
			root_element.appendChild(t)
			t = doc.createElement('timer')
			t.setAttribute("begin", str(timer.begin))
			t.setAttribute("end", str(timer.end))
			t.setAttribute("serviceref", str(timer.service_ref))
			t.setAttribute("repeated", str(timer.repeated))			
			t.setAttribute("name", timer.name)
			t.setAttribute("description", timer.description)
			t.setAttribute("eit", str(timer.eit))
			
			root_element.appendChild(t)
			t = doc.createTextNode("\n")
			root_element.appendChild(t)

		file = open(self.Filename, "w")
		doc.writexml(file)
		file.write("\n")
		file.close()
	
	def record(self, entry):
		print "[Timer] Record " + str(entry)
		entry.Timer = self
		self.addTimerEntry(entry)

	def removeEntry(self, entry):
		print "[Timer] Remove " + str(entry)
		
		entry.repeated = False

		if entry.state == timer.TimerEntry.StateRunning:
			print "remove running timer."
			entry.end = time.time()
			self.timeChanged(entry)
		elif entry.state != timer.TimerEntry.StateEnded:
			entry.activate(timer.TimerEntry.EventAbort)
			self.timer_list.remove(entry)

			self.calcNextActivation()
			print "timer did not yet start - removing"

			# the timer was aborted, and removed.
			return
		else:
			print "timer did already end - doing nothing."
		
		print "state: ", entry.state
		print "in processed: ", entry in self.processed_timers
		print "in running: ", entry in self.timer_list
		# now the timer should be in the processed_timers list. remove it from there.
		self.processed_timers.remove(entry)

	def shutdown(self):
		self.saveTimer()