Skip to content

Commit

Permalink
Update language equals mechanism
Browse files Browse the repository at this point in the history
This commit will avoid the core requesting languages that are not
supported by providers.
  • Loading branch information
vitiko98 committed Jun 14, 2023
1 parent 6ac5bc7 commit 2b0e435
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions libs/subliminal_patch/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -330,6 +330,8 @@ def list_subtitles_provider(self, provider, video, languages):
:rtype: list of :class:`~subliminal.subtitle.Subtitle` or None
"""
logger.debug("Languages requested: %s", languages)

if self.language_hook:
languages_search_base = self.language_hook(provider)
else:
Expand All @@ -354,10 +356,14 @@ def list_subtitles_provider(self, provider, video, languages):
return []

# list subtitles
logger.info('Listing subtitles with provider %r and languages %r', provider, provider_languages)
results = []

to_request = self.lang_equals.translate(provider_languages) & set(provider_registry[provider].languages)

logger.info('Listing subtitles with provider %r and languages %r', provider, to_request)

try:
results = self[provider].list_subtitles(video, self.lang_equals.translate(provider_languages))
results = self[provider].list_subtitles(video, to_request)
seen = []
out = []
for s in results:
Expand Down

0 comments on commit 2b0e435

Please sign in to comment.