monkee.ch/SickGear
1
0
Fork 0
mirror of https://github.com/SickGear/SickGear.git synced 2025-01-08 11:03:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #115 from adam111316/feature/FixNameParserUnittest

Browse source 
Fix name parser unit tests and regex
This commit is contained in:
adam111316 2015-01-02 07:21:05 +08:00
commit 1d30a6b63f
4 changed files with 806 additions and 664 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
CHANGES.md
View file

@ -12,6 +12,9 @@
* Change Search Settings/Torrent/Deluge option texts for improved understanding * Change Search Settings/Torrent/Deluge option texts for improved understanding
* Fix Womble's Index searching (ssl disabled for now, old categories are the new active ones again) * Fix Womble's Index searching (ssl disabled for now, old categories are the new active ones again)
* Fix Add From Trending Show page to work with Trakt changes * Fix Add From Trending Show page to work with Trakt changes
* Add anime unit test cases (port from lad1337/sickbeard)
* Fix normal tv show regex (port from midgetspy/sickbeard)
* Fix anime regex (port from lad1337/sickbeard)
[develop changelog] [develop changelog]
@ -56,6 +59,10 @@
* Fix restart issue * Fix restart issue
* Fix to use new TorrentDay URLs * Fix to use new TorrentDay URLs
* Fix typo in menu item Manage/Update XBMC * Fix typo in menu item Manage/Update XBMC
* Fix NameParser unittests
* Add Anime unittest cases (port from lad1337/sickbeard)
* Fix normal tv show regex (port from midgetspy/sickbeard)
* Fix anime regex (port from lad1337/sickbeard)
### 0.4.0 (2014-12-04 10:50:00 UTC) ### 0.4.0 (2014-12-04 10:50:00 UTC)

70
sickbeard/name_parser/parser.py
View file

@ -26,8 +26,9 @@ import os.path
import regexes import regexes
import sickbeard import sickbeard
from sickbeard import logger, helpers, scene_numbering, common, exceptions, scene_exceptions, encodingKludge as ek, db from sickbeard import logger, helpers, scene_numbering, common, scene_exceptions, encodingKludge as ek, db
from dateutil import parser from dateutil import parser
from sickbeard.exceptions import ex
class NameParser(object): class NameParser(object):
@ -36,13 +37,14 @@ class NameParser(object):
ANIME_REGEX = 2 ANIME_REGEX = 2
def __init__(self, file_name=True, showObj=None, tryIndexers=False, convert=False, def __init__(self, file_name=True, showObj=None, tryIndexers=False, convert=False,
naming_pattern=False): naming_pattern=False, testing=False):
self.file_name = file_name self.file_name = file_name
self.showObj = showObj self.showObj = showObj
self.tryIndexers = tryIndexers self.tryIndexers = tryIndexers
self.convert = convert self.convert = convert
self.naming_pattern = naming_pattern self.naming_pattern = naming_pattern
self.testing = testing
if self.showObj and not self.showObj.is_anime: if self.showObj and not self.showObj.is_anime:
self._compile_regexes(self.NORMAL_REGEX) self._compile_regexes(self.NORMAL_REGEX)
@ -78,7 +80,7 @@ class NameParser(object):
def _compile_regexes(self, regexMode): def _compile_regexes(self, regexMode):
if regexMode == self.ANIME_REGEX: if regexMode == self.ANIME_REGEX:
logger.log(u"Using ANIME regexs", logger.DEBUG) logger.log(u"Using ANIME regexs", logger.DEBUG)
uncompiled_regex = [regexes.anime_regexes, regexes.normal_regexes] uncompiled_regex = [regexes.anime_regexes]
elif regexMode == self.NORMAL_REGEX: elif regexMode == self.NORMAL_REGEX:
logger.log(u"Using NORMAL regexs", logger.DEBUG) logger.log(u"Using NORMAL regexs", logger.DEBUG)
uncompiled_regex = [regexes.normal_regexes] uncompiled_regex = [regexes.normal_regexes]
@ -86,7 +88,8 @@ class NameParser(object):
logger.log(u"Using ALL regexes", logger.DEBUG) logger.log(u"Using ALL regexes", logger.DEBUG)
uncompiled_regex = [regexes.normal_regexes, regexes.anime_regexes] uncompiled_regex = [regexes.normal_regexes, regexes.anime_regexes]
self.compiled_regexes = [] self.compiled_regexes = {0: [], 1: []}
index = 0
for regexItem in uncompiled_regex: for regexItem in uncompiled_regex:
for cur_pattern_num, (cur_pattern_name, cur_pattern) in enumerate(regexItem): for cur_pattern_num, (cur_pattern_name, cur_pattern) in enumerate(regexItem):
try: try:
@ -94,7 +97,8 @@ class NameParser(object):
except re.error, errormsg: except re.error, errormsg:
logger.log(u"WARNING: Invalid episode_pattern, %s. %s" % (errormsg, cur_pattern)) logger.log(u"WARNING: Invalid episode_pattern, %s. %s" % (errormsg, cur_pattern))
else: else:
self.compiled_regexes.append((cur_pattern_num, cur_pattern_name, cur_regex)) self.compiled_regexes[index].append([cur_pattern_num, cur_pattern_name, cur_regex])
index += 1
def _parse_string(self, name): def _parse_string(self, name):
if not name: if not name:
@ -102,8 +106,8 @@ class NameParser(object):
matches = [] matches = []
bestResult = None bestResult = None
for regex in self.compiled_regexes:
for (cur_regex_num, cur_regex_name, cur_regex) in self.compiled_regexes: for (cur_regex_num, cur_regex_name, cur_regex) in self.compiled_regexes[regex]:
match = cur_regex.match(name) match = cur_regex.match(name)
if not match: if not match:
@ -150,13 +154,19 @@ class NameParser(object):
result.ab_episode_numbers = [ep_ab_num] result.ab_episode_numbers = [ep_ab_num]
result.score += 1 result.score += 1
if 'air_date' in named_groups: if 'air_year' in named_groups and 'air_month' in named_groups and 'air_day' in named_groups:
air_date = match.group('air_date') year = int(match.group('air_year'))
month = int(match.group('air_month'))
day = int(match.group('air_day'))
# make an attempt to detect YYYY-DD-MM formats
if month > 12:
tmp_month = month
month = day
day = tmp_month
try: try:
result.air_date = parser.parse(air_date, fuzzy=True).date() result.air_date = datetime.date(year, month, day)
result.score += 1 except ValueError, e:
except: raise InvalidNameException(ex(e))
continue
if 'extra_info' in named_groups: if 'extra_info' in named_groups:
tmp_extra_info = match.group('extra_info') tmp_extra_info = match.group('extra_info')
@ -194,13 +204,16 @@ class NameParser(object):
show = helpers.get_show(bestResult.series_name, self.tryIndexers) show = helpers.get_show(bestResult.series_name, self.tryIndexers)
# confirm passed in show object indexer id matches result show object indexer id # confirm passed in show object indexer id matches result show object indexer id
if show: if show and not self.testing:
if self.showObj and show.indexerid != self.showObj.indexerid: if self.showObj and show.indexerid != self.showObj.indexerid:
show = None show = None
bestResult.show = show bestResult.show = show
elif not show and self.showObj: elif not show and self.showObj:
bestResult.show = self.showObj bestResult.show = self.showObj
if bestResult.show and bestResult.show.is_anime and len(self.compiled_regexes[1]) > 1 and regex != 1:
continue
# if this is a naming pattern test or result doesn't have a show object then return best result # if this is a naming pattern test or result doesn't have a show object then return best result
if not bestResult.show or self.naming_pattern: if not bestResult.show or self.naming_pattern:
return bestResult return bestResult
@ -259,7 +272,7 @@ class NameParser(object):
new_episode_numbers.append(e) new_episode_numbers.append(e)
new_season_numbers.append(s) new_season_numbers.append(s)
elif bestResult.show.is_anime and len(bestResult.ab_episode_numbers): elif bestResult.show.is_anime and len(bestResult.ab_episode_numbers) and not self.testing:
scene_season = scene_exceptions.get_scene_exception_by_name(bestResult.series_name)[1] scene_season = scene_exceptions.get_scene_exception_by_name(bestResult.series_name)[1]
for epAbsNo in bestResult.ab_episode_numbers: for epAbsNo in bestResult.ab_episode_numbers:
a = epAbsNo a = epAbsNo
@ -275,7 +288,7 @@ class NameParser(object):
new_episode_numbers.extend(e) new_episode_numbers.extend(e)
new_season_numbers.append(s) new_season_numbers.append(s)
elif bestResult.season_number and len(bestResult.episode_numbers): elif bestResult.season_number and len(bestResult.episode_numbers) and not self.testing:
for epNo in bestResult.episode_numbers: for epNo in bestResult.episode_numbers:
s = bestResult.season_number s = bestResult.season_number
e = epNo e = epNo
@ -416,7 +429,7 @@ class NameParser(object):
file_name_result = self._parse_string(base_file_name) file_name_result = self._parse_string(base_file_name)
# use only the direct parent dir # use only the direct parent dir
dir_name = os.path.basename(dir_name) dir_name = ek.ek(os.path.basename, dir_name)
# parse the dirname for extra info if needed # parse the dirname for extra info if needed
dir_name_result = self._parse_string(dir_name) dir_name_result = self._parse_string(dir_name)
@ -452,6 +465,10 @@ class NameParser(object):
final_result.quality = self._combine_results(file_name_result, dir_name_result, 'quality') final_result.quality = self._combine_results(file_name_result, dir_name_result, 'quality')
if not final_result.show: if not final_result.show:
if self.testing:
pass
#final_result.which_regex = []
else:
raise InvalidShowException( raise InvalidShowException(
"Unable to parse " + name.encode(sickbeard.SYS_ENCODING, 'xmlcharrefreplace')) "Unable to parse " + name.encode(sickbeard.SYS_ENCODING, 'xmlcharrefreplace'))
@ -506,7 +523,7 @@ class ParseResult(object):
self.air_date = air_date self.air_date = air_date
self.which_regex = [] self.which_regex = None
self.show = show self.show = show
self.score = score self.score = score
@ -530,14 +547,14 @@ class ParseResult(object):
return False return False
if self.ab_episode_numbers != other.ab_episode_numbers: if self.ab_episode_numbers != other.ab_episode_numbers:
return False return False
if self.show != other.show: #if self.show != other.show:
return False # return False
if self.score != other.score: #if self.score != other.score:
return False # return False
if self.quality != other.quality: #if self.quality != other.quality:
return False # return False
if self.version != other.version: #if self.version != other.version:
return False # return False
return True return True
@ -556,6 +573,7 @@ class ParseResult(object):
to_return += str(self.air_date) to_return += str(self.air_date)
if self.ab_episode_numbers: if self.ab_episode_numbers:
to_return += ' [ABS: ' + str(self.ab_episode_numbers) + ']' to_return += ' [ABS: ' + str(self.ab_episode_numbers) + ']'
if self.is_anime:
if self.version: if self.version:
to_return += ' [ANIME VER: ' + str(self.version) + ']' to_return += ' [ANIME VER: ' + str(self.version) + ']'

188
sickbeard/name_parser/regexes.py
View file

@ -30,8 +30,9 @@ normal_regexes = [
e(?P<extra_ep_num>\d+))+ # E03/etc and separator e(?P<extra_ep_num>\d+))+ # E03/etc and separator
[. _-]*((?P<extra_info>.+?) # Source_Quality_Etc- [. _-]*((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group ((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group -(?P<release_group>[^- ]+))?)?$ # Group
'''), '''
),
('fov_repeat', ('fov_repeat',
# Show.Name.1x02.1x03.Source.Quality.Etc-Group # Show.Name.1x02.1x03.Source.Quality.Etc-Group
@ -44,8 +45,9 @@ normal_regexes = [
(?P<extra_ep_num>\d+))+ # 03/etc and separator (?P<extra_ep_num>\d+))+ # 03/etc and separator
[. _-]*((?P<extra_info>.+?) # Source_Quality_Etc- [. _-]*((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group ((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group -(?P<release_group>[^- ]+))?)?$ # Group
'''), '''
),
('standard', ('standard',
# Show.Name.S01E02.Source.Quality.Etc-Group # Show.Name.S01E02.Source.Quality.Etc-Group
@ -62,8 +64,9 @@ normal_regexes = [
(?P<extra_ep_num>(?!(1080|720|480)[pi])\d+))* # additional E03/etc (?P<extra_ep_num>(?!(1080|720|480)[pi])\d+))* # additional E03/etc
[. _-]*((?P<extra_info>.+?) # Source_Quality_Etc- [. _-]*((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group ((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group -(?P<release_group>[^- ]+))?)?$ # Group
'''), '''
),
('fov', ('fov',
# Show_Name.1x02.Source_Quality_Etc-Group # Show_Name.1x02.Source_Quality_Etc-Group
@ -80,31 +83,23 @@ normal_regexes = [
\d+))* # additional x03/etc \d+))* # additional x03/etc
[\]. _-]*((?P<extra_info>.+?) # Source_Quality_Etc- [\]. _-]*((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group ((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group -(?P<release_group>[^- ]+))?)?$ # Group
'''), '''
),
('scene_date_format', ('scene_date_format',
# Show.Name.2010.11.23.Source.Quality.Etc-Group # Show.Name.2010.11.23.Source.Quality.Etc-Group
# Show Name - 2010-11-23 - Ep Name # Show Name - 2010-11-23 - Ep Name
''' '''
^((?P<series_name>.+?)[. _-]+)? # Show_Name and separator ^((?P<series_name>.+?)[. _-]+)? # Show_Name and separator
(?P<air_date>(\d+[. _-]\d+[. _-]\d+)|(\d+\w+[. _-]\w+[. _-]\d+)) (?P<air_year>\d{4})[. _-]+ # 2010 and separator
(?P<air_month>\d{2})[. _-]+ # 11 and separator
(?P<air_day>\d{2}) # 23 and separator
[. _-]*((?P<extra_info>.+?) # Source_Quality_Etc- [. _-]*((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group ((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group -(?P<release_group>[^- ]+))?)?$ # Group
'''),
('scene_sports_format',
# Show.Name.100.Event.2010.11.23.Source.Quality.Etc-Group
# Show.Name.2010.11.23.Source.Quality.Etc-Group
# Show Name - 2010-11-23 - Ep Name
''' '''
^(?P<series_name>.*?(UEFA|MLB|ESPN|WWE|MMA|UFC|TNA|EPL|NASCAR|NBA|NFL|NHL|NRL|PGA|SUPER LEAGUE|FORMULA|FIFA|NETBALL|MOTOGP).*?)[. _-]+ ),
((?P<series_num>\d{1,3})[. _-]+)?
(?P<air_date>(\d+[. _-]\d+[. _-]\d+)|(\d+\w+[. _-]\w+[. _-]\d+))[. _-]+
((?P<extra_info>.+?)((?<![. _-])
(?<!WEB)-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$
'''),
('stupid', ('stupid',
# tpz-abc102 # tpz-abc102
@ -113,7 +108,8 @@ normal_regexes = [
(?!264) # don't count x264 (?!264) # don't count x264
(?P<season_num>\d{1,2}) # 1 (?P<season_num>\d{1,2}) # 1
(?P<ep_num>\d{2})$ # 02 (?P<ep_num>\d{2})$ # 02
'''), '''
),
('verbose', ('verbose',
# Show Name Season 1 Episode 2 Ep Name # Show Name Season 1 Episode 2 Ep Name
@ -124,7 +120,8 @@ normal_regexes = [
episode[. _-]+ # episode and separator episode[. _-]+ # episode and separator
(?P<ep_num>\d+)[. _-]+ # 02 and separator (?P<ep_num>\d+)[. _-]+ # 02 and separator
(?P<extra_info>.+)$ # Source_Quality_Etc- (?P<extra_info>.+)$ # Source_Quality_Etc-
'''), '''
),
('season_only', ('season_only',
# Show.Name.S01.Source.Quality.Etc-Group # Show.Name.S01.Source.Quality.Etc-Group
@ -134,7 +131,7 @@ normal_regexes = [
(?P<season_num>\d+)[. _-]* # S01 and optional separator (?P<season_num>\d+)[. _-]* # S01 and optional separator
[. _-]*((?P<extra_info>.+?) # Source_Quality_Etc- [. _-]*((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group ((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group -(?P<release_group>[^- ]+))?)?$ # Group
''' '''
), ),
@ -149,7 +146,7 @@ normal_regexes = [
(?P<extra_ep_num>(?!(1080|720|480)[pi])(\d+|[ivx]+))[. _-]) # second ep num (?P<extra_ep_num>(?!(1080|720|480)[pi])(\d+|[ivx]+))[. _-]) # second ep num
([. _-]*(?P<extra_info>.+?) # Source_Quality_Etc- ([. _-]*(?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group ((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group -(?P<release_group>[^- ]+))?)?$ # Group
''' '''
), ),
@ -167,22 +164,7 @@ normal_regexes = [
(\d+|([ivx]+(?=[. _-]))))[. _-])* # second ep num (\d+|([ivx]+(?=[. _-]))))[. _-])* # second ep num
([. _-]*(?P<extra_info>.+?) # Source_Quality_Etc- ([. _-]*(?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group ((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group -(?P<release_group>[^- ]+))?)?$ # Group
'''
),
('no_season',
# Show Name - 01 - Ep Name
# 01 - Ep Name
# 01 - Ep Name
'''
^((?P<series_name>.+?)(?:[. _-]{2,}|[. _]))? # Show_Name and separator
(?P<ep_num>\d{1,3}) # 02
(?:-(?P<extra_ep_num>\d{1,3}))* # -03-04-05 etc
\s?of?\s?\d{1,3}? # of joiner (with or without spaces) and series total ep
[. _-]+((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group
''' '''
), ),
@ -193,23 +175,37 @@ normal_regexes = [
(?P<season_num>\d{1,2}) # 1 (?P<season_num>\d{1,2}) # 1
(?P<ep_num>\d{2}) # 02 and separator (?P<ep_num>\d{2}) # 02 and separator
([. _-]+(?P<extra_info>(?!\d{3}[. _-]+)[^-]+) # Source_Quality_Etc- ([. _-]+(?P<extra_info>(?!\d{3}[. _-]+)[^-]+) # Source_Quality_Etc-
(-(?P<release_group>[^- ]+([. _-]\[.*\])?))?)?$ # Group (-(?P<release_group>.+))?)?$ # Group
'''), '''
),
('no_season',
# Show Name - 01 - Ep Name
# 01 - Ep Name
'''
^((?P<series_name>.+?)(?:[. _-]{2,}|[. _]))? # Show_Name and separator
(?P<ep_num>\d{1,2}) # 01
(?:-(?P<extra_ep_num>\d{1,2}))* # 02
[. _-]+((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+))?)?$ # Group
'''
),
] ]
anime_regexes = [ anime_regexes = [
('anime_ultimate', ('anime_ultimate',
""" '''
^(?:\[(?P<release_group>.+?)\][ ._-]*) ^(?:\[(?P<release_group>.+?)\][ ._-]*)
(?P<series_name>.+?)[ ._-]+ (?P<series_name>.+?)[ ._-]+
(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) (?P<ep_ab_num>\d{1,3})
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))?[ ._-]+? (-(?P<extra_ab_ep_num>\d{1,3}))?[ ._-]+?
(?:v(?P<version>[0-9]))? (?:v(?P<version>[0-9]))?
(?:[\w\.]*) (?:[\w\.]*)
(?:(?:(?:[\[\(])(?P<extra_info>\d{3,4}[xp]?\d{0,4}[\.\w\s-]*)(?:[\]\)]))|(?:\d{3,4}[xp])) (?:(?:(?:[\[\(])(?P<extra_info>\d{3,4}[xp]?\d{0,4}[\.\w\s-]*)(?:[\]\)]))|(?:\d{3,4}[xp]))
(?:[ ._]?\[(?P<crc>\w+)\])? (?:[ ._]?\[(?P<crc>\w+)\])?
.*? .*?
""" '''
), ),
('anime_standard', ('anime_standard',
# [Group Name] Show Name.13-14 # [Group Name] Show Name.13-14
@ -221,42 +217,38 @@ anime_regexes = [
''' '''
^(\[(?P<release_group>.+?)\][ ._-]*)? # Release Group and separator ^(\[(?P<release_group>.+?)\][ ._-]*)? # Release Group and separator
(?P<series_name>.+?)[ ._-]+ # Show_Name and separator (?P<series_name>.+?)[ ._-]+ # Show_Name and separator
(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) # E01 (?P<ep_ab_num>\d{1,3}) # E01
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # E02 (-(?P<extra_ab_ep_num>\d{1,3}))? # E02
(v(?P<version>[0-9]))? # version (v(?P<version>[0-9]))? # version
[ ._-]+\[(?P<extra_info>\d{3,4}[xp]?\d{0,4}[\.\w\s-]*)\] # Source_Quality_Etc- [ ._-]+\[(?P<extra_info>\d{3,4}[xp]?\d{0,4}[\.\w\s-]*)\] # Source_Quality_Etc-
(\[(?P<crc>\w{8})\])? # CRC (\[(?P<crc>\w{8})\])? # CRC
.*? # Separator and EOL .*? # Separator and EOL
'''), '''),
('anime_standard_round', ('anime_standard_round',
# TODO examples
# [Stratos-Subs]_Infinite_Stratos_-_12_(1280x720_H.264_AAC)_[379759DB] # [Stratos-Subs]_Infinite_Stratos_-_12_(1280x720_H.264_AAC)_[379759DB]
# [ShinBunBu-Subs] Bleach - 02-03 (CX 1280x720 x264 AAC) # [ShinBunBu-Subs] Bleach - 02-03 (CX 1280x720 x264 AAC)
''' '''
^(\[(?P<release_group>.+?)\][ ._-]*)? # Release Group and separator ^(\[(?P<release_group>.+?)\][ ._-]*)? # Release Group and separator
(?P<series_name>.+?)[ ._-]+ # Show_Name and separator (?P<series_name>.+?)[ ._-]+ # Show_Name and separator
(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) # E01 (?P<ep_ab_num>\d{1,3}) # E01
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # E02 (-(?P<extra_ab_ep_num>\d{1,3}))? # E02
(v(?P<version>[0-9]))? # version (v(?P<version>[0-9]))? # version
[ ._-]+\((?P<extra_info>(CX[ ._-]?)?\d{3,4}[xp]?\d{0,4}[\.\w\s-]*)\) # Source_Quality_Etc- [ ._-]+\((?P<extra_info>(CX[ ._-]?)?\d{3,4}[xp]?\d{0,4}[\.\w\s-]*)\) # Source_Quality_Etc-
(\[(?P<crc>\w{8})\])? # CRC (\[(?P<crc>\w{8})\])? # CRC
.*? # Separator and EOL .*? # Separator and EOL
'''), '''),
('anime_slash', ('anime_slash',
# [SGKK] Bleach 312v1 [720p/MKV] # [SGKK] Bleach 312v1 [720p/MKV]
''' '''
^(\[(?P<release_group>.+?)\][ ._-]*)? # Release Group and separator ^(\[(?P<release_group>.+?)\][ ._-]*)? # Release Group and separator
(?P<series_name>.+?)[ ._-]+ # Show_Name and separator (?P<series_name>.+?)[ ._-]+ # Show_Name and separator
(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) # E01 (?P<ep_ab_num>\d{1,3}) # E01
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # E02 (-(?P<extra_ab_ep_num>\d{1,3}))? # E02
(v(?P<version>[0-9]))? # version (v(?P<version>[0-9]))? # version
[ ._-]+\[(?P<extra_info>\d{3,4}p) # Source_Quality_Etc- [ ._-]+\[(?P<extra_info>\d{3,4}p) # Source_Quality_Etc-
(\[(?P<crc>\w{8})\])? # CRC (\[(?P<crc>\w{8})\])? # CRC
.*? # Separator and EOL .*? # Separator and EOL
'''), '''),
('anime_standard_codec', ('anime_standard_codec',
# [Ayako]_Infinite_Stratos_-_IS_-_07_[H264][720p][EB7838FC] # [Ayako]_Infinite_Stratos_-_IS_-_07_[H264][720p][EB7838FC]
# [Ayako] Infinite Stratos - IS - 07v2 [H264][720p][44419534] # [Ayako] Infinite Stratos - IS - 07v2 [H264][720p][44419534]
@ -264,26 +256,15 @@ anime_regexes = [
''' '''
^(\[(?P<release_group>.+?)\][ ._-]*)? # Release Group and separator ^(\[(?P<release_group>.+?)\][ ._-]*)? # Release Group and separator
(?P<series_name>.+?)[ ._]* # Show_Name and separator (?P<series_name>.+?)[ ._]* # Show_Name and separator
([ ._-]+-[ ._-]+[A-Z]+[ ._-]+)?[ ._-]+ # funny stuff, this is sooo nuts ! this will kick me in the butt one day ([ ._-]+-[ ._-]+[A-Z]+[ ._-]+)?[ ._-]+ # this will kick me in the butt one day
(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) # E01 (?P<ep_ab_num>\d{1,3}) # E01
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # E02 (-(?P<extra_ab_ep_num>\d{1,3}))? # E02
(v(?P<version>[0-9]))? # version (v(?P<version>[0-9]))? # version
([ ._-](\[\w{1,2}\])?\[[a-z][.]?\w{2,4}\])? # codec ([ ._-](\[\w{1,2}\])?\[[a-z][.]?\w{2,4}\])? # codec
[ ._-]*\[(?P<extra_info>(\d{3,4}[xp]?\d{0,4})?[\.\w\s-]*)\] # Source_Quality_Etc- [ ._-]*\[(?P<extra_info>(\d{3,4}[xp]?\d{0,4})?[\.\w\s-]*)\] # Source_Quality_Etc-
(\[(?P<crc>\w{8})\])? (\[(?P<crc>\w{8})\])? # CRC
.*? # Separator and EOL .*? # Separator and EOL
'''), '''),
('anime_codec_crc',
'''
^(?:\[(?P<release_group>.*?)\][ ._-]*)?
(?:(?P<series_name>.*?)[ ._-]*)?
(?:(?P<ep_ab_num>(((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))[ ._-]*).+?
(?:\[(?P<codec>.*?)\][ ._-]*)
(?:\[(?P<crc>\w{8})\])?
.*?
'''),
('anime_and_normal', ('anime_and_normal',
# Bleach - s16e03-04 - 313-314 # Bleach - s16e03-04 - 313-314
# Bleach.s16e03-04.313-314 # Bleach.s16e03-04.313-314
@ -294,15 +275,14 @@ anime_regexes = [
[eE](?P<ep_num>\d+) # epipisode E02 [eE](?P<ep_num>\d+) # epipisode E02
(([. _-]*e|-) # linking e/- char (([. _-]*e|-) # linking e/- char
(?P<extra_ep_num>\d+))* # additional E03/etc (?P<extra_ep_num>\d+))* # additional E03/etc
([ ._-]{2,}|[ ._]+) # if "-" is used to separate at least something else has to be there(->{2,}) "s16e03-04-313-314" would make sens any way ([ ._-]{2,}|[ ._]+) # if "-" is used to separate at least something else has to be
((?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # absolute number # there(->{2,}) "s16e03-04-313-314" would make sens any way
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # "-" as separator and anditional absolute number, all optinal (?P<ep_ab_num>\d{1,3}) # absolute number
(-(?P<extra_ab_ep_num>\d{1,3}))* # "-" as separator and anditional absolute number, all optinal
(v(?P<version>[0-9]))? # the version e.g. "v2" (v(?P<version>[0-9]))? # the version e.g. "v2"
.*? .*?
''' '''
), ),
('anime_and_normal_x', ('anime_and_normal_x',
# Bleach - s16e03-04 - 313-314 # Bleach - s16e03-04 - 313-314
# Bleach.s16e03-04.313-314 # Bleach.s16e03-04.313-314
@ -313,23 +293,23 @@ anime_regexes = [
[xX](?P<ep_num>\d+) # epipisode E02 [xX](?P<ep_num>\d+) # epipisode E02
(([. _-]*e|-) # linking e/- char (([. _-]*e|-) # linking e/- char
(?P<extra_ep_num>\d+))* # additional E03/etc (?P<extra_ep_num>\d+))* # additional E03/etc
([ ._-]{2,}|[ ._]+) # if "-" is used to separate at least something else has to be there(->{2,}) "s16e03-04-313-314" would make sens any way ([ ._-]{2,}|[ ._]+) # if "-" is used to separate at least something else has to be
((?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # absolute number # there(->{2,}) "s16e03-04-313-314" would make sens any way
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # "-" as separator and anditional absolute number, all optinal (?P<ep_ab_num>\d{1,3}) # absolute number
(-(?P<extra_ab_ep_num>\d{1,3}))* # "-" as separator and anditional absolute number, all optinal
(v(?P<version>[0-9]))? # the version e.g. "v2" (v(?P<version>[0-9]))? # the version e.g. "v2"
.*? .*?
''' '''
), ),
('anime_and_normal_reverse', ('anime_and_normal_reverse',
# Bleach - 313-314 - s16e03-04 # Bleach - 313-314 - s16e03-04
''' '''
^(?P<series_name>.+?)[ ._-]+ # start of string and series name and non optinal separator ^(?P<series_name>.+?)[ ._-]+ # start of string and series name and non optinal separator
(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) # absolute number (?P<ep_ab_num>\d{1,3}) # absolute number
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # "-" as separator and anditional absolute number, all optinal (-(?P<extra_ab_ep_num>\d{1,3}))* # "-" as separator and anditional absolute number, all optinal
(v(?P<version>[0-9]))? # the version e.g. "v2" (v(?P<version>[0-9]))? # the version e.g. "v2"
([ ._-]{2,}|[ ._]+) # if "-" is used to separate at least something else has to be there(->{2,}) "s16e03-04-313-314" would make sens any way ([ ._-]{2,}|[ ._]+) # if "-" is used to separate at least something else has to be
# there(->{2,}) "s16e03-04-313-314" would make sens any way
[sS](?P<season_num>\d+)[. _-]* # S01 and optional separator [sS](?P<season_num>\d+)[. _-]* # S01 and optional separator
[eE](?P<ep_num>\d+) # epipisode E02 [eE](?P<ep_num>\d+) # epipisode E02
(([. _-]*e|-) # linking e/- char (([. _-]*e|-) # linking e/- char
@ -337,12 +317,11 @@ anime_regexes = [
.*? .*?
''' '''
), ),
('anime_and_normal_front', ('anime_and_normal_front',
# 165.Naruto Shippuuden.s08e014 # 165.Naruto Shippuuden.s08e014
''' '''
^(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) # start of string and absolute number ^(?P<ep_ab_num>\d{1,3}) # start of string and absolute number
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # "-" as separator and anditional absolute number, all optinal (-(?P<extra_ab_ep_num>\d{1,3}))* # "-" as separator and anditional absolute number, all optinal
(v(?P<version>[0-9]))?[ ._-]+ # the version e.g. "v2" (v(?P<version>[0-9]))?[ ._-]+ # the version e.g. "v2"
(?P<series_name>.+?)[ ._-]+ (?P<series_name>.+?)[ ._-]+
[sS](?P<season_num>\d+)[. _-]* # S01 and optional separator [sS](?P<season_num>\d+)[. _-]* # S01 and optional separator
@ -352,13 +331,12 @@ anime_regexes = [
.*? .*?
''' '''
), ),
('anime_ep_name', ('anime_ep_name',
''' '''
^(?:\[(?P<release_group>.+?)\][ ._-]*) ^(?:\[(?P<release_group>.+?)\][ ._-]*)
(?P<series_name>.+?)[ ._-]+ (?P<series_name>.+?)[ ._-]+
(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) (?P<ep_ab_num>\d{1,3})
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))?[ ._-]*? (-(?P<extra_ab_ep_num>\d{1,3}))*[ ._-]*?
(?:v(?P<version>[0-9])[ ._-]+?)? (?:v(?P<version>[0-9])[ ._-]+?)?
(?:.+?[ ._-]+?)? (?:.+?[ ._-]+?)?
\[(?P<extra_info>\w+)\][ ._-]? \[(?P<extra_info>\w+)\][ ._-]?
@ -366,16 +344,34 @@ anime_regexes = [
.*? .*?
''' '''
), ),
('anime_bare', ('anime_bare',
# One Piece - 102 # One Piece - 102
# [ACX]_Wolf's_Spirit_001.mkv # [ACX]_Wolf's_Spirit_001.mkv
''' '''
^(\[(?P<release_group>.+?)\][ ._-]*)? ^(\[(?P<release_group>.+?)\][ ._-]*)?
(?P<series_name>.+?)[ ._-]+ # Show_Name and separator (?P<series_name>.+?)[ ._-]+ # Show_Name and separator
(?P<ep_ab_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}) # E01 (?<!H.)(?P<ep_ab_num>\d{3})(?!0p) # E01, while avoiding H.264 and 1080p from being matched
(-(?P<extra_ab_ep_num>((?!(1080|720|480)[pi])|(?![hx].?264))\d{1,3}))? # E02 (-(?P<extra_ab_ep_num>\d{3}))* # E02
(v(?P<version>[0-9]))? # v2 (v(?P<version>[0-9]))? # v2
.*? # Separator and EOL .*? # Separator and EOL
''') '''),
('standard',
# Show.Name.S01E02.Source.Quality.Etc-Group
# Show Name - S01E02 - My Ep Name
# Show.Name.S01.E03.My.Ep.Name
# Show.Name.S01E02E03.Source.Quality.Etc-Group
# Show Name - S01E02-03 - My Ep Name
# Show.Name.S01.E02.E03
'''
^((?P<series_name>.+?)[. _-]+)? # Show_Name and separator
s(?P<season_num>\d+)[. _-]* # S01 and optional separator
e(?P<ep_num>\d+) # E02 and separator
(([. _-]*e|-) # linking e/- char
(?P<extra_ep_num>(?!(1080|720|480)[pi])\d+))* # additional E03/etc
[. _-]*((?P<extra_info>.+?) # Source_Quality_Etc-
((?<![. _-])(?<!WEB) # Make sure this is really the release group
-(?P<release_group>[^- ]+))?)?$ # Group
'''
),
] ]

313
tests/name_parser_tests.py
View file

@ -3,12 +3,14 @@ import unittest
import test_lib as test import test_lib as test
import sys, os.path import sys, os.path
sys.path.append(os.path.abspath('..')) sys.path.append(os.path.abspath('..'))
sys.path.append(os.path.abspath('../lib')) sys.path.append(os.path.abspath('../lib'))
from sickbeard.name_parser import parser from sickbeard.name_parser import parser
import sickbeard import sickbeard
sickbeard.SYS_ENCODING = 'UTF-8' sickbeard.SYS_ENCODING = 'UTF-8'
DEBUG = VERBOSE = False DEBUG = VERBOSE = False
@ -66,7 +68,7 @@ simple_test_cases = {
'show.name.2010.222.123.source.quality.etc-group': parser.ParseResult(None, 'show name 2010.222', 1, [23], 'source.quality.etc', 'group'), 'show.name.2010.222.123.source.quality.etc-group': parser.ParseResult(None, 'show name 2010.222', 1, [23], 'source.quality.etc', 'group'),
'Show.Name.102': parser.ParseResult(None, 'Show Name', 1, [2]), 'Show.Name.102': parser.ParseResult(None, 'Show Name', 1, [2]),
'the.event.401.hdtv-lol': parser.ParseResult(None, 'the event', 4, [1], 'hdtv', 'lol'), 'the.event.401.hdtv-lol': parser.ParseResult(None, 'the event', 4, [1], 'hdtv', 'lol'),
'show.name.2010.special.hdtv-blah': None, # 'show.name.2010.special.hdtv-blah': None,
}, },
'stupid': { 'stupid': {
@ -110,6 +112,86 @@ simple_test_cases = {
'2010-11-23 - Ep Name': parser.ParseResult(None, extra_info='Ep Name', air_date=datetime.date(2010, 11, 23)), '2010-11-23 - Ep Name': parser.ParseResult(None, extra_info='Ep Name', air_date=datetime.date(2010, 11, 23)),
'Show.Name.2010.11.23.WEB-DL': parser.ParseResult(None, 'Show Name', None, [], 'WEB-DL', None, datetime.date(2010, 11, 23)), 'Show.Name.2010.11.23.WEB-DL': parser.ParseResult(None, 'Show Name', None, [], 'WEB-DL', None, datetime.date(2010, 11, 23)),
}, },
'anime_ultimate': {
'[Tsuki] Bleach - 301 [1280x720][61D1D4EE]': parser.ParseResult(None, 'Bleach', None, [], '1280x720', 'Tsuki', None, [301]),
'[Tsuki] Fairy Tail - 70 [1280x720][C4807111]': parser.ParseResult(None, 'Fairy Tail', None, [], '1280x720', 'Tsuki', None, [70]),
'[SGKK] Bleach 312v2 [720p MKV]': parser.ParseResult(None, 'Bleach', None, [], '720p MKV', 'SGKK', None, [312]),
'[BSS-Anon] Tengen Toppa Gurren Lagann - 22-23 [1280x720][h264][6039D9AF]': parser.ParseResult(None, 'Tengen Toppa Gurren Lagann', None, [], '1280x720', 'BSS-Anon', None, [22, 23]),
'[SJSUBS]_Naruto_Shippuden_-_02_[480p AAC]': parser.ParseResult(None, 'Naruto Shippuden', None, [], '480p AAC', 'SJSUBS', None, [2]),
'[SFW-Chihiro] Dance in the Vampire Bund - 12 [1920x1080 Blu-ray FLAC][2F6DBC66].mkv': parser.ParseResult(None, 'Dance in the Vampire Bund', None, [], '1920x1080 Blu-ray FLAC', 'SFW-Chihiro', None, [12]),
'[SHiN-gx] Hanasaku Iroha - 01 [1280x720 h.264 AAC][BDC36683]': parser.ParseResult(None, 'Hanasaku Iroha', None, [], '1280x720 h.264 AAC', 'SHiN-gx', None, [1]),
'[SFW-Chihiro] Dance in the Vampire Bund - 02 [1920x1080 Blu-ray FLAC][C1FA0A09]': parser.ParseResult(None, 'Dance in the Vampire Bund', None, [], '1920x1080 Blu-ray FLAC', 'SFW-Chihiro', None, [2]),
'[HorribleSubs] No. 6 - 11 [720p]': parser.ParseResult(None, 'No. 6', None, [], '720p', 'HorribleSubs', None, [11]),
'[HorribleSubs] D Gray-Man - 312 (480p) [F501C9BE]': parser.ParseResult(None, 'D Gray-Man', None, [], '480p', 'HorribleSubs', None, [312]),
'[SGKK] Tengen Toppa Gurren Lagann - 45-46 (720p h264) [F501C9BE]': parser.ParseResult(None, 'Tengen Toppa Gurren Lagann', None, [], '720p h264', 'SGKK', None, [45, 46]),
'[Stratos-Subs]_Infinite_Stratos_-_12_(1280x720_H.264_AAC)_[379759DB]': parser.ParseResult(None, 'Infinite Stratos', None, [], '1280x720_H.264_AAC', 'Stratos-Subs', None, [12]),
'[ShinBunBu-Subs] Bleach - 02-03 (CX 1280x720 x264 AAC)': parser.ParseResult(None, 'Bleach', None, [], 'CX 1280x720 x264 AAC', 'ShinBunBu-Subs', None, [02, 03]),
'[Doki] Hanasaku Iroha - 03 (848x480 h264 AAC) [CB1AA73B]': parser.ParseResult(None, 'Hanasaku Iroha', None, [], '848x480 h264 AAC', 'Doki', None, [03]),
'[UTW]_Fractal_-_01_[h264-720p][96D3F1BF]': parser.ParseResult(None, 'Fractal', None, [], 'h264-720p', 'UTW', None, [1]),
'[a-s]_inuyasha_-_028_rs2_[BFDDF9F2]': parser.ParseResult(None, 'inuyasha', None, [], 'BFDDF9F2', 'a-s', None, [28]),
'[HorribleSubs] Fairy Tail S2 - 37 [1080p]': parser.ParseResult(None,'Fairy Tail S2', None, [], '1080p', 'HorribleSubs', None, [37]),
'[HorribleSubs] Sword Art Online II - 23 [720p]': parser.ParseResult(None, 'Sword Art Online II', None, [], '720p', 'HorribleSubs', None, [23])
},
'anime_ep_name': {
'[TzaTziki]_One_Piece_279_Chopper_Man_1_[720p][8AE5F25D]': parser.ParseResult(None, 'One Piece', None, [], '720p', 'TzaTziki', None, [279]),
"[ACX]Wolf's_Rain_-_04_-_Scars_in_the_Wasteland_[octavarium]_[82B7E357]": parser.ParseResult(None, "Wolf's Rain", None, [], 'octavarium', 'ACX', None, [4]),
'[ACX]Black Lagoon - 02v2 - Mangrove Heaven [SaintDeath] [7481F875]': parser.ParseResult(None, 'Black Lagoon', None, [], 'SaintDeath', 'ACX', None, [2]),
},
"anime_standard_round": {
'[SGKK] Bleach - 312v2 (1280x720 h264 AAC) [F501C9BE]': parser.ParseResult(None, 'Bleach', None, [], '1280x720 h264 AAC', 'SGKK', None, [312]),
},
'anime_slash': {
'[SGKK] Bleach 312v1 [720p/MKV]': parser.ParseResult(None, 'Bleach', None, [], '720p', 'SGKK', None, [312]),
'[SGKK] Bleach 312 [480p/MKV]': parser.ParseResult(None, 'Bleach', None, [], '480p', 'SGKK', None, [312])
},
'anime_standard_codec': {
'[Ayako]_Infinite_Stratos_-_IS_-_07_[H264][720p][EB7838FC]': parser.ParseResult(None, 'Infinite Stratos', None, [], '720p', 'Ayako', None, [7]),
'[Ayako] Infinite Stratos - IS - 07v2 [H264][720p][44419534]': parser.ParseResult(None, 'Infinite Stratos', None, [], '720p', 'Ayako', None, [7]),
'[Ayako-Shikkaku] Oniichan no Koto Nanka Zenzen Suki Janain Dakara ne - 10 [LQ][h264][720p] [8853B21C]': parser.ParseResult(None, 'Oniichan no Koto Nanka Zenzen Suki Janain Dakara ne', None, [], '720p', 'Ayako-Shikkaku', None, [10]),
'[Tsuki] Fairy Tail - 72 [XviD][C4807111]': parser.ParseResult(None, 'Fairy Tail', None, [], 'C4807111', 'Tsuki', None, [72]),
'Bubblegum Crisis Tokyo 2040 - 25 [aX] [F4E2E558]': parser.ParseResult(None, 'Bubblegum Crisis Tokyo 2040', None, [], "aX", None, None, [25]),
},
'anime_and_normal': {
'Bleach - s02e03 - 012 - Name & Name': parser.ParseResult(None, 'Bleach', 2, [3], None, None, None, [12]),
'Bleach - s02e03e04 - 012-013 - Name & Name': parser.ParseResult(None, 'Bleach', 2, [3, 4], None, None, None, [12, 13]),
'Bleach - s16e03-04 - 313-314': parser.ParseResult(None, 'Bleach', 16, [3, 4], None, None, None, [313, 314]),
'Blue Submarine No. 6 s16e03e04 313-314': parser.ParseResult(None, 'Blue Submarine No. 6', 16, [3, 4], None, None, None, [313, 314]),
'Bleach.s16e03-04.313-314': parser.ParseResult(None, 'Bleach', 16, [3, 4], None, None, None, [313, 314]),
'.hack roots s01e01 001.mkv': parser.ParseResult(None, 'hack roots', 1, [1], None, None, None, [1]),
'.hack sign s01e01 001.mkv': parser.ParseResult(None, 'hack sign', 1, [1], None, None, None, [1])
},
'anime_and_normal_reverse': {
'Bleach - 012 - s02e03 - Name & Name': parser.ParseResult(None, 'Bleach', 2, [3], None, None, None, [12]),
'Blue Submarine No. 6 - 012-013 - s02e03e04 - Name & Name': parser.ParseResult(None, 'Blue Submarine No. 6', 2, [3, 4], None, None, None, [12, 13]),
'07-GHOST - 012-013 - s02e03e04 - Name & Name': parser.ParseResult(None, '07-GHOST', 2, [3, 4], None, None, None, [12, 13]),
'3x3 Eyes - 012-013 - s02e03-04 - Name & Name': parser.ParseResult(None, '3x3 Eyes', 2, [3, 4], None, None, None, [12, 13]),
},
'anime_and_normal_front': {
'165.Naruto Shippuuden.s08e014': parser.ParseResult(None, 'Naruto Shippuuden', 8, [14], None, None, None, [165]),
'165-166.Naruto Shippuuden.s08e014e015': parser.ParseResult(None, 'Naruto Shippuuden', 8, [14, 15], None, None, None, [165, 166]),
'165-166.07-GHOST.s08e014-015': parser.ParseResult(None, '07-GHOST', 8, [14, 15], None, None, None, [165, 166]),
'165-166.3x3 Eyes.S08E014E015': parser.ParseResult(None, '3x3 Eyes', 8, [14, 15], None, None, None, [165, 166]),
},
'anime_bare': {
'One Piece 102': parser.ParseResult(None, 'One Piece', None, [], None, None, None, [102]),
'bleach - 010': parser.ParseResult(None, 'bleach', None, [], None, None, None, [10]),
'Naruto Shippuden - 314v2': parser.ParseResult(None, 'Naruto Shippuden', None, [], None, None, None, [314]),
'Blue Submarine No. 6 104-105': parser.ParseResult(None, 'Blue Submarine No. 6', None, [], None, None, None, [104, 105]),
'Samurai X: Trust & Betrayal (OVA) 001-002': parser.ParseResult(None, 'Samurai X: Trust & Betrayal (OVA)', None, [], None, None, None, [1, 2]),
"[ACX]_Wolf's_Spirit_001.mkv": parser.ParseResult(None, "Wolf's Spirit", None, [], None, 'ACX', None, [1])
}
} }
combination_test_cases = [ combination_test_cases = [
@ -159,7 +241,6 @@ failure_cases = ['7sins-jfcs01e09-720p-bluray-x264']
class UnicodeTests(test.SickbeardTestDBCase): class UnicodeTests(test.SickbeardTestDBCase):
def _test_unicode(self, name, result): def _test_unicode(self, name, result):
np = parser.NameParser(True) np = parser.NameParser(True)
@ -175,8 +256,8 @@ class UnicodeTests(test.SickbeardTestDBCase):
for (name, result) in unicode_test_cases: for (name, result) in unicode_test_cases:
self._test_unicode(name, result) self._test_unicode(name, result)
class FailureCaseTests(test.SickbeardTestDBCase):
class FailureCaseTests(test.SickbeardTestDBCase):
def _test_name(self, name): def _test_name(self, name):
np = parser.NameParser(True) np = parser.NameParser(True)
try: try:
@ -192,8 +273,8 @@ class FailureCaseTests(test.SickbeardTestDBCase):
for name in failure_cases: for name in failure_cases:
self.assertTrue(self._test_name(name)) self.assertTrue(self._test_name(name))
class ComboTests(test.SickbeardTestDBCase):
class ComboTests(test.SickbeardTestDBCase):
def _test_combo(self, name, result, which_regexes): def _test_combo(self, name, result, which_regexes):
if VERBOSE: if VERBOSE:
@ -211,7 +292,6 @@ class ComboTests(test.SickbeardTestDBCase):
print test_result, test_result.which_regex print test_result, test_result.which_regex
print result, which_regexes print result, which_regexes
self.assertEqual(test_result, result) self.assertEqual(test_result, result)
for cur_regex in which_regexes: for cur_regex in which_regexes:
self.assertTrue(cur_regex in test_result.which_regex) self.assertTrue(cur_regex in test_result.which_regex)
@ -224,8 +304,8 @@ class ComboTests(test.SickbeardTestDBCase):
# paths when test is run on Windows. # paths when test is run on Windows.
self._test_combo(os.path.normpath(name), result, which_regexes) self._test_combo(os.path.normpath(name), result, which_regexes)
class BasicTests(test.SickbeardTestDBCase):
class BasicTests(test.SickbeardTestDBCase):
def _test_names(self, np, section, transform=None, verbose=False): def _test_names(self, np, section, transform=None, verbose=False):
if VERBOSE or verbose: if VERBOSE or verbose:
@ -246,105 +326,146 @@ class BasicTests(test.SickbeardTestDBCase):
else: else:
test_result = np.parse(cur_test) test_result = np.parse(cur_test)
if DEBUG or verbose: try:
# self.assertEqual(test_result.which_regex, [section])
self.assertEqual(test_result, result)
except:
print 'air_by_date:', test_result.is_air_by_date, 'air_date:', test_result.air_date print 'air_by_date:', test_result.is_air_by_date, 'air_date:', test_result.air_date
print 'anime:', test_result.is_anime, 'ab_episode_numbers:', test_result.ab_episode_numbers print 'anime:', test_result.is_anime, 'ab_episode_numbers:', test_result.ab_episode_numbers
print test_result print test_result
print result print result
self.assertEqual(test_result.which_regex, [section]) raise
self.assertEqual(test_result, result)
#def test_standard_names(self):
# np = parser.NameParser(False) def test_standard_names(self):
# self._test_names(np, 'standard') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'standard')
#def test_standard_repeat_names(self):
# np = parser.NameParser(False) def test_standard_repeat_names(self):
# self._test_names(np, 'standard_repeat') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'standard_repeat')
#def test_fov_names(self):
# np = parser.NameParser(False) def test_fov_names(self):
# self._test_names(np, 'fov') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'fov')
#def test_fov_repeat_names(self):
# np = parser.NameParser(False) def test_fov_repeat_names(self):
# self._test_names(np, 'fov_repeat') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'fov_repeat')
#def test_bare_names(self):
# np = parser.NameParser(False) def test_bare_names(self):
# self._test_names(np, 'bare') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'bare')
#def test_stupid_names(self):
# np = parser.NameParser(False) def test_stupid_names(self):
# self._test_names(np, 'stupid') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'stupid')
#def test_no_season_names(self):
# np = parser.NameParser(False) def test_no_season_names(self):
# self._test_names(np, 'no_season') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'no_season')
#def test_no_season_general_names(self):
# np = parser.NameParser(False) def test_no_season_general_names(self):
# self._test_names(np, 'no_season_general') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'no_season_general')
#def test_no_season_multi_ep_names(self):
# np = parser.NameParser(False) def test_no_season_multi_ep_names(self):
# self._test_names(np, 'no_season_multi_ep') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'no_season_multi_ep')
#def test_season_only_names(self):
# np = parser.NameParser(False) def test_season_only_names(self):
# self._test_names(np, 'season_only') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'season_only')
#def test_scene_date_format_names(self):
# np = parser.NameParser(False) def test_scene_date_format_names(self):
# self._test_names(np, 'scene_date_format') np = parser.NameParser(False, testing=True)
# self._test_names(np, 'scene_date_format')
#def test_standard_file_names(self):
# np = parser.NameParser() def test_standard_file_names(self):
# self._test_names(np, 'standard', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'standard', lambda x: x + '.avi')
#def test_standard_repeat_file_names(self):
# np = parser.NameParser() def test_standard_repeat_file_names(self):
# self._test_names(np, 'standard_repeat', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'standard_repeat', lambda x: x + '.avi')
#def test_fov_file_names(self):
# np = parser.NameParser() def test_fov_file_names(self):
# self._test_names(np, 'fov', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'fov', lambda x: x + '.avi')
#def test_fov_repeat_file_names(self):
# np = parser.NameParser() def test_fov_repeat_file_names(self):
# self._test_names(np, 'fov_repeat', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'fov_repeat', lambda x: x + '.avi')
#def test_bare_file_names(self):
# np = parser.NameParser() def test_bare_file_names(self):
# self._test_names(np, 'bare', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'bare', lambda x: x + '.avi')
#def test_stupid_file_names(self):
# np = parser.NameParser() def test_stupid_file_names(self):
# self._test_names(np, 'stupid', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'stupid', lambda x: x + '.avi')
#def test_no_season_file_names(self):
# np = parser.NameParser() def test_no_season_file_names(self):
# self._test_names(np, 'no_season', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'no_season', lambda x: x + '.avi')
#def test_no_season_general_file_names(self):
# np = parser.NameParser() def test_no_season_general_file_names(self):
# self._test_names(np, 'no_season_general', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'no_season_general', lambda x: x + '.avi')
#def test_no_season_multi_ep_file_names(self):
# np = parser.NameParser() def test_no_season_multi_ep_file_names(self):
# self._test_names(np, 'no_season_multi_ep', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'no_season_multi_ep', lambda x: x + '.avi')
#def test_season_only_file_names(self):
# np = parser.NameParser() def test_season_only_file_names(self):
# self._test_names(np, 'season_only', lambda x: x + '.avi') np = parser.NameParser(testing=True)
# self._test_names(np, 'season_only', lambda x: x + '.avi')
#def test_scene_date_format_file_names(self):
# np = parser.NameParser() def test_scene_date_format_file_names(self):
# self._test_names(np, 'scene_date_format', lambda x: x + '.avi') np = parser.NameParser(testing=True)
self._test_names(np, 'scene_date_format', lambda x: x + '.avi')
def test_combination_names(self): def test_combination_names(self):
pass pass
def test_anime_ultimate(self):
np = parser.NameParser(False, TVShow(is_anime=True), testing=True)
self._test_names(np, 'anime_ultimate')
def test_anime_ep_name(self):
np = parser.NameParser(False, TVShow(is_anime=True), testing=True)
self._test_names(np, 'anime_ep_name')
def test_anime_slash(self):
np = parser.NameParser(False, TVShow(is_anime=True), testing=True)
self._test_names(np, 'anime_slash')
def test_anime_codec(self):
np = parser.NameParser(False, TVShow(is_anime=True), testing=True)
self._test_names(np, 'anime_standard_codec')
def test_anime_and_normal(self):
np = parser.NameParser(False, TVShow(is_anime=True), testing=True)
self._test_names(np, 'anime_and_normal')
def test_anime_and_normal_reverse(self):
np = parser.NameParser(False, TVShow(is_anime=True), testing=True)
self._test_names(np, 'anime_and_normal_reverse')
def test_anime_and_normal_front(self):
np = parser.NameParser(False, TVShow(is_anime=True), testing=True)
self._test_names(np, 'anime_and_normal_front')
def test_anime_bare(self):
np = parser.NameParser(False, TVShow(is_anime=True), testing=True)
self._test_names(np, 'anime_bare')
class TVShow(object):
def __init__(self, is_anime=False):
self.is_anime = is_anime
if __name__ == '__main__': if __name__ == '__main__':
if len(sys.argv) > 1: if len(sys.argv) > 1:
suite = unittest.TestLoader().loadTestsFromName('name_parser_tests.BasicTests.test_' + sys.argv[1]) suite = unittest.TestLoader().loadTestsFromName('name_parser_tests.BasicTests.test_' + sys.argv[1])