use service compare strings instead of eServiceReferences in parental control

2 files changed, 14 insertions, 15 deletions

diff --git a/lib/python/Components/ParentalControl.py b/lib/python/Components/ParentalControl.py
index f8a6dff2..97e9fad5 100644
--- a/lib/python/Components/ParentalControl.py
+++ b/lib/python/Components/ParentalControl.py
@@ -37,26 +37,25 @@ class ParentalControl:
 		self.tries = 3
 		
 	def addWhitelistService(self, service):
-		self.whitelist.append(service.toCompareString())
+		self.whitelist.append(service)
 	
 	def addBlacklistService(self, service):
-		self.blacklist.append(service.toCompareString())
+		self.blacklist.append(service)
 
 	def setServiceLevel(self, service, level):
-		self.serviceLevel[service.toCompareString()] = level
+		self.serviceLevel[service] = level
 
 	def deleteWhitelistService(self, service):
-		self.whitelist.remove(service.toCompareString())
-		if self.serviceLevel.has_key(service.toCompareString()):
-			self.serviceLevel.remove(service.toCompareString())
+		self.whitelist.remove(service)
+		if self.serviceLevel.has_key(service):
+			self.serviceLevel.remove(service)
 	
 	def deleteBlacklistService(self, service):
-		self.blacklist.remove(service.toCompareString())
-		if self.serviceLevel.has_key(service.toCompareString()):
-			self.serviceLevel.remove(service.toCompareString())
+		self.blacklist.remove(service)
+		if self.serviceLevel.has_key(service):
+			self.serviceLevel.remove(service)
 				
-	def isServicePlayable(self, serviceref, callback):
-		service = serviceref.toCompareString()
+	def isServicePlayable(self, service, callback):
 		if not config.ParentalControl.configured.value:
 			return True
 		print "whitelist:", self.whitelist
@@ -80,10 +79,10 @@ class ParentalControl:
 		print "protect"
 		print "config.ParentalControl.type.value:", config.ParentalControl.type.value
 		if config.ParentalControl.type.value == "whitelist":
-			if service.toCompareString() in self.whitelist:
+			if service in self.whitelist:
 				self.deleteWhitelistService(service)
 		else: # blacklist
-			if service.toCompareString() not in self.blacklist:
+			if service not in self.blacklist:
 				self.addBlacklistService(service)
 		print "whitelist:", self.whitelist
 		print "blacklist:", self.blacklist
diff --git a/lib/python/Screens/ChannelSelection.py b/lib/python/Screens/ChannelSelection.py
index 58c57f41..54c48293 100644
--- a/lib/python/Screens/ChannelSelection.py
+++ b/lib/python/Screens/ChannelSelection.py
@@ -139,11 +139,11 @@ class ChannelContextMenu(Screen):
 		self.close()
 
 	def addParentalProtection(self, service):
-		parentalControl.protectService(service)
+		parentalControl.protectService(service.toCompareString())
 		self.close()
 
 	def removeParentalProtection(self, service):
-		self.session.openWithCallback(boundFunction(self.pinEntered, service), PinInput, pinList = [config.ParentalControl.servicepin[0].value], title = _("Enter the service pin"), windowTitle = _("Change pin code"))
+		self.session.openWithCallback(boundFunction(self.pinEntered, service.toCompareString()), PinInput, pinList = [config.ParentalControl.servicepin[0].value], title = _("Enter the service pin"), windowTitle = _("Change pin code"))
 
 	def pinEntered(self, service, result):
 		if result[0]: