diff options
| author | morpheus65535 <[email protected]> | 2021-10-12 12:09:22 -0400 |
|---|---|---|
| committer | morpheus65535 <[email protected]> | 2021-10-12 12:09:22 -0400 |
| commit | 0bc2605439184fba36fcfd8e4a61c4b664a827e8 (patch) | |
| tree | f1500db4b8f9a0663ad1603b518086b610d00a1c | |
| parent | a819fc34b36a45b21fee6def70003412ca66d41f (diff) | |
| download | bazarr-0bc2605439184fba36fcfd8e4a61c4b664a827e8.tar.gz bazarr-0bc2605439184fba36fcfd8e4a61c4b664a827e8.zip | |
Improved language selection for Titrari.ro provider.
| -rw-r--r-- | libs/subliminal_patch/providers/titrari.py | 17 |
1 files changed, 11 insertions, 6 deletions
diff --git a/libs/subliminal_patch/providers/titrari.py b/libs/subliminal_patch/providers/titrari.py index 8710f578a..5a42e0ac8 100644 --- a/libs/subliminal_patch/providers/titrari.py +++ b/libs/subliminal_patch/providers/titrari.py @@ -139,10 +139,10 @@ class TitrariProvider(Provider, ProviderSubtitleArchiveMixin): def terminate(self): self.session.close() - def query(self, languages=None, title=None, imdb_id=None, video=None): + def query(self, language=None, title=None, imdb_id=None, video=None): subtitles = [] - params = self.getQueryParams(imdb_id, title) + params = self.getQueryParams(imdb_id, title, language) search_response = self.session.get(self.api_url, params=params, timeout=15) search_response.raise_for_status() @@ -194,7 +194,7 @@ class TitrariProvider(Provider, ProviderSubtitleArchiveMixin): logger.error("Error parsing comments.") episode_number = video.episode if isinstance(video, Episode) else None - subtitle = self.subtitle_class(next(iter(languages)), download_link, index, comments, title, sub_imdb_id, + subtitle = self.subtitle_class(language, download_link, index, comments, title, sub_imdb_id, year, downloads, isinstance(video, Episode), episode_number) logger.debug('[#### Provider: titrari.ro] Found subtitle %r', str(subtitle)) subtitles.append(subtitle) @@ -231,7 +231,7 @@ class TitrariProvider(Provider, ProviderSubtitleArchiveMixin): # z8 = language (-1: all, 1: ron, 2: eng) # z9 = genre (All: all, Action: action etc.) # z11 = type (0: any, 1: movie, 2: series) - def getQueryParams(self, imdb_id, title): + def getQueryParams(self, imdb_id, title, language): queryParams = { 'page': self.query_advanced_search, 'z7': '', @@ -249,6 +249,11 @@ class TitrariProvider(Provider, ProviderSubtitleArchiveMixin): elif title is not None: queryParams["z7"] = title + if language == 'ro': + queryParams["z8"] = '1' + elif language == 'en': + queryParams["z8"] = '2' + return queryParams def list_subtitles(self, video, languages): @@ -262,8 +267,8 @@ class TitrariProvider(Provider, ProviderSubtitleArchiveMixin): except: logger.error("[#### Provider: titrari.ro] Error parsing video.imdb_id.") - subtitles = [s for s in - self.query(languages, title, imdb_id, video)] + subtitles = [s for lang in languages for s in + self.query(lang, title, imdb_id, video)] return subtitles def download_subtitle(self, subtitle): |