moparisthebest/SickRage
1
0
Fork 0
mirror of https://github.com/moparisthebest/SickRage synced 2024-12-14 03:52:22 -05:00
Code Issues Releases Wiki Activity

Improved name parser scoring

Browse Source
This commit is contained in:
echel0n 2014-07-18 17:44:06 -07:00
parent c4a0f314fd
commit e0e10dd289
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
sickbeard/name_parser/parser.py
View File

@ -107,16 +107,12 @@ class NameParser(object):
bestResult = None bestResult = None
for regexMode in self.regexModes: for regexMode in self.regexModes:
if doneSearch:
break
self._compile_regexes(regexMode) self._compile_regexes(regexMode)
for (cur_regexMode, cur_regex_name, cur_regex) in self.compiled_regexes: for (cur_regexMode, cur_regex_name, cur_regex) in self.compiled_regexes:
time.sleep(0.02) time.sleep(0.02)
if doneSearch:
break
match = cur_regex.match(name) match = cur_regex.match(name)
if not match: if not match:
@ -134,6 +130,7 @@ class NameParser(object):
if result.series_name: if result.series_name:
result.series_name = self.clean_series_name(result.series_name) result.series_name = self.clean_series_name(result.series_name)
if not result.show:
if self.showObj and self.showObj.name.lower() == result.series_name.lower(): if self.showObj and self.showObj.name.lower() == result.series_name.lower():
result.show = self.showObj result.show = self.showObj
else: else:
@ -219,11 +216,11 @@ class NameParser(object):
if result.show: if result.show:
if regexMode == self.NORMAL_REGEX and not (result.show.is_anime or result.show.is_sports): if regexMode == self.NORMAL_REGEX and not (result.show.is_anime or result.show.is_sports):
doneSearch = True result.score += 1
elif regexMode == self.SPORTS_REGEX and result.show.is_sports: elif regexMode == self.SPORTS_REGEX and result.show.is_sports:
doneSearch = True result.score += 1
elif regexMode == self.ANIME_REGEX and result.show.is_anime: elif regexMode == self.ANIME_REGEX and result.show.is_anime:
doneSearch = True result.score += 1
matches.append(result) matches.append(result)