Limit result_range_max

nationalarchives · Jan 10, 2024 · d388c4d · d388c4d
1 parent 56dc9d8
commit d388c4d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/app/schemas/api_response.py b/app/schemas/api_response.py
@@ -28,7 +28,8 @@ def get_result_range_min(self):
         return (self.results_per_page * (self.page - 1)) + 1
 
     def get_result_range_max(self):
-        return self.results_per_page * self.page
+        page_max = self.results_per_page * self.page
+        return page_max if page_max <= self.count else self.count
 
     def page_in_range(self):
         return self.page > 0 and self.page <= self.get_pages()