Embedded Subtitles Provider: handle KeyError for cached paths

10 months ago · cdf7296dd4
parent deef13d11b
commit cdf7296dd4
1 changed files with 5 additions and 1 deletions
--- a/libs/subliminal_patch/providers/embeddedsubtitles.py
+++ b/libs/subliminal_patch/providers/embeddedsubtitles.py
@ -190,7 +190,11 @@ class EmbeddedSubtitlesProvider(Provider):

    @blacklist_on(ExtractionError)
    def download_subtitle(self, subtitle: EmbeddedSubtitle):
-        path = self._get_subtitle_path(subtitle)
+        try:
+            path = self._get_subtitle_path(subtitle)
+        except KeyError: # TODO: add MustGetBlacklisted support
+            logger.error("Couldn't get subtitle path")
+            return None

        modifiers = _type_modifiers.get(subtitle.stream.codec_name) or set()
        logger.debug("Found modifiers for %s type: %s", subtitle.stream, modifiers)