github-mirrors/bazarr
1
0
Fork 0
mirror of https://github.com/morpheus65535/bazarr.git synced 2025-04-23 14:17:46 -04:00
Code Issues Projects Releases Packages Wiki Activity

Language-equals: add compatibility for more providers

Browse source 
Some providers directly uses language codes within their query
implementations. This 'translator' method will take care of that.
This commit is contained in:
Vitiko 2023-05-28 22:06:43 -04:00
parent 77531090b2
commit 4725496313
1 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

15
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
@ -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: