mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-23 16:28:49 -05:00
269 lines
8.7 KiB
Python
269 lines
8.7 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of SickRage.
|
|
#
|
|
# SickRage is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# SickRage is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with SickRage. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import datetime
|
|
import os
|
|
|
|
import sickbeard
|
|
from sickbeard import encodingKludge as ek
|
|
from sickbeard import tv
|
|
from sickbeard import common
|
|
from sickbeard import logger
|
|
from sickbeard.name_parser.parser import NameParser, InvalidNameException
|
|
|
|
from common import Quality, DOWNLOADED
|
|
|
|
name_presets = ('%SN - %Sx%0E - %EN',
|
|
'%S.N.S%0SE%0E.%E.N',
|
|
'%Sx%0E - %EN',
|
|
'S%0SE%0E - %EN',
|
|
'Season %0S/%S.N.S%0SE%0E.%Q.N-%RG'
|
|
)
|
|
|
|
name_anime_presets = name_presets
|
|
|
|
name_abd_presets = ('%SN - %A-D - %EN',
|
|
'%S.N.%A.D.%E.N.%Q.N',
|
|
'%Y/%0M/%S.N.%A.D.%E.N-%RG'
|
|
)
|
|
|
|
name_sports_presets = ('%SN - %A-D - %EN',
|
|
'%S.N.%A.D.%E.N.%Q.N',
|
|
'%Y/%0M/%S.N.%A.D.%E.N-%RG'
|
|
)
|
|
|
|
class TVShow():
|
|
def __init__(self):
|
|
self.name = "Show Name"
|
|
self.genre = "Comedy"
|
|
self.indexerid = 00001
|
|
self.air_by_date = 0
|
|
self.sports = 0
|
|
self.anime = 0
|
|
self.scene = 0
|
|
|
|
def _is_anime(self):
|
|
if (self.anime > 0):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
is_anime = property(_is_anime)
|
|
|
|
def _is_sports(self):
|
|
if (self.sports > 0):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
is_sports = property(_is_sports)
|
|
|
|
def _is_scene(self):
|
|
if (self.scene > 0):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
is_scene = property(_is_scene)
|
|
|
|
|
|
class TVEpisode(tv.TVEpisode):
|
|
def __init__(self, season, episode, absolute_number, name):
|
|
self.relatedEps = []
|
|
self._name = name
|
|
self._season = season
|
|
self._episode = episode
|
|
self._absolute_number = absolute_number
|
|
self.scene_season = season
|
|
self.scene_episode = episode
|
|
self.scene_absolute_number = absolute_number
|
|
self._airdate = datetime.date(2010, 3, 9)
|
|
self.show = TVShow()
|
|
self._status = Quality.compositeStatus(common.DOWNLOADED, common.Quality.SDTV)
|
|
self._release_name = 'Show.Name.S02E03.HDTV.XviD-RLSGROUP'
|
|
self._is_proper = True
|
|
|
|
|
|
def check_force_season_folders(pattern=None, multi=None, anime_type=None):
|
|
"""
|
|
Checks if the name can still be parsed if you strip off the folders to determine if we need to force season folders
|
|
to be enabled or not.
|
|
|
|
Returns true if season folders need to be forced on or false otherwise.
|
|
"""
|
|
if pattern == None:
|
|
pattern = sickbeard.NAMING_PATTERN
|
|
|
|
if anime_type == None:
|
|
anime_type = sickbeard.NAMING_ANIME
|
|
|
|
valid = not validate_name(pattern, None, anime_type, file_only=True)
|
|
|
|
if multi != None:
|
|
valid = valid or not validate_name(pattern, multi, anime_type, file_only=True)
|
|
|
|
return valid
|
|
|
|
|
|
def check_valid_naming(pattern=None, multi=None, anime_type=None):
|
|
"""
|
|
Checks if the name is can be parsed back to its original form for both single and multi episodes.
|
|
|
|
Returns true if the naming is valid, false if not.
|
|
"""
|
|
if pattern == None:
|
|
pattern = sickbeard.NAMING_PATTERN
|
|
|
|
if anime_type == None:
|
|
anime_type = sickbeard.NAMING_ANIME
|
|
|
|
logger.log(u"Checking whether the pattern " + pattern + " is valid for a single episode", logger.DEBUG)
|
|
valid = validate_name(pattern, None, anime_type)
|
|
|
|
if multi != None:
|
|
logger.log(u"Checking whether the pattern " + pattern + " is valid for a multi episode", logger.DEBUG)
|
|
valid = valid and validate_name(pattern, multi, anime_type)
|
|
|
|
return valid
|
|
|
|
|
|
def check_valid_abd_naming(pattern=None):
|
|
"""
|
|
Checks if the name is can be parsed back to its original form for an air-by-date format.
|
|
|
|
Returns true if the naming is valid, false if not.
|
|
"""
|
|
if pattern == None:
|
|
pattern = sickbeard.NAMING_PATTERN
|
|
|
|
logger.log(u"Checking whether the pattern " + pattern + " is valid for an air-by-date episode", logger.DEBUG)
|
|
valid = validate_name(pattern, abd=True)
|
|
|
|
return valid
|
|
|
|
def check_valid_sports_naming(pattern=None):
|
|
"""
|
|
Checks if the name is can be parsed back to its original form for an sports format.
|
|
|
|
Returns true if the naming is valid, false if not.
|
|
"""
|
|
if pattern == None:
|
|
pattern = sickbeard.NAMING_PATTERN
|
|
|
|
logger.log(u"Checking whether the pattern " + pattern + " is valid for an sports episode", logger.DEBUG)
|
|
valid = validate_name(pattern, sports=True)
|
|
|
|
return valid
|
|
|
|
def validate_name(pattern, multi=None, anime_type=None, file_only=False, abd=False, sports=False):
|
|
ep = generate_sample_ep(multi, abd, sports, anime_type)
|
|
|
|
new_name = ep.formatted_filename(pattern, multi, anime_type) + '.ext'
|
|
new_path = ep.formatted_dir(pattern, multi)
|
|
if not file_only:
|
|
new_name = ek.ek(os.path.join, new_path, new_name)
|
|
|
|
if not new_name:
|
|
logger.log(u"Unable to create a name out of " + pattern, logger.DEBUG)
|
|
return False
|
|
|
|
logger.log(u"Trying to parse " + new_name, logger.DEBUG)
|
|
|
|
parser = NameParser(True, showObj=ep.show, naming_pattern=True)
|
|
|
|
try:
|
|
result = parser.parse(new_name)
|
|
except Exception, e:
|
|
logger.log(u"Unable to parse " + new_name + ", not valid", logger.DEBUG)
|
|
return False
|
|
|
|
logger.log("The name " + new_name + " parsed into " + str(result), logger.DEBUG)
|
|
|
|
if abd or sports:
|
|
if result.air_date != ep.airdate:
|
|
logger.log(u"Air date incorrect in parsed episode, pattern isn't valid", logger.DEBUG)
|
|
return False
|
|
elif anime_type != 3:
|
|
if len(result.ab_episode_numbers) and result.ab_episode_numbers != [x.absolute_number for x in [ep] + ep.relatedEps]:
|
|
logger.log(u"Absolute numbering incorrect in parsed episode, pattern isn't valid", logger.DEBUG)
|
|
return False
|
|
else:
|
|
if result.season_number != ep.season:
|
|
logger.log(u"Season number incorrect in parsed episode, pattern isn't valid", logger.DEBUG)
|
|
return False
|
|
if result.episode_numbers != [x.episode for x in [ep] + ep.relatedEps]:
|
|
logger.log(u"Episode numbering incorrect in parsed episode, pattern isn't valid", logger.DEBUG)
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def generate_sample_ep(multi=None, abd=False, sports=False, anime_type=None):
|
|
# make a fake episode object
|
|
ep = TVEpisode(2, 3, 3, "Ep Name")
|
|
|
|
ep._status = Quality.compositeStatus(DOWNLOADED, Quality.HDTV)
|
|
ep._airdate = datetime.date(2011, 3, 9)
|
|
|
|
if abd:
|
|
ep._release_name = 'Show.Name.2011.03.09.HDTV.XviD-RLSGROUP'
|
|
ep.show.air_by_date = 1
|
|
elif sports:
|
|
ep._release_name = 'Show.Name.2011.03.09.HDTV.XviD-RLSGROUP'
|
|
ep.show.sports = 1
|
|
else:
|
|
if anime_type != 3:
|
|
ep.show.anime = 1
|
|
ep._release_name = 'Show.Name.003.HDTV.XviD-RLSGROUP'
|
|
else:
|
|
ep._release_name = 'Show.Name.S02E03.HDTV.XviD-RLSGROUP'
|
|
|
|
if multi != None:
|
|
ep._name = "Ep Name (1)"
|
|
|
|
if anime_type != 3:
|
|
ep.show.anime = 1
|
|
|
|
ep._release_name = 'Show.Name.003-004.HDTV.XviD-RLSGROUP'
|
|
|
|
secondEp = TVEpisode(2, 4, 4, "Ep Name (2)")
|
|
secondEp._status = Quality.compositeStatus(DOWNLOADED, Quality.HDTV)
|
|
secondEp._release_name = ep._release_name
|
|
|
|
ep.relatedEps.append(secondEp)
|
|
else:
|
|
ep._release_name = 'Show.Name.S02E03E04E05.HDTV.XviD-RLSGROUP'
|
|
|
|
secondEp = TVEpisode(2, 4, 4, "Ep Name (2)")
|
|
secondEp._status = Quality.compositeStatus(DOWNLOADED, Quality.HDTV)
|
|
secondEp._release_name = ep._release_name
|
|
|
|
thirdEp = TVEpisode(2, 5, 5, "Ep Name (3)")
|
|
thirdEp._status = Quality.compositeStatus(DOWNLOADED, Quality.HDTV)
|
|
thirdEp._release_name = ep._release_name
|
|
|
|
ep.relatedEps.append(secondEp)
|
|
ep.relatedEps.append(thirdEp)
|
|
|
|
return ep
|
|
|
|
|
|
def test_name(pattern, multi=None, abd=False, sports=False, anime_type=None):
|
|
ep = generate_sample_ep(multi, abd, sports, anime_type)
|
|
|
|
return {'name': ep.formatted_filename(pattern, multi, anime_type), 'dir': ep.formatted_dir(pattern, multi)} |