X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/52492210cfa975235fb07171f0348e9471edd8ef..e8f4b9242fc67bcea9b561d3e7bc34c25b3ea0a1:/lib/python/Screens/EpgSelection.py

diff --git a/lib/python/Screens/EpgSelection.py b/lib/python/Screens/EpgSelection.py
index 5613c8b1..331fe75e 100644
--- a/lib/python/Screens/EpgSelection.py
+++ b/lib/python/Screens/EpgSelection.py
@@ -6,6 +6,7 @@ from Components.EpgList import *
 from Components.ActionMap import ActionMap
 from Components.ScrollLabel import ScrollLabel
 from Screens.EventView import EventViewSimple
+from TimeDateInput import TimeDateInput
 from enigma import eServiceReference, eServiceEventPtr
 from Screens.FixedMenu import FixedMenu
 from RecordTimer import RecordTimerEntry, parseEvent
@@ -18,8 +19,10 @@ from time import localtime
 import xml.dom.minidom
 
 class EPGSelection(Screen):
-	def __init__(self, session, service, zapFunc=None, eventid=None):
+	def __init__(self, session, service, zapFunc=None, eventid=None, bouquetChangeCB=None):
 		Screen.__init__(self, session)
+		self.bouquetChangeCB = bouquetChangeCB
+		self.ask_time = -1 #now
 		self["key_red"] = Button("")
 		self.closeRecursive = False
 		if isinstance(service, str) and eventid != None:
@@ -67,12 +70,33 @@ class EPGSelection(Screen):
 				"yellow": self.yellowButtonPressed,
 				"blue": self.blueButtonPressed,
 				"info": self.infoKeyPressed,
-				"zapTo": self.zapTo
+				"red": self.zapTo,
+				"input_date_time": self.enterDateTime,
+				"nextBouquet": self.nextBouquet,
+				"prevBouquet": self.prevBouquet
 			})
 		self["actions"].csel = self
 
 		self.onLayoutFinish.append(self.onCreate)
 
+	def nextBouquet(self):
+		if self.bouquetChangeCB:
+			self.bouquetChangeCB(1, self)
+
+	def prevBouquet(self):
+		if self.bouquetChangeCB:
+			self.bouquetChangeCB(-1, self)
+
+	def enterDateTime(self):
+		if self.type == EPG_TYPE_MULTI:
+			self.session.openWithCallback(self.onDateTimeInputClosed, TimeDateInput)
+
+	def onDateTimeInputClosed(self, ret):
+		if len(ret) > 1:
+			if ret[0]:
+				self.ask_time=ret[1]
+				self["list"].fillMultiEPG(self.services, ret[1])
+
 	def closeScreen(self):
 		self.close(self.closeRecursive)
 
@@ -93,12 +117,16 @@ class EPGSelection(Screen):
 	def openSimilarList(self, eventid, refstr):
 		self.session.open(EPGSelection, refstr, None, eventid)
 
+	def setServices(self, services):
+		self.services = services
+		self.onCreate()
+
 	#just used in multipeg
 	def onCreate(self):
 		l = self["list"]
 		if self.type == EPG_TYPE_MULTI:
 			l.recalcEntrySize()
-			l.fillMultiEPG(self.services)
+			l.fillMultiEPG(self.services, self.ask_time)
 		elif self.type == EPG_TYPE_SINGLE:
 			if SINGLE_CPP == 0:
 				l.recalcEntrySize()
@@ -125,16 +153,16 @@ class EPGSelection(Screen):
 				setEvent(cur[0])
 
 	def zapTo(self): # just used in multiepg
-		if self.zapFunc != None:
-			self.closeRecursive = True
-			ref = self["list"].getCurrent()[1]
-			self.zapFunc(ref.ref)
+		if self.zapFunc and self["key_red"].getText() == "Zap":
+			lst = self["list"]
+			count = lst.getCurrentChangeCount()
+			if count == 0:
+				self.closeRecursive = True
+				ref = lst.getCurrent()[1]
+				self.zapFunc(ref.ref)
 
 	def eventSelected(self):
-		if self.type == EPG_TYPE_MULTI:
-			self.zapTo()
-		else:
-			self.infoKeyPressed()
+		self.infoKeyPressed()
 
 	def yellowButtonPressed(self):
 		if self.type == EPG_TYPE_MULTI:
@@ -170,31 +198,47 @@ class EPGSelection(Screen):
 		self["list"].moveDown()
 
 	def applyButtonState(self, state):
-		if state == 1:
-			self["now_button_sel"].show()
+		if state == 0:
 			self["now_button"].hide()
-		else:
-			self["now_button"].show()
 			self["now_button_sel"].hide()
-
-		if state == 2:
-			self["next_button_sel"].show()
 			self["next_button"].hide()
-		else:
-			self["next_button"].show()
 			self["next_button_sel"].hide()
-
-		if state == 3:
-			self["more_button_sel"].show()
 			self["more_button"].hide()
-		else:
-			self["more_button"].show()
 			self["more_button_sel"].hide()
+			self["now_text"].hide()
+			self["next_text"].hide()
+			self["more_text"].hide()
+			self["key_red"].setText("")
+		else:
+			if state == 1:
+				self["key_red"].setText("Zap")
+				self["now_button_sel"].show()
+				self["now_button"].hide()
+			else:
+				self["now_button"].show()
+				self["now_button_sel"].hide()
+				self["key_red"].setText("")
+
+			if state == 2:
+				self["next_button_sel"].show()
+				self["next_button"].hide()
+			else:
+				self["next_button"].show()
+				self["next_button_sel"].hide()
+
+			if state == 3:
+				self["more_button_sel"].show()
+				self["more_button"].hide()
+			else:
+				self["more_button"].show()
+				self["more_button_sel"].hide()
 
 	def onSelectionChanged(self):
 		if self.type == EPG_TYPE_MULTI:
 			count = self["list"].getCurrentChangeCount()
-			if count > 1:
+			if self.ask_time != -1:
+				self.applyButtonState(0)
+			elif count > 1:
 				self.applyButtonState(3)
 			elif count > 0:
 				self.applyButtonState(2)