Cleaned up usage of helper functions

This commit is contained in:
Dean Gardiner committed 2013-10-16 14:58:50 +13:00
1 parent da87e68fad
commit b902186389
3 files changed
+7 -16

No files matched your search

+3
View File
@@ -211,3 +211,6 @@ def randomString(size = 8, chars = string.ascii_uppercase + string.digits):
def splitString(str, split_on = ',', clean = True): def splitString(str, split_on = ',', clean = True):
list = [x.strip() for x in str.split(split_on)] if str else [] list = [x.strip() for x in str.split(split_on)] if str else []
return filter(None, list) if clean else list return filter(None, list) if clean else list
def dictIsSubset(a, b):
return all([k in b and b[k] == v for k, v in a.items()])
@@ -38,7 +38,6 @@ class ShowSearcher(Plugin):
addEvent('searcher.get_media_identifier', self.getMediaIdentifier) addEvent('searcher.get_media_identifier', self.getMediaIdentifier)
addEvent('searcher.get_media_root', self.getMediaRoot) addEvent('searcher.get_media_root', self.getMediaRoot)
addEvent('searcher.get_media_searcher_id', self.getMediaSearcherId) addEvent('searcher.get_media_searcher_id', self.getMediaSearcherId)
addEvent('searcher.get_media_titles', self.getMediaTitles)
def single(self, media, search_protocols = None, manual = False): def single(self, media, search_protocols = None, manual = False):
if media['type'] == 'show': if media['type'] == 'show':
+4 -15
View File
@@ -2,7 +2,8 @@ import pprint
from caper import Caper from caper import Caper
from couchpotato import CPLog, tryInt from couchpotato import CPLog, tryInt
from couchpotato.core.event import addEvent, fireEvent from couchpotato.core.event import addEvent, fireEvent
from couchpotato.core.helpers.variable import possibleTitles from couchpotato.core.helpers.encoding import simplifyString
from couchpotato.core.helpers.variable import possibleTitles, dictIsSubset
from couchpotato.core.plugins.base import Plugin from couchpotato.core.plugins.base import Plugin
log = CPLog(__name__) log = CPLog(__name__)
@@ -40,7 +41,7 @@ class Matcher(Plugin):
for match in chain.info[group]: for match in chain.info[group]:
for ck, cv in match.items(): for ck, cv in match.items():
if ck in tags and self.cleanMatchValue(cv) in tags[ck]: if ck in tags and simplifyString(cv) in tags[ck]:
found_tags.append(ck) found_tags.append(ck)
@@ -49,18 +50,6 @@ class Matcher(Plugin):
return set([key for key, value in tags.items() if None not in value]) == set(found_tags) return set([key for key, value in tags.items() if None not in value]) == set(found_tags)
def cleanMatchValue(self, value):
value = value.lower()
value = value.strip()
for ch in [' ', '-', '.']:
value = value.replace(ch, '')
return value
def dictIsSubset(self, a, b):
return all([k in b and b[k] == v for k, v in a.items()])
def correctIdentifier(self, chain, media): def correctIdentifier(self, chain, media):
required_id = fireEvent('searcher.get_media_identifier', media['library'], single = True) required_id = fireEvent('searcher.get_media_identifier', media['library'], single = True)
@@ -78,7 +67,7 @@ class Matcher(Plugin):
for k, v in identifier.items(): for k, v in identifier.items():
identifier[k] = tryInt(v, None) identifier[k] = tryInt(v, None)
if not self.dictIsSubset(required_id, identifier): if not dictIsSubset(required_id, identifier):
log.info2('Wrong: required identifier %s does not match release identifier %s', (str(required_id), str(identifier))) log.info2('Wrong: required identifier %s does not match release identifier %s', (str(required_id), str(identifier)))
return False return False