Cleaned up usage of helper functions
This commit is contained in:
1 parent
da87e68fad
commit
b902186389
3 files changed
+7
-16
No files matched your search
@@ -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':
|
||||||
|
|||||||
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user