fixes bug introduced in the increment logic for file naming

beaufour · Nov 9, 2022 · 55ba6ce · 55ba6ce
1 parent 80c2b43
commit 55ba6ce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flickr_download/filename_handlers.py b/flickr_download/filename_handlers.py
@@ -95,7 +95,7 @@ def title_increment(pset: Optional[Photoset], photo: Photo, suffix: Optional[str
     index = pset.id if pset else "1"
     photo_index = INCREMENT_INDEX[index][photo.title]
     if photo_index:
-        extra = f"({photo.index})"
+        extra = f"({photo_index})"
     INCREMENT_INDEX[index][photo.title] += 1
     return get_filename(f"{photo.title}{suffix}{extra}")
 

diff --git a/tests/test_filename_handlers.py b/tests/test_filename_handlers.py
@@ -8,7 +8,7 @@
 class TestFilenameHandlers:
     def setup_method(self) -> None:
         self._pset = Mock(Photoset, title="Some Set", id=999)
-        self._photo = Mock(Photo, title="Some Photo", id=123, index=1)
+        self._photo = Mock(Photo, title="Some Photo", id=123)
         self._suffix = ""
 
     def teardown_method(self) -> None: