From 9be61914784d16668ebc6289e3d965ece1e0a578 Mon Sep 17 00:00:00 2001 From: josdion Date: Sun, 31 May 2020 11:43:41 +0300 Subject: [PATCH] set yifysubtitles video_types to movie only --- README.md | 1 + libs/subliminal_patch/providers/yifysubtitles.py | 9 +++++---- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 542e769dc..84bd854fd 100644 --- a/README.md +++ b/README.md @@ -71,6 +71,7 @@ If you need something that is not already part of Bazarr, feel free to create a * Wizdom * XSubs * Yavka.net +* YIFY Subtitles * Zimuku ## Screenshot diff --git a/libs/subliminal_patch/providers/yifysubtitles.py b/libs/subliminal_patch/providers/yifysubtitles.py index 3e085851d..560806efe 100644 --- a/libs/subliminal_patch/providers/yifysubtitles.py +++ b/libs/subliminal_patch/providers/yifysubtitles.py @@ -96,6 +96,7 @@ class YifySubtitlesProvider(Provider): languages = {Language(l, c) for (_, l, c) in YifyLanguages} server_url = 'https://www.yifysubtitles.com' + video_types = (Movie,) def initialize(self): self.session = Session() @@ -129,11 +130,11 @@ class YifySubtitlesProvider(Provider): return [] - def _query(self, languages, video): + def query(self, languages, imdb_id): subtitles = [] - logger.info('Searching subtitle %r', video.imdb_id) - response = self.session.get(self.server_url + '/movie-imdb/' + video.imdb_id, + logger.info('Searching subtitle %r', imdb_id) + response = self.session.get(self.server_url + '/movie-imdb/' + imdb_id, allow_redirects=False, timeout=10, headers={'Referer': self.server_url}) response.raise_for_status() @@ -157,7 +158,7 @@ class YifySubtitlesProvider(Provider): return subtitles def list_subtitles(self, video, languages): - return self._query(languages, video) if isinstance(video, Movie) and video.imdb_id else [] + return self.query(languages, video.imdb_id) if isinstance(video, Movie) and video.imdb_id else [] def download_subtitle(self, subtitle): logger.info('Downloading subtitle %r', subtitle.sub_link)