Refactor news API methods to use 'limit' parameter instead of 'total' for consistency across wrappers

src/app/news/news_api.py

@@ -26,22 +26,25 @@ class NewsApiWrapper(NewsWrapper):
         self.language = "en" # TODO Only English articles for now?
         self.max_page_size = 100
 
-    def get_top_headlines(self, total: int = 100) -> list[Article]:
-        page_size = min(self.max_page_size, total)
-        pages = (total // page_size) + (1 if total % page_size > 0 else 0)
+    def __calc_pages(self, limit: int, page_size: int) -> tuple[int, int]:
+        page_size = min(self.max_page_size, limit)
+        pages = (limit // page_size) + (1 if limit % page_size > 0 else 0)
+        return pages, page_size
 
+    def get_top_headlines(self, limit: int = 100) -> list[Article]:
+        pages, page_size = self.__calc_pages(limit, self.max_page_size)
         articles = []
+
         for page in range(1, pages + 1):
             headlines = self.client.get_top_headlines(q="", category=self.category, language=self.language, page_size=page_size, page=page)
             results = [result_to_article(article) for article in headlines.get("articles", [])]
             articles.extend(results)
         return articles
 
-    def get_latest_news(self, query: str, total: int = 100) -> list[Article]:
-        page_size = min(self.max_page_size, total)
-        pages = (total // page_size) + (1 if total % page_size > 0 else 0)
-
+    def get_latest_news(self, query: str, limit: int = 100) -> list[Article]:
+        pages, page_size = self.__calc_pages(limit, self.max_page_size)
         articles = []
+
         for page in range(1, pages + 1):
             everything = self.client.get_everything(q=query, language=self.language, sort_by="publishedAt", page_size=page_size, page=page)
             results = [result_to_article(article) for article in everything.get("articles", [])]