Fix pagination

Author: TheEdgeOfRage

pyazo_api/domain/images/actions/get_images.py

@@ -10,5 +10,9 @@ class GetImagesAction:
     def __init__(self, image_repository: Annotated[ImageRepository, Depends()]):
         self.image_repository: ImageRepository = image_repository
 
-    async def __call__(self, owner: UserGet, pagination: PaginationRequest) -> PaginatedResults:
-        return await self.image_repository.get_paginated_images_by_owner_id(owner.id, pagination)
+    async def __call__(
+        self, owner: UserGet, pagination: PaginationRequest
+    ) -> PaginatedResults:
+        return await self.image_repository.get_paginated_images_by_owner_id(
+            owner.id, pagination
+        )

pyazo_api/util/auth.py

@@ -38,13 +38,11 @@ async def get_user(
             algorithms=[settings.jwt_algorithm],
         )
     except PyJWTError as e:
-        print("AAAAAAAAAAAAAAAAAA", str(e))
         return None
 
     try:
         data = TokenData.model_validate(payload)
     except ValidationError as e:
-        print("BBBBBBBBBBBBBBBBBB", str(e))
         return None
 
     return await user_repository.get_by_username(data.sub)

pyazo_api/util/pagination.py

@@ -20,12 +20,12 @@ def offset(self) -> int:
         return self.page * self.limit
 
 
-class PaginatedResults(BaseModel):
-    results: list[BaseModel]
+class PaginatedResults[T](BaseModel):
+    results: list[T]
     count: int
     next_page: int
 
-    def get_results(self) -> Iterator[BaseModel]:
+    def get_results(self) -> Iterator[T]:
         for result in self.results:
             yield result