take care of running recording before realy shutdown e2

1 files changed, 76 insertions, 3 deletions

diff --git a/lib/python/Screens/Standby.py b/lib/python/Screens/Standby.py
index 06c44703..cb0a957e 100644
--- a/lib/python/Screens/Standby.py
+++ b/lib/python/Screens/Standby.py
@@ -5,6 +5,7 @@ from Components.AVSwitch import AVSwitch
 from enigma import eDVBVolumecontrol, eDBoxLCD, eServiceReference
 from Components.Sources.Clock import Clock
 
+inStandby = False
 
 class Standby(Screen):
 	def Power(self):
@@ -19,7 +20,7 @@ class Standby(Screen):
 		#set brightness of lcd
 		eDBoxLCD.getInstance().setLCDBrightness(config.lcd.bright.value * 20)
 		#kill me
-		self.close()
+		self.close(True)
 
 	def setMute(self):
 		if (eDVBVolumecontrol.getInstance().isMuted()):
@@ -33,9 +34,8 @@ class Standby(Screen):
 		if self.wasMuted == 0:
 			eDVBVolumecontrol.getInstance().volumeToggleMute()
 
-	def __init__(self, session, infobar):
+	def __init__(self, session):
 		Screen.__init__(self, session)
-		self.infobar = infobar
 		self.avswitch = AVSwitch()
 
 		print "enter standby"
@@ -55,10 +55,21 @@ class Standby(Screen):
 		self.avswitch.setInput("SCART")
 		#set lcd brightness to standby value
 		eDBoxLCD.getInstance().setLCDBrightness(config.lcd.standby.value * 20)
+		self.onShow.append(self.__onShow)
+		self.onHide.append(self.__onHide)
 
 	def createSummary(self):
 		return StandbySummary
 
+	def __onShow(self):
+		global inStandby
+		inStandby = True
+
+	def __onHide(self):
+		global inStandby
+		inStandby = False
+
+
 class StandbySummary(Screen):
 	skin = """
 	<screen position="0,0" size="132,64">
@@ -70,3 +81,65 @@ class StandbySummary(Screen):
 	def __init__(self, session, parent):
 		Screen.__init__(self, session)
 		self["CurrentTime"] = Clock()
+
+from enigma import quitMainloop, iRecordableService
+from Screens.MessageBox import MessageBox
+from time import time
+
+inTryQuitMainloop = False
+
+class TryQuitMainloop(MessageBox):
+	def __init__(self, session, retvalue=1):
+		self.retval=retvalue
+		recordings = len(session.nav.getRecordings())
+		self.connected = False
+		next_rec_time = -1
+		if not recordings:
+			next_rec_time = session.nav.RecordTimer.getNextRecordingTime()
+		if recordings or (next_rec_time > 0 and (next_rec_time - time()) < 360):
+			if retvalue == 1:
+				MessageBox.__init__(self, session, _("Recording(s) are in progress or comming up in few seconds... really shutdown now?"), type = MessageBox.TYPE_YESNO, timeout = -1, close_on_any_key = False)
+			elif retvalue == 2:
+				MessageBox.__init__(self, session, _("Recording(s) are in progress or comming up in few seconds... really reboot now?"), type = MessageBox.TYPE_YESNO, timeout = -1, close_on_any_key = False)
+			elif retvalue == 4:
+				pass
+			else:
+				MessageBox.__init__(self, session, _("Recording(s) are in progress or comming up in few seconds... really restart now?"), type = MessageBox.TYPE_YESNO, timeout = -1, close_on_any_key = False)
+			self.skinName = "MessageBox"
+			session.nav.record_event.append(self.getRecordEvent)
+			self.connected = True
+			self.onShow.append(self.__onShow)
+			self.onHide.append(self.__onHide)
+		else:
+			self.skin = """<screen position="0,0" size="0,0"/>"""
+			Screen.__init__(self, session)
+			self.close(True)
+
+	def getRecordEvent(self, recservice, event):
+		if event == iRecordableService.evEnd:
+			recordings = self.session.nav.getRecordings()
+			if not len(recordings): # no more recordings exist
+				rec_time = self.session.nav.RecordTimer.getNextRecordingTime()
+				if rec_time > 0 and (rec_time - time()) < 360:
+					self.initTimeout(360) # wait for next starting timer
+				else:
+					self.close(True) # immediate shutdown
+		elif event == iRecordableService.evStart:
+			self.stopTimer()
+
+	def close(self, value):
+		if self.connected:
+			self.conntected=False
+			self.session.nav.record_event.remove(self.getRecordEvent)
+		if value:
+			quitMainloop(self.retval)
+		else:
+			MessageBox.close(self, True)
+
+	def __onShow(self):
+		global inTryQuitMainloop
+		inTryQuitMainloop = True
+
+	def __onHide(self):
+		global inTryQuitMainloop
+		inTryQuitMainloop = False