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.
87 lines
3.4 KiB
87 lines
3.4 KiB
12 years ago
|
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
|
||
|
|
|
||
|
|
log = CPLog(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class ShowMatcher(MatcherBase):
|
||
|
|
|
||
|
|
type = ['show', 'season', 'episode']
|
||
|
|
|
||
|
|
# 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(ShowMatcher, self).__init__()
|
||
|
|
|
||
|
|
for type in toIterable(self.type):
|
||
|
|
addEvent('%s.matcher.correct' % type, self.correct)
|
||
|
|
addEvent('%s.matcher.correct_identifier' % 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('show.matcher.correct_identifier', 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):
|
||
|
|
required_id = fireEvent('library.identifier', media['library'], single = True)
|
||
|
|
|
||
|
|
if 'identifier' not in chain.info:
|
||
|
|
return False
|
||
|
|
|
||
|
|
# TODO could be handled better?
|
||
|
|
if len(chain.info['identifier']) != 1:
|
||
|
|
return False
|
||
|
|
identifier = chain.info['identifier'][0]
|
||
|
|
|
||
|
|
# TODO air by date episodes
|
||
|
|
|
||
|
|
# TODO this should support identifiers with characters 'a', 'b', etc..
|
||
|
|
for k, v in identifier.items():
|
||
|
|
identifier[k] = tryInt(v, None)
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
# 'episode' is required in identifier for subset matching
|
||
|
|
if 'episode' not in identifier:
|
||
|
|
identifier['episode'] = None
|
||
|
|
|
||
|
|
if not dictIsSubset(required_id, identifier):
|
||
|
|
log.info2('Wrong: required identifier %s does not match release identifier %s', (str(required_id), str(identifier)))
|
||
|
|
return False
|
||
|
|
|
||
|
|
return True
|