mirror
/
bazarr
mirror of https://github.com/morpheus65535/bazarr
1
0
Fork 0
Code Issues Releases Wiki Activity

Merge remote-tracking branch 'origin/development' into development

This commit is contained in:
morpheus65535 2023-05-29 06:39:15 -04:00
parent 2346f3ed58 4725496313
commit bbdb61a514
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
libs/subliminal_patch/core.py
View File

@ -155,6 +155,19 @@ class _LanguageEquals(list):
if len(item) != 2 or not any(isinstance(i, Language) for i in item):
raise ValueError(f"Not a valid equal tuple: {item}")
def translate(self, items: set):
translated = set()
for equals in self:
from_, to_ = equals
if to_ in items:
logger.debug("Translating %s -> %s", to_, from_)
translated.add(from_)
else:
translated.add(to_)
return translated or items
def check_set(self, items: set):
""" Check a set of languages. For example, if the set is {Language('es')} and one of the
equals of the instance is (Language('es'), Language('es', 'MX')), the set will now have
@ -168,12 +181,12 @@ class _LanguageEquals(list):
for equals in self:
from_, to_ = equals
if from_ in items:
logger.debug("Adding %s to %s", to_, items)
logger.debug("Adding %s to %s item(s) set", to_, len(items))
to_add.append(to_)
new_items = items.copy()
new_items.update(to_add)
logger.debug("New set: %s", new_items)
logger.debug("New set: %s items", len(new_items))
return new_items
def update_subtitle(self, subtitle):
@ -343,7 +356,7 @@ class SZProviderPool(ProviderPool):
logger.info('Listing subtitles with provider %r and languages %r', provider, provider_languages)
results = []
try:
results = self[provider].list_subtitles(video, provider_languages)
results = self[provider].list_subtitles(video, self.lang_equals.translate(provider_languages))
seen = []
out = []
for s in results: