From 333b40c33969328121aea9047ea7500eb726efdd Mon Sep 17 00:00:00 2001
From: Kharec <sandro@cazzaniga.fr>
Date: Tue, 18 Nov 2025 21:04:07 +0100
Subject: [PATCH] feat: extract duplicated pagination logic into a helper
 private method

---
 main.py | 82 ++++++++++++++++++++++++---------------------------------
 1 file changed, 34 insertions(+), 48 deletions(-)

diff --git a/main.py b/main.py
index 1bcf857..a8c084c 100644
--- a/main.py
+++ b/main.py
@@ -161,34 +161,41 @@ class Auditui:
         print("-" * 80)
         print(f"Total: {len(items)} books")
 
+    def _fetch_all_pages(self, response_groups):
+        all_items = []
+        page = 1
+        page_size = 50
+
+        while True:
+            library = self.client.get(
+                path="library",
+                num_results=page_size,
+                page=page,
+                response_groups=response_groups
+            )
+
+            items = library.get("items", [])
+
+            if not items:
+                break
+
+            all_items.extend(items)
+            print(f"Fetched page {page} ({len(items)} items)...", end="\r")
+
+            if len(items) < page_size:
+                break
+
+            page += 1
+
+        return all_items
+
     def list_library(self):
         try:
             print("\nFetching your library...")
 
-            all_items = []
-            page = 1
-            page_size = 50
-
-            while True:
-                library = self.client.get(
-                    path="library",
-                    num_results=page_size,
-                    page=page,
-                    response_groups="contributors,media,product_attrs,product_desc,product_details,rating"
-                )
-
-                items = library.get("items", [])
-
-                if not items:
-                    break
-
-                all_items.extend(items)
-                print(f"Fetched page {page} ({len(items)} items)...", end="\r")
-
-                if len(items) < page_size:
-                    break
-
-                page += 1
+            all_items = self._fetch_all_pages(
+                "contributors,media,product_attrs,product_desc,product_details,rating"
+            )
 
             print(f"\nFetched {len(all_items)} books total.\n")
 
@@ -205,30 +212,9 @@ class Auditui:
         try:
             print("\nFetching your library...")
 
-            all_items = []
-            page = 1
-            page_size = 50
-
-            while True:
-                library = self.client.get(
-                    path="library",
-                    num_results=page_size,
-                    page=page,
-                    response_groups="contributors,media,product_attrs,product_desc,product_details,rating,is_finished,listening_status,percent_complete"
-                )
-
-                items = library.get("items", [])
-
-                if not items:
-                    break
-
-                all_items.extend(items)
-                print(f"Fetched page {page} ({len(items)} items)...", end="\r")
-
-                if len(items) < page_size:
-                    break
-
-                page += 1
+            all_items = self._fetch_all_pages(
+                "contributors,media,product_attrs,product_desc,product_details,rating,is_finished,listening_status,percent_complete"
+            )
 
             print(f"\nFetched {len(all_items)} books total.\n")