Adds support for moving with original filename (#43)

jkwill87 · Mar 6, 2022 · 15ac312 · 15ac312
1 parent 9f34c79
commit 15ac312
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/mnamer/metadata.py b/mnamer/metadata.py
@@ -45,6 +45,7 @@ class Metadata:
     quality: Optional[str] = None
     synopsis: Optional[str] = None
     media: Union[MediaType, str, None] = None
+    original: Optional[str] = None
 
     def __setattr__(self, key: str, value: Any):
         converter_map: Dict[str, Callable] = {

diff --git a/mnamer/target.py b/mnamer/target.py
@@ -143,6 +143,7 @@ def _parse(self, file_path: Path):
             None: Metadata,
         }[media_type]
         self.metadata = meta_cls(language=self._settings.language)
+        self.metadata.original = self.source.name
         self.metadata.quality = (
             " ".join(
                 path_data[key]

diff --git a/tests/e2e/test_moving.py b/tests/e2e/test_moving.py
@@ -217,3 +217,15 @@ def test_ambiguous_language_deletction(e2e_run, setup_test_files):
     )
     result = e2e_run("--batch", ".")
     assert result.code == 0
+
+
+@pytest.mark.usefixtures("setup_test_dir")
+def test_original_filename(e2e_run, setup_test_files):
+    setup_test_files("archer.2009.s10e07.webrip.x264-lucidtv.mp4")
+    result = e2e_run(
+        "--batch",
+        "--episode-format='{original}'",
+        ".",
+    )
+    assert result.code == 0
+    assert "archer.2009.s10e07.webrip.x264-lucidtv.mp4" in result.out