You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
4.7 KiB
127 lines
4.7 KiB
from couchpotato import CPLog
|
|
from couchpotato.core.event import addEvent, fireEvent
|
|
from couchpotato.core.helpers.variable import dictIsSubset, tryInt, toIterable
|
|
from couchpotato.core.media._base.matcher.base import MatcherBase
|
|
from couchpotato.core.providers.base import MultiProvider
|
|
|
|
log = CPLog(__name__)
|
|
|
|
|
|
class ShowMatcher(MultiProvider):
|
|
|
|
def getTypes(self):
|
|
return [Season, Episode]
|
|
|
|
|
|
class Base(MatcherBase):
|
|
# TODO come back to this later, think this could be handled better, this is starting to get out of hand....
|
|
quality_map = {
|
|
'bluray_1080p': {'resolution': ['1080p'], 'source': ['bluray']},
|
|
'bluray_720p': {'resolution': ['720p'], 'source': ['bluray']},
|
|
|
|
'bdrip_1080p': {'resolution': ['1080p'], 'source': ['BDRip']},
|
|
'bdrip_720p': {'resolution': ['720p'], 'source': ['BDRip']},
|
|
|
|
'brrip_1080p': {'resolution': ['1080p'], 'source': ['BRRip']},
|
|
'brrip_720p': {'resolution': ['720p'], 'source': ['BRRip']},
|
|
|
|
'webdl_1080p': {'resolution': ['1080p'], 'source': ['webdl', ['web', 'dl']]},
|
|
'webdl_720p': {'resolution': ['720p'], 'source': ['webdl', ['web', 'dl']]},
|
|
'webdl_480p': {'resolution': ['480p'], 'source': ['webdl', ['web', 'dl']]},
|
|
|
|
'hdtv_720p': {'resolution': ['720p'], 'source': ['hdtv']},
|
|
'hdtv_sd': {'resolution': ['480p', None], 'source': ['hdtv']},
|
|
}
|
|
|
|
def __init__(self):
|
|
super(Base, self).__init__()
|
|
|
|
addEvent('%s.matcher.correct_identifier' % self.type, self.correctIdentifier)
|
|
|
|
def correct(self, chain, release, media, quality):
|
|
log.info("Checking if '%s' is valid", release['name'])
|
|
log.info2('Release parsed as: %s', chain.info)
|
|
|
|
if not fireEvent('matcher.correct_quality', chain, quality, self.quality_map, single = True):
|
|
log.info('Wrong: %s, quality does not match', release['name'])
|
|
return False
|
|
|
|
if not fireEvent('%s.matcher.correct_identifier' % self.type, chain, media):
|
|
log.info('Wrong: %s, identifier does not match', release['name'])
|
|
return False
|
|
|
|
if not fireEvent('matcher.correct_title', chain, media):
|
|
log.info("Wrong: '%s', undetermined naming.", (' '.join(chain.info['show_name'])))
|
|
return False
|
|
|
|
return True
|
|
|
|
def correctIdentifier(self, chain, media):
|
|
raise NotImplementedError()
|
|
|
|
def getChainIdentifier(self, chain):
|
|
if 'identifier' not in chain.info:
|
|
return None
|
|
|
|
identifier = self.flattenInfo(chain.info['identifier'])
|
|
|
|
# Try cast values to integers
|
|
for key, value in identifier.items():
|
|
if isinstance(value, list):
|
|
if len(value) <= 1:
|
|
value = value[0]
|
|
else:
|
|
log.warning('Wrong: identifier contains multiple season or episode values, unsupported')
|
|
return None
|
|
|
|
identifier[key] = tryInt(value, value)
|
|
|
|
return identifier
|
|
|
|
|
|
class Episode(Base):
|
|
type = 'episode'
|
|
|
|
def correctIdentifier(self, chain, media):
|
|
identifier = self.getChainIdentifier(chain)
|
|
if not identifier:
|
|
log.info2('Wrong: release identifier is not valid (unsupported or missing identifier)')
|
|
return False
|
|
|
|
# TODO - Parse episode ranges from identifier to determine if they are multi-part episodes
|
|
if any([x in identifier for x in ['episode_from', 'episode_to']]):
|
|
log.info2('Wrong: releases with identifier ranges are not supported yet')
|
|
return False
|
|
|
|
required = fireEvent('library.identifier', media['library'], single = True)
|
|
|
|
# TODO - Support air by date episodes
|
|
# TODO - Support episode parts
|
|
|
|
if identifier != required:
|
|
log.info2('Wrong: required identifier (%s) does not match release identifier (%s)', (required, identifier))
|
|
return False
|
|
|
|
return True
|
|
|
|
class Season(Base):
|
|
type = 'season'
|
|
|
|
def correctIdentifier(self, chain, media):
|
|
identifier = self.getChainIdentifier(chain)
|
|
if not identifier:
|
|
log.info2('Wrong: release identifier is not valid (unsupported or missing identifier)')
|
|
return False
|
|
|
|
# TODO - Parse episode ranges from identifier to determine if they are season packs
|
|
if any([x in identifier for x in ['episode_from', 'episode_to']]):
|
|
log.info2('Wrong: releases with identifier ranges are not supported yet')
|
|
return False
|
|
|
|
required = fireEvent('library.identifier', media['library'], single = True)
|
|
|
|
if identifier != required:
|
|
log.info2('Wrong: required identifier (%s) does not match release identifier (%s)', (required, identifier))
|
|
return False
|
|
|
|
return True
|
|
|