From d0730679239914b1fd778c30ff0fe973412e290b Mon Sep 17 00:00:00 2001 From: Steve Kowalik Date: Sun, 25 Aug 2024 18:16:59 +1000 Subject: [PATCH] Replace spaces with dots in SourceFile.name --- sinking/source.py | 5 +++-- tests/test_source.py | 6 ++++++ 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/sinking/source.py b/sinking/source.py index 439790c..34701e7 100644 --- a/sinking/source.py +++ b/sinking/source.py @@ -16,9 +16,10 @@ class SourceFile: @property def name(self) -> str: o = Options() + name = self.path.name if o.rename and self.needs_rename: - return f"{self.path.parts[-2]}{self.path.suffix}" - return self.path.name + name = f"{self.path.parts[-2]}{self.path.suffix}" + return name.replace(" ", ".") @property def pattern(self) -> str: diff --git a/tests/test_source.py b/tests/test_source.py index e92a1bb..465e6d1 100644 --- a/tests/test_source.py +++ b/tests/test_source.py @@ -25,6 +25,12 @@ def test_name_with_rename_false(self): s = SourceFile(pathlib.Path("/foo/bar/baz/bar")) self.assertEqual(s.name, "bar") + def test_name_with_spaces(self): + p = pathlib.Path("/foo/bar/FooBarBaz.S03E05/FooBarBaz S03E05") + s = SourceFile(p) + self.assertFalse(s.needs_rename) + self.assertEqual(s.name, "FooBarBaz.S03E05") + def test_needs_rename(self): s = SourceFile(pathlib.Path("/foo/bar/baz/foo")) self.assertEqual(s.name, "baz")