feature: page number pagination added to list endpoints

Author: DataRohit

src/authors/routes.py

@@ -134,8 +134,10 @@ async def update_author_profile_image(
 
 
 @author_router.get("/list", status_code=status.HTTP_200_OK)
-async def list_authors(session: AsyncSession = Depends(get_session)):
-    authors = await author_service.list_authors(session)
+async def list_authors(request: Request, session: AsyncSession = Depends(get_session)):
+    page = int(request.query_params.get("page", 1))
+
+    authors = await author_service.list_authors(page, session)
 
     return JSONResponse(
         status_code=status.HTTP_200_OK,
@@ -153,9 +155,13 @@ async def list_authors(session: AsyncSession = Depends(get_session)):
 
 @author_router.get("/list/{nationality}", status_code=status.HTTP_200_OK)
 async def list_authors_by_nationality(
-    nationality: str, session: AsyncSession = Depends(get_session)
+    request: Request, nationality: str, session: AsyncSession = Depends(get_session)
 ):
-    authors = await author_service.list_authors_by_nationality(nationality, session)
+    page = int(request.query_params.get("page", 1))
+
+    authors = await author_service.list_authors_by_nationality(
+        nationality, page, session
+    )
 
     return JSONResponse(
         status_code=status.HTTP_200_OK,

src/authors/service.py

@@ -75,15 +75,18 @@ async def get_author_by_name(
         return author
 
     async def list_authors_by_nationality(
-        self, nationality: str, session: AsyncSession
+        self, nationality: str, page: int, session: AsyncSession
     ):
         result = await session.execute(
-            select(Author).where(Author.nationality == nationality)
+            select(Author)
+            .where(Author.nationality == nationality)
+            .offset((page - 1) * 10)
+            .limit(10)
         )
         authors = result.scalars().all()
         return authors
 
-    async def list_authors(self, session: AsyncSession):
-        result = await session.execute(select(Author))
+    async def list_authors(self, page: int, session: AsyncSession):
+        result = await session.execute(select(Author).offset((page - 1) * 10).limit(10))
         authors = result.scalars().all()
         return authors

src/books/routes.py

@@ -98,8 +98,12 @@ async def update_book_category(
 
 
 @book_category_router.get("/list", status_code=status.HTTP_200_OK)
-async def list_book_categories(session: AsyncSession = Depends(get_session)):
-    book_categories = await book_category_service.list_book_categories(session)
+async def list_book_categories(
+    request: Request, session: AsyncSession = Depends(get_session)
+):
+    page = int(request.query_params.get("page", 1))
+
+    book_categories = await book_category_service.list_book_categories(page, session)
 
     return JSONResponse(
         status_code=status.HTTP_200_OK,
@@ -232,8 +236,12 @@ async def update_book_genre(
 
 
 @book_genre_router.get("/list", status_code=status.HTTP_200_OK)
-async def list_book_genres(session: AsyncSession = Depends(get_session)):
-    book_genres = await book_genre_service.list_book_genres(session)
+async def list_book_genres(
+    request: Request, session: AsyncSession = Depends(get_session)
+):
+    page = int(request.query_params.get("page", 1))
+
+    book_genres = await book_genre_service.list_book_genres(page, session)
 
     return JSONResponse(
         status_code=status.HTTP_200_OK,
@@ -363,8 +371,10 @@ async def update_book(
 
 
 @book_router.get("/list", status_code=status.HTTP_200_OK)
-async def list_books(session: AsyncSession = Depends(get_session)):
-    books = await book_service.list_books(session)
+async def list_books(request: Request, session: AsyncSession = Depends(get_session)):
+    page = int(request.query_params.get("page", 1))
+
+    books = await book_service.list_books(page, session)
 
     return JSONResponse(
         status_code=status.HTTP_200_OK,
@@ -443,9 +453,11 @@ async def get_book_by_title(title: str, session: AsyncSession = Depends(get_sess
 
 @book_router.get("/list/category/{category}", status_code=status.HTTP_200_OK)
 async def list_books_by_category(
-    category: str, session: AsyncSession = Depends(get_session)
+    request: Request, category: str, session: AsyncSession = Depends(get_session)
 ):
-    books = await book_service.list_books_by_category(category, session)
+    page = int(request.query_params.get("page", 1))
+
+    books = await book_service.list_books_by_category(category, page, session)
 
     return JSONResponse(
         status_code=status.HTTP_200_OK,
@@ -460,8 +472,12 @@ async def list_books_by_category(
 
 
 @book_router.get("/list/genre/{genre}", status_code=status.HTTP_200_OK)
-async def list_books_by_genre(genre: str, session: AsyncSession = Depends(get_session)):
-    books = await book_service.list_books_by_genre(genre, session)
+async def list_books_by_genre(
+    request: Request, genre: str, session: AsyncSession = Depends(get_session)
+):
+    page = int(request.query_params.get("page", 1))
+
+    books = await book_service.list_books_by_genre(genre, page, session)
 
     return JSONResponse(
         status_code=status.HTTP_200_OK,
@@ -477,9 +493,11 @@ async def list_books_by_genre(genre: str, session: AsyncSession = Depends(get_se
 
 @book_router.get("/list/author/{author}", status_code=status.HTTP_200_OK)
 async def list_books_by_author(
-    author: str, session: AsyncSession = Depends(get_session)
+    request: Request, author: str, session: AsyncSession = Depends(get_session)
 ):
-    books = await book_service.list_books_by_author(author, session)
+    page = int(request.query_params.get("page", 1))
+
+    books = await book_service.list_books_by_author(author, page, session)
 
     return JSONResponse(
         status_code=status.HTTP_200_OK,

src/books/service.py

@@ -44,8 +44,10 @@ async def get_book_category_by_category(self, category: str, session: AsyncSessi
         book_category = result.scalars().first()
         return book_category
 
-    async def list_book_categories(self, session: AsyncSession):
-        result = await session.execute(select(BookCategory))
+    async def list_book_categories(self, page: int, session: AsyncSession):
+        result = await session.execute(
+            select(BookCategory).offset((page - 1) * 10).limit(10)
+        )
         book_categories = result.scalars().all()
         return book_categories
 
@@ -88,8 +90,10 @@ async def get_book_genre_by_genre(self, genre: str, session: AsyncSession):
         book_genre = result.scalars().first()
         return book_genre
 
-    async def list_book_genres(self, session: AsyncSession):
-        result = await session.execute(select(BookGenre))
+    async def list_book_genres(self, page: int, session: AsyncSession):
+        result = await session.execute(
+            select(BookGenre).offset((page - 1) * 10).limit(10)
+        )
         book_genres = result.scalars().all()
         return book_genres
 
@@ -136,26 +140,38 @@ async def get_book_by_title(self, title: str, session: AsyncSession):
         book = result.scalars().first()
         return book
 
-    async def list_books(self, session: AsyncSession):
-        result = await session.execute(select(Book))
+    async def list_books(self, page: int, session: AsyncSession):
+        result = await session.execute(select(Book).offset((page - 1) * 10).limit(10))
         books = result.scalars().all()
         return books
 
-    async def list_books_by_category(self, category: str, session: AsyncSession):
+    async def list_books_by_category(
+        self, category: str, page: int, session: AsyncSession
+    ):
         result = await session.execute(
-            select(Book).where(Book.categories.contains(category))
+            select(Book).where(
+                Book.categories.contains(category).offset((page - 1) * 10).limit(10)
+            )
         )
         books = result.scalars().all()
         return books
 
-    async def list_books_by_genre(self, genre: str, session: AsyncSession):
-        result = await session.execute(select(Book).where(Book.genres.contains(genre)))
+    async def list_books_by_genre(self, genre: str, page: int, session: AsyncSession):
+        result = await session.execute(
+            select(Book)
+            .where(Book.genres.contains(genre))
+            .offset((page - 1) * 10)
+            .limit(10)
+        )
         books = result.scalars().all()
         return books
 
-    async def list_books_by_author(self, author: str, session: AsyncSession):
+    async def list_books_by_author(self, author: str, page: int, session: AsyncSession):
         result = await session.execute(
-            select(Book).where(Book.authors.contains(author))
+            select(Book)
+            .where(Book.authors.contains(author))
+            .offset((page - 1) * 10)
+            .limit(10)
         )
         books = result.scalars().all()
         return books