bazarr/libs/subzero/score_range.py

25 lines
916 B
Python

# 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)))]))