From 26118abb2a385a30667f4cf1802730641a03dad1 Mon Sep 17 00:00:00 2001 From: morpheus65535 <5130500+morpheus65535@users.noreply.github.com> Date: Sun, 30 Sep 2018 20:23:12 -0400 Subject: [PATCH] Real fix for #158 --- get_subtitle.py | 9 +++++---- libs/subliminal/cli.py | 2 +- libs/subliminal/core.py | 3 +-- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/get_subtitle.py b/get_subtitle.py index d7bcd6d55..b94e32e89 100644 --- a/get_subtitle.py +++ b/get_subtitle.py @@ -9,6 +9,7 @@ import time from datetime import datetime, timedelta from babelfish import Language from subliminal import region, scan_video, Video, download_best_subtitles, compute_score, save_subtitles, AsyncProviderPool, score, list_subtitles, download_subtitles +from subliminal.subtitle import get_subtitle_path from get_languages import language_from_alpha3, alpha2_from_alpha3, alpha3_from_alpha2 from bs4 import UnicodeDammit from get_settings import get_general_settings, pp_replace, path_replace, path_replace_movie, path_replace_reverse, path_replace_reverse_movie @@ -78,11 +79,11 @@ def download_subtitle(path, language, hi, providers, providers_auth, sceneName, logging.error('Error saving subtitles file to disk.') return None else: - downloaded_provider = str(result[0][0]).strip('<>').split(' ')[0][:-8] + downloaded_provider = str(result[0]).strip('<>').split(' ')[0][:-8] downloaded_language = language_from_alpha3(language) downloaded_language_code2 = alpha2_from_alpha3(language) downloaded_language_code3 = language - downloaded_path = result[1] + downloaded_path = get_subtitle_path(path, language=lang_obj) if used_sceneName == True: message = downloaded_language + " subtitles downloaded from " + downloaded_provider + " with a score of " + unicode(score) + "% using this scene name: " + sceneName else: @@ -221,11 +222,11 @@ def manual_download_subtitle(path, language, hi, subtitle, provider, providers_a logging.exception('Error saving subtitles file to disk.') return None else: - downloaded_provider = str(result[0][0]).strip('<>').split(' ')[0][:-8] + downloaded_provider = str(result[0]).strip('<>').split(' ')[0][:-8] downloaded_language = language_from_alpha3(language) downloaded_language_code2 = alpha2_from_alpha3(language) downloaded_language_code3 = language - downloaded_path = result[1] + downloaded_path = get_subtitle_path(path, language=lang_obj) message = downloaded_language + " subtitles downloaded from " + downloaded_provider + " with a score of " + unicode(score) + "% using manual search." if use_postprocessing is True: diff --git a/libs/subliminal/cli.py b/libs/subliminal/cli.py index 0b8dcee02..2eb502fbc 100644 --- a/libs/subliminal/cli.py +++ b/libs/subliminal/cli.py @@ -407,7 +407,7 @@ def download(obj, provider, refiner, language, age, directory, encoding, single, # save subtitles total_subtitles = 0 for v, subtitles in downloaded_subtitles.items(): - saved_subtitles = save_subtitles(v, subtitles, single=single, directory=directory, encoding=encoding)[0] + saved_subtitles = save_subtitles(v, subtitles, single=single, directory=directory, encoding=encoding) total_subtitles += len(saved_subtitles) if verbose > 0: diff --git a/libs/subliminal/core.py b/libs/subliminal/core.py index 2c13d6975..e879eb80f 100644 --- a/libs/subliminal/core.py +++ b/libs/subliminal/core.py @@ -750,7 +750,6 @@ def save_subtitles(video, subtitles, single=False, directory=None, encoding=None continue # create subtitle path - subtitle_path = None subtitle_path = get_subtitle_path(video.name, None if single else subtitle.language) if directory is not None: subtitle_path = os.path.join(directory, os.path.split(subtitle_path)[1]) @@ -769,4 +768,4 @@ def save_subtitles(video, subtitles, single=False, directory=None, encoding=None if single: break - return [saved_subtitles, subtitle_path] + return saved_subtitles