bazarr/custom_libs/subzero/score_range.py

# coding=utf-8

from __future__ import absolute_import
from __future__ import print_function
import sys
from itertools import chain, combinations, permutations

from subliminal.video import Episode


def permute(x):
    return [base_score + i + j for i in x for j in x]

if __name__ == "__main__":
    scores = Episode.scores
    base_score_keys = ["series", "season", "episode"]
    leftover_keys = list(set(scores.keys()) - set(base_score_keys))
    base_score = sum([val for key, val in scores.items() if key in base_score_keys])
    leftover_scores = set([score for key, score in scores.items() if key in leftover_keys])
    print("base score:", base_score)
    print("leftover:", sorted(set(leftover_scores)))
    # print sum_possible_scores(base_score, leftover_scores)
    # print list(permutations(leftover_scores))
    print(",\n".join(['"%s"' % x for x in ["66"] + sorted(set(permute(leftover_scores)))]))