61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
# coding=utf-8
|
|
|
|
import logging
|
|
from subliminal.video import Episode, Movie
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def compute_score(matches, video, scores=None):
|
|
"""Compute the score of the `matches` against the `video`.
|
|
Some matches count as much as a combination of others in order to level the final score:
|
|
* `hash` removes everything else
|
|
* For :class:`~subliminal.video.Episode`
|
|
* `imdb_id` removes `series`, `tvdb_id`, `season`, `episode`, `title` and `year`
|
|
* `tvdb_id` removes `series` and `year`
|
|
* `title` removes `season` and `episode`
|
|
:param video: the video to get the score with.
|
|
:type video: :class:`~subliminal.video.Video`
|
|
:param dict scores: scores to use, if `None`, the :attr:`~subliminal.video.Video.scores` from the video are used.
|
|
:return: score of the subtitle.
|
|
:rtype: int
|
|
|
|
# patch: remove score cap for enabling individual boost
|
|
"""
|
|
final_matches = matches.copy()
|
|
scores = scores or video.scores
|
|
|
|
logger.info('Computing score for matches %r and %r', matches, video)
|
|
|
|
is_episode = isinstance(video, Episode)
|
|
|
|
episode_hash_valid_if = {"series", "season", "episode", "format"}
|
|
movie_hash_valid_if = {"title", "format", "video_codec"}
|
|
|
|
# remove equivalent match combinations
|
|
if 'hash' in final_matches:
|
|
# hash is error-prone, try to fix that
|
|
hash_valid_if = episode_hash_valid_if if is_episode else movie_hash_valid_if
|
|
|
|
if hash_valid_if <= set(final_matches):
|
|
# series, season and episode matched, hash is valid
|
|
logger.debug('Using valid hash, as %s are correct (%r) and (%r)', hash_valid_if, matches, video)
|
|
final_matches &= {'hash', 'hearing_impaired'}
|
|
else:
|
|
# no match, invalidate hash
|
|
logger.debug('Ignoring hash as other matches are wrong (missing: %r) and (%r)', hash_valid_if - matches, video)
|
|
final_matches -= {"hash"}
|
|
|
|
elif is_episode:
|
|
if 'imdb_id' in final_matches:
|
|
final_matches -= {'series', 'tvdb_id', 'season', 'episode', 'title', 'year'}
|
|
if 'tvdb_id' in final_matches:
|
|
final_matches -= {'series', 'year'}
|
|
|
|
# compute score
|
|
logger.debug('Final matches: %r', final_matches)
|
|
score = sum((scores[match] for match in final_matches))
|
|
logger.info('Computed score %d', score)
|
|
|
|
return score
|