Spaces:
Running
Running
| import pytest | |
| from bot.parser import ( | |
| DEFAULT_TEMPLATE, | |
| apply_remove_words, | |
| compute_new_name, | |
| finalize_name, | |
| build_values, | |
| missing_variables, | |
| padded_episode, | |
| parse_caption, | |
| parse_info, | |
| render_template, | |
| sanitize, | |
| split_ext, | |
| ) | |
| def test_parse_info(name, title, season, episode, quality, year): | |
| info = parse_info(name) | |
| assert (info.title, info.season, info.episode, info.quality, info.year) == ( | |
| title, season, episode, quality, year | |
| ) | |
| def test_resolution_is_not_an_episode(): | |
| info = parse_info("Some Movie 1920x1080.mkv") | |
| assert info.episode == "" | |
| def test_codec_detection(): | |
| assert parse_info("Show.S01E01.1080p.x265-GRP.mkv").codec == "x265" | |
| assert parse_info("Show.S01E01.1080p.H.264.mkv").codec == "H264" | |
| assert parse_info("Show - 01 [HEVC].mkv").codec == "HEVC" | |
| def test_title_only_marker_at_start(): | |
| info = parse_info("S01E05 Some Title.mkv") | |
| assert info.title == "Some Title" | |
| assert (info.season, info.episode) == ("01", "05") | |
| def test_split_ext(): | |
| assert split_ext("a.b.mkv") == ("a.b", ".mkv") | |
| assert split_ext("Season 1.5", known_only=True) == ("Season 1.5", "") | |
| assert split_ext("Season 1.5") == ("Season 1.5", "") | |
| assert split_ext("noext") == ("noext", "") | |
| assert split_ext("x.MKV", known_only=True) == ("x", ".MKV") | |
| assert split_ext("weird.abc", known_only=True) == ("weird.abc", "") | |
| assert split_ext("weird.abc") == ("weird", ".abc") | |
| def test_render_template_drops_group_with_empty_variable(): | |
| values = {"title": "Show", "season": "", "episode": "", "quality": "720p", "year": "", "codec": "", "filename": "x", "ext": "mkv"} | |
| assert render_template(DEFAULT_TEMPLATE, values) == "Show [720p]" | |
| values["quality"] = "" | |
| assert render_template(DEFAULT_TEMPLATE, values) == "Show" | |
| def test_render_template_full(): | |
| values = {"title": "Show", "season": "01", "episode": "05", "quality": "720p", "year": "", "codec": "", "filename": "x", "ext": "mkv"} | |
| assert render_template(DEFAULT_TEMPLATE, values) == "[S01E05] Show [720p]" | |
| assert render_template("{title} - {episode}", values) == "Show - 05" | |
| assert render_template("{title}.S{season}E{episode}", values) == "Show.S01E05" | |
| def test_render_template_values_are_not_reparsed(): | |
| values = {"title": "{quality} [x]", "season": "", "episode": "", "quality": "720p", "year": "", "codec": "", "filename": "x", "ext": "mkv"} | |
| assert render_template("{title} [{quality}]", values) == "{quality} [x] [720p]" | |
| def test_render_template_unknown_token_kept(): | |
| values = {"title": "T", "season": "", "episode": "", "quality": "", "year": "", "codec": "", "filename": "x", "ext": "mkv"} | |
| assert render_template("{title} {nope}", values) == "T {nope}" | |
| def test_compute_new_name_filename_mode(): | |
| name, info = compute_new_name( | |
| template=DEFAULT_TEMPLATE, | |
| mode="filename", | |
| original_name="[SubsPlease] Tokyo Revengers - 05 (1080p) [ABCD1234].mkv", | |
| original_ext=".mkv", | |
| ) | |
| assert name == "[S01E05] Tokyo Revengers [1080p].mkv" | |
| assert info.episode == "05" | |
| def test_compute_new_name_caption_mode_uses_caption(): | |
| name, _ = compute_new_name( | |
| template=DEFAULT_TEMPLATE, | |
| mode="caption", | |
| original_name="random_file_123.mkv", | |
| original_ext=".mkv", | |
| caption="Bleach S01E12 720p", | |
| ) | |
| assert name == "[S01E12] Bleach [720p].mkv" | |
| def test_compute_new_name_caption_mode_falls_back_to_filename(): | |
| name, _ = compute_new_name( | |
| template=DEFAULT_TEMPLATE, | |
| mode="caption", | |
| original_name="One.Piece.E1000.1080p.mkv", | |
| original_ext=".mkv", | |
| caption="Hello everyone, enjoy!", | |
| ) | |
| assert name == "[S01E1000] One Piece [1080p].mkv" | |
| def test_compute_new_name_caption_fills_missing_from_filename(): | |
| name, _ = compute_new_name( | |
| template=DEFAULT_TEMPLATE, | |
| mode="caption", | |
| original_name="Show.S01E03.1080p.mkv", | |
| original_ext=".mkv", | |
| caption="Show Name S01E03", | |
| ) | |
| assert name == "[S01E03] Show Name [1080p].mkv" | |
| def test_caption_title_from_first_line_details_from_rest(): | |
| info = parse_caption("My Show\nEpisode: 5\nQuality: 720p") | |
| assert info.title == "My Show" | |
| assert info.quality == "720p" | |
| def test_compute_new_name_remove_words_and_illegal_chars(): | |
| name, _ = compute_new_name( | |
| template="{title} {quality}", | |
| mode="filename", | |
| original_name="Re:Zero - 03 [720p] @SomeChannel.mkv", | |
| original_ext=".mkv", | |
| remove_words=["@SomeChannel"], | |
| ) | |
| assert name == "Re Zero 720p.mkv" | |
| def test_compute_new_name_keeps_extension_and_no_double_ext(): | |
| name, _ = compute_new_name( | |
| template="{title}.{ext}", | |
| mode="filename", | |
| original_name="Movie.2020.720p.mkv", | |
| original_ext=".mkv", | |
| ) | |
| assert name == "Movie.mkv" | |
| def test_compute_new_name_missing_everything_falls_back_to_original(): | |
| name, _ = compute_new_name( | |
| template="[S{season}E{episode}]", | |
| mode="filename", | |
| original_name="holiday video.mp4", | |
| original_ext=".mp4", | |
| ) | |
| assert name == "holiday video.mp4" | |
| def test_compute_new_name_uses_original_ext_when_name_has_none(): | |
| name, _ = compute_new_name( | |
| template="{title}", mode="filename", original_name="Movie", original_ext=".mp4" | |
| ) | |
| assert name == "Movie.mp4" | |
| def test_finalize_truncates_by_bytes_and_keeps_ext(): | |
| out = finalize_name("é" * 300, ".mkv") | |
| assert out.endswith(".mkv") | |
| assert len(out.encode("utf-8")) <= 200 | |
| def test_sanitize_and_remove_words(): | |
| assert sanitize(" ..hidden:name ") == "hidden name" | |
| assert apply_remove_words("Show [Group] 720p", ["[group]"]) == "Show 720p" | |
| # ---- episode numbers written first / bare / in brackets (e.g. "1147 - One Piece [Dual] 720p @Channel") ---- | |
| def test_episode_without_a_marker_word(name, title, episode, quality): | |
| info = parse_info(name) | |
| assert (info.title, info.season, info.episode, info.quality) == (title, "01", episode, quality) | |
| def test_numbers_that_belong_to_titles_are_not_episodes(name, title): | |
| info = parse_info(name) | |
| assert info.episode == "" and info.title == title | |
| def test_an_explicit_marker_beats_a_leading_number(): | |
| info = parse_info("86 - Eighty Six - 05 (1080p).mkv") | |
| assert (info.title, info.episode, info.quality) == ("86 - Eighty Six", "05", "1080p") | |
| def test_channel_tags_end_the_title(): | |
| assert parse_info("One Piece @Anime_X_Universe 720p.mkv").title == "One Piece" | |
| assert parse_info("Some @ Title.mkv").title == "Some @ Title" | |
| def test_the_users_one_piece_files_get_their_episode_number(): | |
| for number in ("1145", "1146", "1147"): | |
| name, _info = compute_new_name( | |
| template="One Piece - E{episode}", mode="filename", original_ext=".mkv", | |
| original_name=f"{number} - One Piece [Dual] 720p @Anime_X_Universe.mkv", | |
| ) | |
| assert name == f"One Piece - E{number}.mkv" | |
| name, _ = compute_new_name( | |
| template=DEFAULT_TEMPLATE, mode="filename", original_ext=".mkv", | |
| original_name="1147 - One Piece [Dual] 720p @Anime_X_Universe.mkv", | |
| ) | |
| assert name == "[S01E1147] One Piece [720p].mkv" | |
| def test_missing_variables_only_reports_holes_outside_bracket_groups(): | |
| info = parse_info("Plain Movie.mkv") | |
| assert missing_variables("One Piece - E{episode}", info) == ["episode"] | |
| assert missing_variables("S{season}E{episode} {title}", info) == ["season", "episode"] | |
| assert missing_variables("{title} {quality} {quality}", info) == ["quality"] | |
| assert missing_variables(DEFAULT_TEMPLATE, info) == [] # bracket groups vanish on purpose | |
| assert missing_variables("{title} ({year})", info) == [] | |
| assert missing_variables("{title} {nope}", info) == [] | |
| assert missing_variables("One Piece - E{episode}", parse_info("1147 - One Piece 720p.mkv")) == [] | |
| # --------------------------------------------------------------------------- | |
| # {1episode} {2episode} {3episode}: the episode with zeros in front | |
| # --------------------------------------------------------------------------- | |
| def test_padded_episode(detected, zeros, expected): | |
| assert padded_episode(detected, zeros) == expected | |
| def test_the_padded_variables_do_not_depend_on_how_the_episode_was_written(written): | |
| info = parse_info(f"Show {written} 720p.mkv") | |
| values = build_values(info, "Show", ".mkv") | |
| assert [values[name] for name in ("episode", "1episode", "2episode", "3episode")] == ["04", "04", "004", "0004"] | |
| assert "4episode" not in values # three padded variables: up to four digits | |
| def test_the_padded_variables_in_a_template(): | |
| name, _ = compute_new_name( | |
| template="{title} - {1episode} - {2episode} - {3episode}", mode="filename", | |
| original_name="One Piece - 4 [1080p].mkv", original_ext=".mkv", | |
| ) | |
| assert name == "One Piece - 04 - 004 - 0004.mkv" | |
| name, _ = compute_new_name( | |
| template="{title} {2episode}", mode="filename", original_name="1147 - One Piece [Dual] 720p.mkv", original_ext=".mkv", | |
| ) | |
| assert name == "One Piece 1147.mkv" # a number that is wide enough keeps its digits | |
| def test_third_padded_episode_is_four_digits_wide(original, expected): | |
| name, _ = compute_new_name(template="{title} {3episode}", mode="filename", original_name=original, original_ext=".mkv") | |
| assert name == expected | |
| def test_a_fourth_padded_variable_does_not_exist(): | |
| values = build_values(parse_info("One Piece - 4 [1080p].mkv"), "One Piece", ".mkv") | |
| assert "4episode" not in values | |
| name, _ = compute_new_name( | |
| template="{title} {4episode}", mode="filename", original_name="One Piece - 4 [1080p].mkv", original_ext=".mkv" | |
| ) | |
| assert name == "One Piece {4episode}.mkv" # an unknown variable is left as plain text | |
| def test_without_an_episode_a_group_with_a_padded_variable_goes_away(): | |
| info = parse_info("Plain Movie.mkv") | |
| values = build_values(info, "Plain Movie", ".mkv") | |
| assert values["1episode"] == values["3episode"] == "" | |
| assert render_template("{title} [E{3episode}]", values) == "Plain Movie" | |
| assert missing_variables("{title} E{2episode}", info) == ["2episode"] # a bare one leaves a hole in the name | |
| assert missing_variables("{title} [E{2episode}] {episode} {3episode}", info) == ["episode", "3episode"] | |
| assert missing_variables("{title} E{2episode}", parse_info("One Piece - 05.mkv")) == [] | |