X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/98de70a04df6f603ce86e2eaf40b1e532def6d7c..a6f7c725e21a4b6ed064e92b453331837d12f267:/lib/python/Screens/SubservicesQuickzap.py

diff --git a/lib/python/Screens/SubservicesQuickzap.py b/lib/python/Screens/SubservicesQuickzap.py
index 84c8317c..3bcc3c46 100644
--- a/lib/python/Screens/SubservicesQuickzap.py
+++ b/lib/python/Screens/SubservicesQuickzap.py
@@ -9,10 +9,17 @@ from Components.ServiceEventTracker import InfoBarBase
 
 from enigma import eTimer
 
-class SubservicesQuickzap(InfoBarBase, InfoBarShowHide, InfoBarMenu, InfoBarInstantRecord, InfoBarSeek, InfoBarTimeshift, InfoBarTimeshiftState, InfoBarExtensions, InfoBarSubtitleSupport, InfoBarAudioSelection, Screen):
+class SubservicesQuickzap(InfoBarBase, InfoBarShowHide, InfoBarMenu, \
+		InfoBarInstantRecord, InfoBarSeek, InfoBarTimeshift, \
+		InfoBarTimeshiftState, InfoBarExtensions, InfoBarSubtitleSupport, \
+		InfoBarAudioSelection, Screen):
+
 	def __init__(self, session, subservices):
 		Screen.__init__(self, session)
-		for x in [InfoBarBase, InfoBarShowHide, InfoBarMenu, InfoBarInstantRecord, InfoBarSeek, InfoBarTimeshift, InfoBarTimeshiftState, InfoBarSubtitleSupport, InfoBarExtensions, InfoBarAudioSelection]:
+		for x in InfoBarBase, InfoBarShowHide, InfoBarMenu, \
+				InfoBarInstantRecord, InfoBarSeek, InfoBarTimeshift, \
+				InfoBarTimeshiftState, InfoBarSubtitleSupport, \
+				InfoBarExtensions, InfoBarAudioSelection:
 			x.__init__(self)
 
 		self.restoreService = self.session.nav.getCurrentlyPlayingServiceReference()
@@ -46,6 +53,11 @@ class SubservicesQuickzap(InfoBarBase, InfoBarShowHide, InfoBarMenu, InfoBarInst
 				"9": self.keyNumberGlobal,
 				"0": self.keyNumberGlobal
 			}, 0)
+		
+		self.onClose.append(self.__onClose)
+
+	def __onClose(self):
+		self.session.nav.playService(self.restoreService, False)
 
 	def onLayoutFinished(self):
 		self.timer.start(0,True)
@@ -92,12 +104,15 @@ class SubservicesQuickzap(InfoBarBase, InfoBarShowHide, InfoBarMenu, InfoBarInst
 	def showSelection(self):
 		self.updateSubservices()
 		tlist = []
-		if self.n is not None:
-			for x in range(self.n):
-				i = self.subservices.getSubservice(x)
-				tlist.append((i.getName(), x))
+		n = self.n or 0
+		if n:
+			idx = 0
+			while idx < n:
+				i = self.subservices.getSubservice(idx)
+				tlist.append((i.getName(), idx))
+				idx += 1
 
-		keys = [ "", "1", "2", "3", "4", "5", "6", "7", "8", "9" ] + [""] * self.n
+		keys = [ "", "1", "2", "3", "4", "5", "6", "7", "8", "9" ] + [""] * n
 		self.session.openWithCallback(self.subserviceSelected, ChoiceBox, title=_("Please select a subservice..."), list = tlist, selection = self.currentlyPlayingSubservice, keys = keys)
 	
 	def subserviceSelected(self, service):
@@ -113,7 +128,6 @@ class SubservicesQuickzap(InfoBarBase, InfoBarShowHide, InfoBarMenu, InfoBarInst
 	
 	def quit(self, answer):
 		if answer:
-			self.session.nav.playService(self.restoreService, False)
 			self.close()
 		
 	def playSubservice(self, number = 0):