From ab7b043d86859d4cbb221743f5a85b478484f219 Mon Sep 17 00:00:00 2001 From: zx900930 Date: Thu, 18 Feb 2021 19:29:04 +0100 Subject: [PATCH] fix for #1276 --- bazarr/list_subtitles.py | 33 +++++++++++++++++++-------------- 1 file changed, 19 insertions(+), 14 deletions(-) diff --git a/bazarr/list_subtitles.py b/bazarr/list_subtitles.py index cc819e28f..b166a35d7 100644 --- a/bazarr/list_subtitles.py +++ b/bazarr/list_subtitles.py @@ -58,10 +58,12 @@ def store_subtitles(original_path, reversed_path): brazilian_portuguese = [".pt-br", ".pob", "pb"] brazilian_portuguese_forced = [".pt-br.forced", ".pob.forced", "pb.forced"] - simplified_chinese = [".chs", ".sc", ".zhs", ".hans", ".gb", u"简", u"双语"] - simplified_chinese_forced = [".chs.forced", ".sc.forced", ".zhs.forced", ".hans.forced", ".gb.forced", u"简体中文.forced", u"双语.forced"] - traditional_chinese = [".cht", ".tc", ".zht", ".hant", ".big5", u"繁", u"雙語", "zh-tw"] - traditional_chinese_forced = [".cht.forced", ".tc.forced", ".zht.forced", ".hant.forced", ".big5.forced", u"繁體中文.forced", u"雙語.forced", "zh-tw.forced"] + simplified_chinese_long = [".simplified"] + simplified_chinese_short = [".chs", ".sc", ".zhs", "hans", ".gb", u"简", u"双语"] + simplified_chinese_forced = [".chs.forced", ".sc.forced", ".zhs.forced", "hans.forced", ".gb.forced", u"简体中文.forced", u"双语.forced"] + traditional_chinese_long = [".traditional"] + traditional_chinese_short = [".cht", ".tc", ".zh-tw", ".zht", "hant", "big5", u"繁", u"雙語"] + traditional_chinese_forced = [".cht.forced", ".tc.forced", ".zht.forced", "hant.forced", ".big5.forced", u"繁體中文.forced", u"雙語.forced", "zh-tw.forced"] try: dest_folder = get_subtitle_destination_folder() @@ -89,7 +91,7 @@ def store_subtitles(original_path, reversed_path): logging.debug("BAZARR external subtitles detected: " + "pb:forced") actual_subtitles.append( [str("pb:forced"), path_mappings.path_replace_reverse(subtitle_path)]) - elif any(ext in (str(os.path.splitext(subtitle)[0]).lower())[-12:] for ext in simplified_chinese): + elif any(ext in (str(subtitle_path).lower())[:-7] for ext in simplified_chinese_short) or (str(subtitle_path).lower())[:-11] in simplified_chinese_long: logging.debug("BAZARR external subtitles detected: " + "zh") actual_subtitles.append( [str("zh"), path_mappings.path_replace_reverse(subtitle_path)]) @@ -97,7 +99,7 @@ def store_subtitles(original_path, reversed_path): logging.debug("BAZARR external subtitles detected: " + "zh:forced") actual_subtitles.append( [str("zh:forced"), path_mappings.path_replace_reverse(subtitle_path)]) - elif any(ext in (str(os.path.splitext(subtitle)[0]).lower())[-12:] for ext in traditional_chinese): + elif any(ext in (str(subtitle_path).lower())[:-7] for ext in traditional_chinese_short) or (str(subtitle_path).lower())[:-12] in traditional_chinese_long: logging.debug("BAZARR external subtitles detected: " + "zt") actual_subtitles.append( [str("zt"), path_mappings.path_replace_reverse(subtitle_path)]) @@ -170,10 +172,12 @@ def store_subtitles_movie(original_path, reversed_path): brazilian_portuguese = [".pt-br", ".pob", "pb"] brazilian_portuguese_forced = [".pt-br.forced", ".pob.forced", "pb.forced"] - simplified_chinese = [".chs", ".sc", ".zhs", ".hans", ".gb", u"简", u"双语"] - simplified_chinese_forced = [".chs.forced", ".sc.forced", ".zhs.forced", ".hans.forced", ".gb.forced", u"简体中文.forced", u"双语.forced"] - traditional_chinese = [".cht", ".tc", ".zht", ".hant", ".big5", u"繁", u"雙語", "zh-tw"] - traditional_chinese_forced = [".cht.forced", ".tc.forced", ".zht.forced",".hant.forced", ".big5.forced", u"繁體中文.forced", u"雙語.forced", "zh-tw.forced"] + simplified_chinese_long = [".simplified"] + simplified_chinese_short = [".chs", ".sc", ".zhs", "hans", ".gb", u"简", u"双语"] + simplified_chinese_forced = [".chs.forced", ".sc.forced", ".zhs.forced", "hans.forced", ".gb.forced", u"简体中文.forced", u"双语.forced"] + traditional_chinese_long = [".traditional"] + traditional_chinese_short = [".cht", ".tc", ".zh-tw", ".zht", "hant", "big5", u"繁", u"雙語"] + traditional_chinese_forced = [".cht.forced", ".tc.forced", ".zht.forced", "hant.forced", ".big5.forced", u"繁體中文.forced", u"雙語.forced", "zh-tw.forced"] try: dest_folder = get_subtitle_destination_folder() or '' core.CUSTOM_PATHS = [dest_folder] if dest_folder else [] @@ -197,13 +201,13 @@ def store_subtitles_movie(original_path, reversed_path): elif str(os.path.splitext(subtitle)[0]).lower().endswith(tuple(brazilian_portuguese_forced)): logging.debug("BAZARR external subtitles detected: " + "pb:forced") actual_subtitles.append([str("pb:forced"), path_mappings.path_replace_reverse_movie(subtitle_path)]) - elif any(ext in (str(os.path.splitext(subtitle)[0]).lower())[-12:] for ext in simplified_chinese): + elif any(ext in (str(subtitle_path).lower())[:-7] for ext in simplified_chinese_short) or (str(subtitle_path).lower())[:-11] in simplified_chinese_long: logging.debug("BAZARR external subtitles detected: " + "zh") actual_subtitles.append([str("zh"), path_mappings.path_replace_reverse_movie(subtitle_path)]) elif any(ext in (str(os.path.splitext(subtitle)[0]).lower())[-12:] for ext in simplified_chinese_forced): logging.debug("BAZARR external subtitles detected: " + "zh:forced") actual_subtitles.append([str("zh:forced"), path_mappings.path_replace_reverse_movie(subtitle_path)]) - elif any(ext in (str(os.path.splitext(subtitle)[0]).lower())[-12:] for ext in traditional_chinese): + elif any(ext in (str(subtitle_path).lower())[:-7] for ext in traditional_chinese_short) or (str(subtitle_path).lower())[:-12] in traditional_chinese_long: logging.debug("BAZARR external subtitles detected: " + "zt") actual_subtitles.append([str("zt"), path_mappings.path_replace_reverse_movie(subtitle_path)]) elif any(ext in (str(os.path.splitext(subtitle)[0]).lower())[-12:] for ext in traditional_chinese_forced): @@ -545,8 +549,9 @@ def guess_external_subtitles(dest_folder, subtitles): detected_language = guess_language(text) #add simplified and traditional chinese detection if detected_language == 'zh': - traditional_chinese = [".cht", ".tc", ".traditional",".zh-tw", ".zht", "hant", "big5", u"繁", u"雙語"] - if any(ext in (str(subtitle_path).lower())[:-16] for ext in traditional_chinese): + traditional_chinese_long = [".traditional"] + traditional_chinese_short = [".cht", ".tc", ".zh-tw", ".zht", "hant", "big5", u"繁", u"雙語"] + if any(ext in (str(subtitle_path).lower())[:-7] for ext in traditional_chinese_short) or (str(subtitle_path).lower())[:-12]==traditional_chinese_long: detected_language == 'zt' except UnicodeDecodeError: detector = Detector()