add parental control (still somehow buggy and some minor features missing... don't trust it

yet to protect your children)

1 files changed, 264 insertions, 0 deletions

diff --git a/lib/python/Screens/ParentalControlSetup.py b/lib/python/Screens/ParentalControlSetup.py
new file mode 100644
index 00000000..2a907466
--- /dev/null
+++ b/lib/python/Screens/ParentalControlSetup.py
@@ -0,0 +1,264 @@
+from Screen import Screen
+from Components.ConfigList import ConfigList, ConfigListScreen
+from Components.ActionMap import NumberActionMap
+from Components.config import config, getConfigListEntry, ConfigNothing, NoSave, ConfigPIN
+from Components.ServiceList import ServiceList
+from Components.ParentalControlList import ParentalControlEntryComponent, ParentalControlList 
+from Components.ParentalControl import parentalControl
+from Screens.ChoiceBox import ChoiceBox
+from Screens.MessageBox import MessageBox
+from Screens.InputBox import InputBox, Input, PinInput
+from Tools.Directories import resolveFilename, SCOPE_CONFIG
+from Tools.BoundFunction import boundFunction
+from ServiceReference import ServiceReference
+from enigma import eServiceCenter, eServiceReference
+import os
+import operator
+
+class ProtectedScreen:
+	def __init__(self):
+		if self.isProtected():
+			self.onFirstExecBegin.append(boundFunction(self.session.openWithCallback, self.pinEntered, PinInput, pinList = [self.protectedWithPin()], title = self.getPinText(), windowTitle = _("Change pin code")))
+
+	def getPinText(self):
+		return _("Please enter the correct pin code")
+	
+	def isProtected(self):
+		return True
+	
+	def protectedWithPin(self):
+		return config.ParentalControl.setuppin.value
+	
+	def pinEntered(self, result):
+		if result[0] is None:
+			self.close()
+		if not result[0]:
+			print result, "-", self.protectedWithPin()
+			self.session.openWithCallback(self.close, MessageBox, _("The pin code you entered is wrong."), MessageBox.TYPE_ERROR)
+
+class ParentalControlSetup(Screen, ConfigListScreen, ProtectedScreen):
+	def __init__(self, session):
+		Screen.__init__(self, session)
+		
+		ProtectedScreen.__init__(self)
+		
+		self["actions"] = NumberActionMap(["SetupActions"],
+		{
+			"cancel": self.keyCancel
+		}, -2)
+		
+		self.list = []
+		ConfigListScreen.__init__(self, self.list)
+		self.createSetup()
+	
+	def isProtected(self):
+		return config.ParentalControl.setuppinactive.value
+	
+	def createSetup(self):
+		self.editListEntry = None
+		self.changePin = None
+		self.changeSetupPin = None
+		
+		self.list = []
+		self.list.append(getConfigListEntry(_("Enable parental control"), config.ParentalControl.configured))
+		print "config.ParentalControl.configured.value", config.ParentalControl.configured.value
+		if config.ParentalControl.configured.value:
+			#self.list.append(getConfigListEntry(_("Configuration mode"), config.ParentalControl.mode))
+			self.list.append(getConfigListEntry(_("Protect setup"), config.ParentalControl.setuppinactive))
+			if config.ParentalControl.setuppinactive.value:
+				self.changeSetupPin = getConfigListEntry(_("Change setup pin"), NoSave(ConfigNothing()))
+				self.list.append(self.changeSetupPin)
+			self.list.append(getConfigListEntry(_("Protect services"), config.ParentalControl.servicepinactive))
+			if config.ParentalControl.servicepinactive.value:
+				self.list.append(getConfigListEntry(_("Parental control type"), config.ParentalControl.type))
+				if config.ParentalControl.mode.value == "complex":
+					self.changePin = getConfigListEntry(_("Change service pins"), NoSave(ConfigNothing()))
+					self.list.append(self.changePin)
+				elif config.ParentalControl.mode.value == "simple":	
+					self.changePin = getConfigListEntry(_("Change service pin"), NoSave(ConfigNothing()))
+					self.list.append(self.changePin)
+				self.list.append(getConfigListEntry(_("Remember service pin"), config.ParentalControl.storeservicepin))	
+				self.editListEntry = getConfigListEntry(_("Edit services list"), NoSave(ConfigNothing()))
+				self.list.append(self.editListEntry)
+				
+		self["config"].list = self.list
+		self["config"].setList(self.list)
+		
+	def keyOK(self):
+		print "self[\"config\"].l.getCurrentSelection()", self["config"].l.getCurrentSelection()
+		if self["config"].l.getCurrentSelection() == self.editListEntry:
+			self.session.open(ParentalControlEditor)
+		elif self["config"].l.getCurrentSelection() == self.changePin:
+			if config.ParentalControl.mode.value == "complex":
+				pass
+			else:
+				self.session.open(ParentalControlChangePin, config.ParentalControl.servicepin[0], _("service pin"))
+		elif self["config"].l.getCurrentSelection() == self.changeSetupPin:
+			self.session.open(ParentalControlChangePin, config.ParentalControl.setuppin, _("setup pin"))
+		else:
+			ConfigListScreen.keyRight(self)
+			print "current selection:", self["config"].l.getCurrentSelection()
+			self.createSetup()
+			
+	def keyLeft(self):
+		ConfigListScreen.keyLeft(self)
+		print "current selection:", self["config"].l.getCurrentSelection()
+		self.createSetup()		
+			
+	def keyRight(self):
+		ConfigListScreen.keyRight(self)
+		print "current selection:", self["config"].l.getCurrentSelection()
+		self.createSetup()
+	
+	def keyCancel(self):
+		for x in self["config"].list:
+			x[1].save()
+		self.close()
+	
+	def keyNumberGlobal(self, number):
+		pass
+
+class ParentalControlEditor(Screen):
+	def __init__(self, session):
+		Screen.__init__(self, session)
+
+		self.list = []
+		self.servicelist = ParentalControlList(self.list)
+		self["servicelist"] = self.servicelist;
+		
+		#self.onShown.append(self.chooseLetter)
+		self.currentLetter = ''
+		
+		self.readServiceList()
+		
+		self["actions"] = NumberActionMap(["DirectionActions", "ColorActions", "OkCancelActions", "NumberActions"],
+		{
+			"ok": self.select,
+			"cancel": self.cancel,
+			"red": self.chooseLetter,
+			#"left": self.keyLeft,
+			#"right": self.keyRight,
+			"1": self.keyNumberGlobal,
+			"2": self.keyNumberGlobal,
+			"3": self.keyNumberGlobal,
+			"4": self.keyNumberGlobal,
+			"5": self.keyNumberGlobal,
+			"6": self.keyNumberGlobal,
+			"7": self.keyNumberGlobal,
+			"8": self.keyNumberGlobal,
+			"9": self.keyNumberGlobal,
+			"0": self.keyNumberGlobal
+		}, -1)
+		
+	def cancel(self):
+		parentalControl.save()
+		self.close()
+		
+	def select(self):
+		self.servicelist.toggleSelectedLock()
+	
+	def keyNumberGlobal(self, number):
+		pass
+	
+	def readServiceList(self):
+		serviceHandler = eServiceCenter.getInstance()
+		self.service_types_tv = '1:7:1:0:0:0:0:0:0:0:(type == 1) || (type == 17) || (type == 195) || (type == 25)'
+		refstr = '%s ORDER BY name' % (self.service_types_tv)
+		self.root = eServiceReference(refstr)
+		
+		self.servicesList = {}
+		
+		list = serviceHandler.list(self.root)
+		if list is not None:
+			while 1:
+				s = list.getNext()
+				if s.valid():
+					service = s.toString()
+					name = ServiceReference(service).getServiceName()
+					key = name.lower()[0]
+					if key < 'a' or key > 'z':
+						key = '&'
+					#key = str(key)
+					if not self.servicesList.has_key(key):
+						self.servicesList[key] = []
+					# (servicestring, eServiceRef, name)
+					self.servicesList[key].append((service, s, name))
+				else:
+					break
+		print self.servicesList
+
+	def chooseLetter(self):
+		print "choose letter"
+		list = []
+		for x in self.servicesList.keys():
+			if x == '&':
+				x = ("special characters", x)
+			else:
+				x = (x, x)
+			list.append(x)
+		print "sorted list:", sorted(list, key=operator.itemgetter(1))
+		print self.servicesList.keys()
+		self.session.openWithCallback(self.letterChosen, ChoiceBox, title=_("Show services beginning with"), list=list)
+		
+	def letterChosen(self, result):
+		if result is not None:
+			print "result:", result
+			self.currentLetter = result[1]
+			self.list = []
+			for x in self.servicesList[result[1]]:
+				self.list.append(ParentalControlEntryComponent(x[1], x[2], parentalControl.getProtectionLevel(x[1]) != -1))
+			self.servicelist.setList(self.list)			
+
+class ParentalControlChangePin(Screen, ConfigListScreen, ProtectedScreen):
+	def __init__(self, session, pin, pinname):
+		Screen.__init__(self, session)
+
+		self.pin = pin
+
+		self.list = []
+		self.pin1 = ConfigPIN(default = 1111, censor = "*")
+		self.pin2 = ConfigPIN(default = 1112, censor = "*")
+		self.list.append(getConfigListEntry(_("New pin"), NoSave(self.pin1)))
+		self.list.append(getConfigListEntry(_("Reenter new pin"), NoSave(self.pin2)))
+		ConfigListScreen.__init__(self, self.list)
+		
+#		print "old pin:", pin
+		#if pin.value != "aaaa":
+			#self.onFirstExecBegin.append(boundFunction(self.session.openWithCallback, self.pinEntered, PinInput, pinList = [self.pin.value], title = _("please enter the old pin"), windowTitle = _("Change pin code")))
+		ProtectedScreen.__init__(self)
+		
+		self["actions"] = NumberActionMap(["DirectionActions", "ColorActions", "OkCancelActions"],
+		{
+			"cancel": self.cancel,
+		}, -1)
+		
+	def getPinText(self):
+		return _("Please enter the old pin code")
+
+	def isProtected(self):
+		return (self.pin.value != "aaaa")
+
+	def protectedWithPin(self):
+		return self.pin.value
+		
+#	def pinEntered(self, result):
+		#if result[0] is None:
+			#self.close()
+		#if not result[0]:
+			#print result, "-", self.pin.value
+			#self.session.openWithCallback(self.close, MessageBox, _("The pin code you entered is wrong."), MessageBox.TYPE_ERROR)
+	
+	def keyOK(self):
+		if self.pin1.value == self.pin2.value:
+			self.pin.value = self.pin1.value
+			self.pin.save()
+			self.session.openWithCallback(self.close, MessageBox, _("The pin code has been changed successfully."), MessageBox.TYPE_INFO)
+		else:
+			self.session.open(MessageBox, _("The pin codes you entered are different."), MessageBox.TYPE_ERROR)
+	
+	def cancel(self):
+		self.close(None)
+	
+	def keyNumberGlobal(self, number):
+		ConfigListScreen.keyNumberGlobal(self, number)
+