Merge branch 'master' into async_picload

4 files changed, 147 insertions, 102 deletions

diff --git a/lib/python/Screens/InputBox.py b/lib/python/Screens/InputBox.py
index 43b8a8b8..61ce356a 100644
--- a/lib/python/Screens/InputBox.py
+++ b/lib/python/Screens/InputBox.py
@@ -8,12 +8,14 @@ from Tools.BoundFunction import boundFunction
 from time import time
 
 class InputBox(Screen):
-	def __init__(self, session, title = "", windowTitle = _("Input"), **kwargs):
+	def __init__(self, session, title = "", windowTitle = _("Input"), useableChars = None, **kwargs):
 		Screen.__init__(self, session)
 
 		self["text"] = Label(title)
 		self["input"] = Input(**kwargs)
 		self.onShown.append(boundFunction(self.setTitle, windowTitle))
+		if useableChars is not None:
+			self["input"].setUseableChars(useableChars)
 
 		self["actions"] = NumberActionMap(["WizardActions", "InputBoxActions", "InputAsciiActions", "KeyboardInputActions"], 
 		{
diff --git a/lib/python/Screens/LocationBox.py b/lib/python/Screens/LocationBox.py
index 7cd6cbf9..68d4f772 100644
--- a/lib/python/Screens/LocationBox.py
+++ b/lib/python/Screens/LocationBox.py
@@ -503,3 +503,32 @@ class MovieLocationBox(LocationBox):
 	def __init__(self, session, text, dir, minFree = None):
 		inhibitDirs = ["/bin", "/boot", "/dev", "/etc", "/lib", "/proc", "/sbin", "/sys", "/usr", "/var"]
 		LocationBox.__init__(self, session, text = text, currDir = dir, bookmarks = config.movielist.videodirs, autoAdd = True, editDir = True, inhibitDirs = inhibitDirs, minFree = minFree)
+
+class TimeshiftLocationBox(LocationBox):
+
+	skinName = "LocationBox" # XXX: though we could use a custom skin or inherit the hardcoded one we stick with the original :-)
+
+	def __init__(self, session):
+		inhibitDirs = ["/bin", "/boot", "/dev", "/etc", "/lib", "/proc", "/sbin", "/sys", "/usr", "/var"]
+		LocationBox.__init__(
+				self,
+				session,
+				text = _("Where to save temporary timeshift recordings?"),
+				currDir = config.usage.timeshift_path.value,
+				bookmarks = config.usage.allowed_timeshift_paths,
+				autoAdd = True,
+				editDir = True,
+				inhibitDirs = inhibitDirs,
+				minFree = 1024 # XXX: the same requirement is hardcoded in servicedvb.cpp
+		)
+
+	def cancel(self):
+		config.usage.timeshift_path.cancel()
+		LocationBox.cancel(self)
+
+	def selectConfirmed(self, ret):
+		if ret:
+			config.usage.timeshift_path.value = self.getPreferredFolder()
+			config.usage.timeshift_path.save()
+			LocationBox.selectConfirmed(self, ret)
+
diff --git a/lib/python/Screens/MovieSelection.py b/lib/python/Screens/MovieSelection.py
index c05f145f..5951653f 100644
--- a/lib/python/Screens/MovieSelection.py
+++ b/lib/python/Screens/MovieSelection.py
@@ -29,6 +29,28 @@ config.movielist.description = ConfigInteger(default=MovieList.HIDE_DESCRIPTION)
 config.movielist.last_videodir = ConfigText(default=resolveFilename(SCOPE_HDD))
 config.movielist.last_timer_videodir = ConfigText(default=resolveFilename(SCOPE_HDD))
 config.movielist.videodirs = ConfigLocations(default=[resolveFilename(SCOPE_HDD)])
+config.movielist.first_tags = ConfigText(default="")
+config.movielist.second_tags = ConfigText(default="")
+
+
+def setPreferredTagEditor(te):
+	global preferredTagEditor
+	try:
+		if preferredTagEditor == None:
+			preferredTagEditor = te
+			print "Preferred tag editor changed to ", preferredTagEditor
+		else:
+			print "Preferred tag editor already set to ", preferredTagEditor
+			print "ignoring ", te
+	except:
+		preferredTagEditor = te
+		print "Preferred tag editor set to ", preferredTagEditor
+
+def getPreferredTagEditor():
+	global preferredTagEditor
+	return preferredTagEditor
+
+setPreferredTagEditor(None)
 
 class MovieContextMenu(Screen):
 	def __init__(self, session, csel, service):
@@ -71,10 +93,8 @@ class MovieContextMenu(Screen):
 
 	def sortBy(self, newType):
 		config.movielist.moviesort.value = newType
-		self.csel.selectedmovie = self.csel.getCurrent()
 		self.csel.setSortType(newType)
 		self.csel.reloadList()
-		self.csel.moveTo()
 		self.close()
 
 	def listType(self, newType):
@@ -123,7 +143,7 @@ class MovieContextMenu(Screen):
 			self.session.openWithCallback(self.close, MessageBox, _("Delete failed!"), MessageBox.TYPE_ERROR)
 		else:
 			self.csel["list"].removeService(self.service)
- 			self.csel["freeDiskSpace"].update()
+			self.csel["freeDiskSpace"].update()
 			self.close()
 
 class SelectionEventInfo:
@@ -149,6 +169,7 @@ class MovieSelection(Screen, HelpableScreen, SelectionEventInfo):
 
 		self.tags = [ ]
 		self.selected_tags = None
+		self.selected_tags_ele = None
 
 		self.movemode = False
 		self.bouquet_mark_edit = False
@@ -178,7 +199,7 @@ class MovieSelection(Screen, HelpableScreen, SelectionEventInfo):
 		# Need list for init
 		SelectionEventInfo.__init__(self)
 
-		self["key_red"] = Button(_("All..."))
+		self["key_red"] = Button(_("All"))
 		self["key_green"] = Button("")
 		self["key_yellow"] = Button("")
 		self["key_blue"] = Button("")
@@ -201,9 +222,9 @@ class MovieSelection(Screen, HelpableScreen, SelectionEventInfo):
 		self["ColorActions"] = HelpableActionMap(self, "ColorActions",
 			{
 				"red": (self.showAll, _("show all")),
-				"green": (self.showTagsFirst, _("show first tag")),
-				"yellow": (self.showTagsSecond, _("show second tag")),
-				"blue": (self.showTagsMenu, _("show tag menu")),
+				"green": (self.showTagsFirst, _("show first selected tag")),
+				"yellow": (self.showTagsSecond, _("show second selected tag")),
+				"blue": (self.showTagsSelect, _("show tag menu")),
 			})
 
 		self["OkCancelActions"] = HelpableActionMap(self, "OkCancelActions",
@@ -246,11 +267,8 @@ class MovieSelection(Screen, HelpableScreen, SelectionEventInfo):
 			self.updateDescription()
 
 	def updateHDDData(self):
- 		self.reloadList()
- 		if self.selectedmovie is not None:
-			self.moveTo()
+ 		self.reloadList(self.selectedmovie)
 		self["waitingtext"].visible = False
-		self.updateTags()
 
 	def moveTo(self):
 		self["list"].moveTo(self.selectedmovie)
@@ -285,26 +303,29 @@ class MovieSelection(Screen, HelpableScreen, SelectionEventInfo):
 	def updateTags(self):
 		# get a list of tags available in this list
 		self.tags = list(self["list"].tags)
-		
-		# by default, we do not display any filtering options
-		self.tag_first = ""
-		self.tag_second = ""
-		
-		# when tags are present, however, the first two are 
-		# directly mapped to the second, third ("green", "yellow") buttons
-		if len(self.tags) > 0:
-			self.tag_first = self.getTagDescription(self.tags[0])
-		
-		if len(self.tags) > 1:
-			self.tag_second = self.getTagDescription(self.tags[1])
-		
+
+		if not self.tags:
+			# by default, we do not display any filtering options
+			self.tag_first = ""
+			self.tag_second = ""
+		else:
+			tmp = config.movielist.first_tags.value
+			if tmp in self.tags:
+				self.tag_first = tmp
+			else:
+				self.tag_first = "<"+_("Tag 1")+">"
+			tmp = config.movielist.second_tags.value
+			if tmp in self.tags:
+				self.tag_second = tmp
+			else:
+				self.tag_second = "<"+_("Tag 2")+">"
 		self["key_green"].text = self.tag_first
 		self["key_yellow"].text = self.tag_second
 		
 		# the rest is presented in a list, available on the
 		# fourth ("blue") button
-		if len(self.tags) > 2:
-			self["key_blue"].text = _("Other...")
+		if self.tags:
+			self["key_blue"].text = _("Tags")+"..."
 		else:
 			self["key_blue"].text = ""
 
@@ -317,20 +338,26 @@ class MovieSelection(Screen, HelpableScreen, SelectionEventInfo):
 	def setSortType(self, type):
 		self["list"].setSortType(type)
 
-	def reloadList(self):
+	def reloadList(self, sel = None, home = False):
 		if not pathExists(config.movielist.last_videodir.value):
 			path = resolveFilename(SCOPE_HDD)
 			config.movielist.last_videodir.value = path
 			config.movielist.last_videodir.save()
 			self.current_ref = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + path)
 			self["freeDiskSpace"].path = path
+		if sel is None:
+			sel = self.getCurrent()
 		self["list"].reload(self.current_ref, self.selected_tags)
 		title = _("Recorded files...")
-		if self.selected_tags is not None:
-			title += " - " + ','.join(self.selected_tags)
 		if config.usage.setup_level.index >= 2: # expert+
 			title += "  " + config.movielist.last_videodir.value
+		if self.selected_tags is not None:
+			title += " - " + ','.join(self.selected_tags)
 		self.setTitle(title)
+ 		if not (sel and self["list"].moveTo(sel)):
+			if home:
+				self["list"].moveToIndex(0)
+		self.updateTags()
 		self["freeDiskSpace"].update()
 
 	def doPathSelect(self):
@@ -348,7 +375,7 @@ class MovieSelection(Screen, HelpableScreen, SelectionEventInfo):
 				config.movielist.last_videodir.save()
 				self.current_ref = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + res)
 				self["freeDiskSpace"].path = res
-				self.reloadList()
+				self.reloadList(home = True)
 			else:
 				self.session.open(
 					MessageBox,
@@ -358,35 +385,41 @@ class MovieSelection(Screen, HelpableScreen, SelectionEventInfo):
 					)
 
 	def showAll(self):
+		self.selected_tags_ele = None
 		self.selected_tags = None
-		self.reloadList()
+		self.reloadList(home = True)
 
-	def showTagsN(self, n):
-		if len(self.tags) < n:
+	def showTagsN(self, tagele):
+		if not self.tags:
 			self.showTagWarning()
+		elif not tagele or self.selected_tags_ele == tagele or not tagele.value in self.tags:
+			self.showTagsMenu(tagele)
 		else:
-			print "select tag #%d, %s, %s" % (n, self.tags[n - 1], ','.join(self.tags))
-			self.selected_tags = set([self.tags[n - 1]])
-			self.reloadList()
+			self.selected_tags_ele = tagele
+			self.selected_tags = set([tagele.value])
+			self.reloadList(home = True)
 
 	def showTagsFirst(self):
-		self.showTagsN(1)
+		self.showTagsN(config.movielist.first_tags)
 
 	def showTagsSecond(self):
-		self.showTagsN(2)
+		self.showTagsN(config.movielist.second_tags)
+
+	def showTagsSelect(self):
+		self.showTagsN(None)
 
 	def tagChosen(self, tag):
 		if tag is not None:
 			self.selected_tags = set([tag[0]])
-			self.reloadList()
+			if self.selected_tags_ele:
+				self.selected_tags_ele.value = tag[0]
+				self.selected_tags_ele.save()
+			self.reloadList(home = True)
 
-	def showTagsMenu(self):
-		if len(self.tags) < 3:
-			self.showTagWarning()
-		else:
-			list = [(tag, self.getTagDescription(tag)) for tag in self.tags ]
-			self.session.openWithCallback(self.tagChosen, ChoiceBox, title=_("Please select keyword to filter..."), list = list)
+	def showTagsMenu(self, tagele):
+		self.selected_tags_ele = tagele
+		list = [(tag, self.getTagDescription(tag)) for tag in self.tags ]
+		self.session.openWithCallback(self.tagChosen, ChoiceBox, title=_("Please select tag to filter..."), list = list)
 
 	def showTagWarning(self):
-		# TODO
-		self.session.open(MessageBox, _("You need to define some keywords first!\nPress the menu-key to define keywords.\nDo you want to define keywords now?"), MessageBox.TYPE_ERROR)
+		self.session.open(MessageBox, _("No tags are set on these movies."), MessageBox.TYPE_ERROR)
diff --git a/lib/python/Screens/TimerEntry.py b/lib/python/Screens/TimerEntry.py
index c4dfff72..1774061d 100644
--- a/lib/python/Screens/TimerEntry.py
+++ b/lib/python/Screens/TimerEntry.py
@@ -8,6 +8,7 @@ from Components.MenuList import MenuList
 from Components.Button import Button
 from Components.Label import Label
 from Components.Pixmap import Pixmap
+from Screens.MovieSelection import getPreferredTagEditor
 from Screens.LocationBox import MovieLocationBox
 from Screens.ChoiceBox import ChoiceBox
 from RecordTimer import AFTEREVENT
@@ -21,8 +22,7 @@ class TimerEntry(Screen, ConfigListScreen):
 		Screen.__init__(self, session)
 		self.timer = timer
 
-		self.entryStartDate = None
-		self.entryEndDate = None
+		self.entryDate = None
 		self.entryService = None
 
 		self["oktext"] = Label(_("OK"))
@@ -88,13 +88,13 @@ class TimerEntry(Screen, ConfigListScreen):
 			self.timerentry_type = ConfigSelection(choices = [("once",_("once")), ("repeated", _("repeated"))], default = type)
 			self.timerentry_name = ConfigText(default = self.timer.name, visible_width = 50, fixed_size = False)
 			self.timerentry_description = ConfigText(default = self.timer.description, visible_width = 50, fixed_size = False)
+			self.timerentry_tags = self.timer.tags + []
+			self.timerentry_tagsset = ConfigSelection(choices = [len(self.timerentry_tags) == 0 and "None" or " ".join(self.timerentry_tags)])
 
 			self.timerentry_repeated = ConfigSelection(default = repeated, choices = [("daily", _("daily")), ("weekly", _("weekly")), ("weekdays", _("Mon-Fri")), ("user", _("user defined"))])
 
-			self.timerentry_startdate = ConfigDateTime(default = self.timer.begin, formatstring = _("%d.%B %Y"), increment = 86400)
+			self.timerentry_date = ConfigDateTime(default = self.timer.begin, formatstring = _("%d.%B %Y"), increment = 86400)
 			self.timerentry_starttime = ConfigClock(default = self.timer.begin)
-
-			self.timerentry_enddate = ConfigDateTime(default = self.timer.end, formatstring =  _("%d.%B %Y"), increment = 86400)
 			self.timerentry_endtime = ConfigClock(default = self.timer.end)
 
 			default = self.timer.dirname or resolveFilename(SCOPE_HDD)
@@ -120,19 +120,6 @@ class TimerEntry(Screen, ConfigListScreen):
 			self.timerentry_service_ref = self.timer.service_ref
 			self.timerentry_service = ConfigSelection([servicename])
 
-			self.timerentry_startdate.addNotifier(self.checkDate)
-			self.timerentry_enddate.addNotifier(self.checkDate)
-
-	def checkDate(self, configElement):
-		if configElement is self.timerentry_startdate:
-			if self.timerentry_enddate.value < self.timerentry_startdate.value:
-				self.timerentry_enddate.value = self.timerentry_startdate.value
-				self["config"].invalidate(self.entryEndDate)
-		if configElement is self.timerentry_enddate:
-			if (self.timerentry_enddate.value < self.timerentry_startdate.value):
-				self.timerentry_startdate.value = self.timerentry_enddate.value
-				self["config"].invalidate(self.entryStartDate)
-
 	def createSetup(self, widget):
 		self.list = []
 		self.list.append(getConfigListEntry(_("Name"), self.timerentry_name))
@@ -165,34 +152,24 @@ class TimerEntry(Screen, ConfigListScreen):
 				self.list.append(getConfigListEntry(_("Saturday"), self.timerentry_day[5]))
 				self.list.append(getConfigListEntry(_("Sunday"), self.timerentry_day[6]))
 
-			#self.list.append(getConfigListEntry("StartDate", self.timerentry_startdate))
-#		self.list.append(getConfigListEntry("Weekday", self.timerentry_weekday))
-
-		self.entryStartDate = getConfigListEntry(_("Start"), self.timerentry_startdate)
-		if self.timerentry_type.value == "once":
-			self.list.append(self.entryStartDate)
-			self.list.append(getConfigListEntry(" ", self.timerentry_starttime))
-		else:
-			self.list.append(getConfigListEntry(_("StartTime"), self.timerentry_starttime))
-
-		self.entryEndDate = getConfigListEntry(_("End"), self.timerentry_enddate)
+		self.entryDate = getConfigListEntry(_("Date"), self.timerentry_date)
 		if self.timerentry_type.value == "once":
-			if self.timerentry_justplay.value != "zap":
-				self.list.append(self.entryEndDate)
-				self.list.append(getConfigListEntry(" ", self.timerentry_endtime))
-		else:
-			if self.timerentry_justplay.value != "zap":
-				self.list.append(getConfigListEntry(_("EndTime"), self.timerentry_endtime))
+			self.list.append(self.entryDate)
+		self.list.append(getConfigListEntry(_("StartTime"), self.timerentry_starttime))
+		if self.timerentry_justplay.value != "zap":
+			self.list.append(getConfigListEntry(_("EndTime"), self.timerentry_endtime))
+		self.channelEntry = getConfigListEntry(_("Channel"), self.timerentry_service)
+		self.list.append(self.channelEntry)
 
+		self.dirname = getConfigListEntry(_("Location"), self.timerentry_dirname)
+		self.tagsSet = getConfigListEntry(_("Tags"), self.timerentry_tagsset)
 		if self.timerentry_justplay.value != "zap":
 			if config.usage.setup_level.index >= 2: # expert+
-				self.dirname = getConfigListEntry(_("Location"), self.timerentry_dirname)
 				self.list.append(self.dirname)
+			if getPreferredTagEditor():
+				self.list.append(self.tagsSet)
 			self.list.append(getConfigListEntry(_("After event"), self.timerentry_afterevent))
 
-		self.channelEntry = getConfigListEntry(_("Channel"), self.timerentry_service)
-		self.list.append(self.channelEntry)
-
 		self[widget].list = self.list
 		self[widget].l.setList(self.list)
 
@@ -206,14 +183,14 @@ class TimerEntry(Screen, ConfigListScreen):
 			self.createSetup("config")
 
 	def keyLeft(self):
-		if self["config"].getCurrent() is self.channelEntry:
+		if self["config"].getCurrent() in [self.channelEntry, self.tagsSet]:
 			self.keySelect()
 		else:
 			ConfigListScreen.keyLeft(self)
 			self.newConfig()
 
 	def keyRight(self):
-		if self["config"].getCurrent() is self.channelEntry:
+		if self["config"].getCurrent() in [self.channelEntry, self.tagsSet]:
 			self.keySelect()
 		else:
 			ConfigListScreen.keyRight(self)
@@ -235,6 +212,12 @@ class TimerEntry(Screen, ConfigListScreen):
 				self.timerentry_dirname.value,
 				minFree = 100 # We require at least 100MB free space
 			)
+		elif getPreferredTagEditor() and cur == self.tagsSet:
+			self.session.openWithCallback(
+				self.tagEditFinished,
+				getPreferredTagEditor(),
+				self.timerentry_tags
+			)
 		else:
 			self.keyGo()
 
@@ -249,24 +232,15 @@ class TimerEntry(Screen, ConfigListScreen):
 		dt = datetime(d.tm_year, d.tm_mon, d.tm_mday, mytime[0], mytime[1])
 		return int(mktime(dt.timetuple()))
 
-	def buildRepeatedBegin(self, rep_time, start_time):
-		d = localtime(rep_time)
-		dt = datetime(d.tm_year, d.tm_mon, d.tm_mday, start_time[0], start_time[1])
-		return int(mktime(dt.timetuple()))
-
 	def getBeginEnd(self):
-		enddate = self.timerentry_enddate.value
+		date = self.timerentry_date.value
 		endtime = self.timerentry_endtime.value
-
-		startdate = self.timerentry_startdate.value
 		starttime = self.timerentry_starttime.value
 
-		begin = self.getTimestamp(startdate, starttime)
-		end = self.getTimestamp(enddate, endtime)
+		begin = self.getTimestamp(date, starttime)
+		end = self.getTimestamp(date, endtime)
 
-		# because of the dateChecks, startdate can't be < enddate.
-		# however, the endtime can be less than the starttime.
-		# in this case, add 1 day.
+		# if the endtime is less than the starttime, add 1 day.
 		if end < begin:
 			end += 86400
 		return begin, end
@@ -278,6 +252,7 @@ class TimerEntry(Screen, ConfigListScreen):
 		self.timer.resetRepeated()
 		self.timer.afterEvent = {"nothing": AFTEREVENT.NONE, "deepstandby": AFTEREVENT.DEEPSTANDBY, "standby": AFTEREVENT.STANDBY}[self.timerentry_afterevent.value]
 		self.timer.service_ref = self.timerentry_service_ref
+		self.timer.tags = self.timerentry_tags
 
 		self.timer.dirname = self.timerentry_dirname.value
 		config.movielist.last_timer_videodir.value = self.timer.dirname
@@ -352,6 +327,12 @@ class TimerEntry(Screen, ConfigListScreen):
 				self.timerentry_dirname.setChoices(config.movielist.videodirs.value, default=res)
 			self.timerentry_dirname.value = res
 
+	def tagEditFinished(self, ret):
+		if ret is not None:
+			self.timerentry_tags = ret
+			self.timerentry_tagsset.setChoices([len(ret) == 0 and "None" or " ".join(ret)])
+			self["config"].invalidate(self.tagsSet)
+
 class TimerLog(Screen):
 	def __init__(self, session, timer):
 		Screen.__init__(self, session)