From 5a0993479669bf34007a155e0a7be627306e188a Mon Sep 17 00:00:00 2001
From: Brian Pepple <bdpepple@gmail.com>
Date: Fri, 15 Sep 2023 16:12:44 -0400
Subject: [PATCH] Add remove_page() method to `Comic` class (#68)

---
 darkseid/comic.py   |  6 ++++++
 tests/test_comic.py | 21 +++++++++++++++++++++
 2 files changed, 27 insertions(+)

diff --git a/darkseid/comic.py b/darkseid/comic.py
index b70ad5b..be42c32 100644
--- a/darkseid/comic.py
+++ b/darkseid/comic.py
@@ -376,6 +376,12 @@ def remove_metadata(self: "Comic") -> bool:
             return self._successful_write(write_success, False, None)
         return True
 
+    def remove_page(self: "Comic", page_idx: int) -> bool:
+        """Remove page from the archive."""
+        page = self.get_page_name(page_idx)
+        write_success = self.archiver.remove_file(page)
+        return self._successful_write(write_success, False, None)
+
     def _successful_write(
         self: "Comic",
         write_success: bool,
diff --git a/tests/test_comic.py b/tests/test_comic.py
index bf2945e..0731a2a 100644
--- a/tests/test_comic.py
+++ b/tests/test_comic.py
@@ -17,6 +17,27 @@
 #######
 # CBZ #
 #######
+def test_archive_delete_page(tmp_path: Path, fake_metadata: Metadata) -> None:
+    z_file = tmp_path / "test.cbz"
+    with zipfile.ZipFile(z_file, "w") as zf:
+        for p in IMG_DIR.iterdir():
+            zf.write(p)
+    # Prep test file
+    ca = Comic(z_file)
+    test_md = Metadata()
+    test_md.set_default_page_list(ca.get_number_of_pages())
+    test_md.overlay(fake_metadata)
+    ca.write_metadata(test_md)
+
+    old_num_pages = ca.get_number_of_pages()
+    result = ca.remove_page(1)
+    ca.get_page_name_list()
+    num_pages = ca.get_number_of_pages()
+    assert result is True
+    assert old_num_pages - 1 == num_pages
+    assert ca.has_metadata()
+
+
 def test_archive_from_img_dir(tmp_path: Path, fake_metadata: Metadata) -> None:
     z_file: Path = tmp_path / "Aquaman v1 #001 (of 08) (1994).cbz"
     with zipfile.ZipFile(z_file, "w") as zf: