Rework Curator into a Jellyfin-first library manager

2026-08-31 18:34:03 +02:00
parent 1c78ca9013
commit 6a2baf243f
25 changed files with 2420 additions and 1502 deletions
@@ -0,0 +1,114 @@
+"""Tests for series naming/scanning (src.core.series) + FileManager episodes."""
+import pytest
+
+from src.core.series import (
+    episode_filename,
+    is_canonical_episode,
+    parse_episode,
+    scan_series,
+    season_folder_name,
+    series_folder_stem,
+)
+
+
+@pytest.mark.parametrize("name,expected", [
+    ("Breaking Bad S01E02.mkv", (1, 2)),
+    ("bb.s1e2.mkv", (1, 2)),
+    ("Show 01x02 x264.mkv", (1, 2)),
+    ("Series.S01E02E03.mkv", (1, 2)),   # multi-ep → first
+    ("Series S12E145.mkv", (12, 145)),
+    ("No code here.mkv", None),
+])
+def test_parse_episode(name, expected):
+    assert parse_episode(name) == expected
+
+
+def test_season_folder_name():
+    assert season_folder_name(0) == "Specials"
+    assert season_folder_name(1) == "Season 01"
+    assert season_folder_name(12) == "Season 12"
+
+
+def test_episode_filename():
+    assert episode_filename("Breaking Bad", 1, 2, ".mkv") == "Breaking Bad S01E02.mkv"
+
+
+def test_series_folder_stem():
+    assert series_folder_stem("Breaking Bad", 2008, 1396) == "Breaking Bad (2008) [tmdbid-1396]"
+    assert series_folder_stem("Neznámý", None, 42) == "Neznámý [tmdbid-42]"
+    assert series_folder_stem("Jen název", None, None) == "Jen název"
+
+
+def test_is_canonical_episode(tmp_path):
+    folder = tmp_path / "Breaking Bad (2008) [tmdbid-1396]" / "Season 01"
+    folder.mkdir(parents=True)
+    good = folder / "Breaking Bad S01E02.mkv"
+    good.write_bytes(b"x")
+    bad = folder / "bb.s01e02.mkv"
+    bad.write_bytes(b"x")
+    assert is_canonical_episode(good, "Breaking Bad")
+    assert not is_canonical_episode(bad, "Breaking Bad")
+
+
+def test_scan_series(tmp_path):
+    lib = tmp_path / "Seriály"
+    s1 = lib / "Breaking Bad (2008) [tmdbid-1396]" / "Season 01"
+    s1.mkdir(parents=True)
+    (s1 / "Breaking Bad S01E01.mkv").write_bytes(b"x")
+    (s1 / "Breaking Bad S01E02.mkv").write_bytes(b"x")
+    (s1 / "poster.jpg").write_bytes(b"img")  # sidecar → skipped
+
+    found = scan_series(lib, "Seriály")
+    assert len(found) == 1
+    ser = found[0]
+    assert ser.title == "Breaking Bad" and ser.year == "2008" and ser.tmdb_id == 1396
+    assert ser.library == "Seriály"
+    assert len(ser.episodes) == 2
+    seasons = ser.seasons()
+    assert list(seasons.keys()) == [1] and len(seasons[1]) == 2
+
+
+class TestFileManagerSeries:
+    @pytest.fixture
+    def file_manager(self, tmp_path, monkeypatch):
+        import src.core.config as config_module
+        monkeypatch.setattr(config_module, "GLOBAL_CONFIG_FILE", tmp_path / ".gtag")
+        from src.core.tag_manager import TagManager
+        from src.core.file_manager import FileManager
+        fm = FileManager(TagManager())
+        fm.set_pool_dir(tmp_path / "pool")
+        return fm
+
+    def test_import_episodes_places_them(self, file_manager, tmp_path):
+        for n in ("bb.S01E01.mkv", "bb.S01E02.mkv", "notes.txt_S00.dummy"):
+            (tmp_path / n).write_bytes(b"x")
+        result = file_manager.import_episodes(
+            [tmp_path / "bb.S01E01.mkv", tmp_path / "bb.S01E02.mkv"],
+            "Breaking Bad", year=2008, tmdb_id=1396)
+
+        assert result["imported"] == 2
+        base = (tmp_path / "pool" / "Seriály"
+                / "Breaking Bad (2008) [tmdbid-1396]" / "Season 01")
+        assert (base / "Breaking Bad S01E01.mkv").exists()
+        assert (base / "Breaking Bad S01E02.mkv").exists()
+
+    def test_import_episodes_skips_uncoded(self, file_manager, tmp_path):
+        (tmp_path / "random.mkv").write_bytes(b"x")
+        result = file_manager.import_episodes(
+            [tmp_path / "random.mkv"], "Breaking Bad", year=2008)
+        assert result["imported"] == 0
+        assert result["skipped_no_code"] == ["random.mkv"]
+
+    def test_rename_episodes_to_canonical(self, file_manager, tmp_path):
+        # a mis-named episode sitting in the wrong place
+        folder = (tmp_path / "pool" / "Seriály"
+                  / "Breaking Bad (2008) [tmdbid-1396]")
+        folder.mkdir(parents=True)
+        (folder / "bb.1x03.HDTV.mkv").write_bytes(b"x")
+
+        series_list = file_manager.load_series()
+        assert len(series_list) == 1
+        result = file_manager.rename_episodes_to_canonical(series_list[0])
+
+        assert result["renamed"] == [("bb.1x03.HDTV.mkv", "Breaking Bad S01E03.mkv")]
+        assert (folder / "Season 01" / "Breaking Bad S01E03.mkv").exists()