diff options
-rw-r--r-- | libs/subliminal_patch/providers/embeddedsubtitles.py | 5 | ||||
-rw-r--r-- | libs/subliminal_patch/providers/utils.py | 22 |
2 files changed, 24 insertions, 3 deletions
diff --git a/libs/subliminal_patch/providers/embeddedsubtitles.py b/libs/subliminal_patch/providers/embeddedsubtitles.py index 828648432..b1d7b12f9 100644 --- a/libs/subliminal_patch/providers/embeddedsubtitles.py +++ b/libs/subliminal_patch/providers/embeddedsubtitles.py @@ -1,8 +1,8 @@ # -*- coding: utf-8 -*- import functools -import logging import hashlib +import logging import os import re import shutil @@ -14,10 +14,12 @@ from fese import container from fese import FFprobeSubtitleStream from fese import FFprobeVideoContainer from fese import tags +from fese.exceptions import ExtractionError from fese.exceptions import InvalidSource from subliminal_patch.core import Episode from subliminal_patch.core import Movie from subliminal_patch.providers import Provider +from subliminal_patch.providers.utils import blacklist_on from subliminal_patch.subtitle import Subtitle from subzero.language import Language @@ -185,6 +187,7 @@ class EmbeddedSubtitlesProvider(Provider): "series" if isinstance(video, Episode) else "movie", ) + @blacklist_on(ExtractionError) def download_subtitle(self, subtitle: EmbeddedSubtitle): path = self._get_subtitle_path(subtitle) diff --git a/libs/subliminal_patch/providers/utils.py b/libs/subliminal_patch/providers/utils.py index f06f75e34..032e7eebf 100644 --- a/libs/subliminal_patch/providers/utils.py +++ b/libs/subliminal_patch/providers/utils.py @@ -12,6 +12,7 @@ from guessit import guessit import pysubs2 import rarfile from subliminal.subtitle import fix_line_ending +from subliminal_patch.exceptions import MustGetBlacklisted from subliminal_patch.core import Episode from subliminal_patch.subtitle import guess_matches @@ -23,6 +24,22 @@ logger = logging.getLogger(__name__) _MatchingSub = namedtuple("_MatchingSub", ("file", "priority", "context")) +def blacklist_on(*exc_types): + "Raise MustGetBlacklisted if any of the exc_types are raised." + + def decorator(method): + def wrapper(self, subtitle): + try: + return method(self, subtitle) + except exc_types: + logger.error("Sending blacklist exception", exc_info=True) + raise MustGetBlacklisted(subtitle.id, subtitle.media_type) + + return wrapper + + return decorator + + def _get_matching_sub( sub_names, forced=False, episode=None, episode_title=None, **kwargs ): @@ -169,11 +186,12 @@ def update_matches( video, release_info: Union[str, Iterable[str]], split="\n", - **guessit_options + **guessit_options, ): """Update matches set from release info string or Iterable. - Use the split parameter to iterate over the set delimiter; set None to avoid split.""" + Use the split parameter to iterate over the set delimiter; set None to avoid split. + """ guessit_options["type"] = "episode" if isinstance(video, Episode) else "movie" |