Init

fastapi_book_server/app/views/book.py

@@ -0,0 +1,81 @@
+from typing import Union
+
+from fastapi import APIRouter, Depends, HTTPException, status
+
+from fastapi_pagination import Params
+from fastapi_pagination.ext.ormar import paginate
+from app.utils.pagination import CustomPage
+
+from app.models import Book as BookDB, Author as AuthorDB, AuthorAnnotation as AuthorAnnotationDB
+from app.serializers.book import Book, CreateBook, UpdateBook, CreateRemoteBook
+from app.services.book import BookTGRMSearchService, BookCreator
+
+
+book_router = APIRouter(
+    prefix="/api/v1/books",
+    tags=["book"],
+)
+
+
+@book_router.get("/", response_model=CustomPage[Book], dependencies=[Depends(Params)])
+async def get_books():
+    return await paginate(
+        BookDB.objects.select_related("authors")
+    )
+
+
+@book_router.post("/", response_model=Book)
+async def create_book(data: Union[CreateBook, CreateRemoteBook]):
+    book = await BookCreator.create(data)
+
+    return await BookDB.objects.select_related("authors").get(id=book.id)
+
+
+@book_router.get("/{id}", response_model=Book)
+async def get_book(id: int):
+    book = await BookDB.objects.select_related("authors").get_or_none(id=id)
+    
+    if book is None:
+        raise HTTPException(status.HTTP_404_NOT_FOUND)
+
+    return book
+
+
+@book_router.put("/{id}", response_model=Book)
+async def update_book(id: int, data: UpdateBook):
+    book = await BookDB.objects.select_related("authors").get_or_none(id=id)
+
+    if book is None:
+        raise HTTPException(status.HTTP_404_NOT_FOUND)
+
+    for author in list(book.authors):
+        await book.authors.remove(author)
+
+    data_dict = data.dict()
+
+    author_ids = data_dict.pop("authors", [])
+    authors = await AuthorDB.objects.filter(id__in=author_ids).all()
+
+    book = await BookDB.objects.create(
+        **data_dict
+    )
+
+    for author in authors:
+        await book.authors.add(author)
+
+    return book
+
+
+@book_router.get("/{id}/annotation")
+async def get_book_annotation(id: int):
+    annotation = await AuthorAnnotationDB.objects.get(book__id=id)
+
+    if annotation is None:
+        raise HTTPException(status.HTTP_404_NOT_FOUND)
+
+    return annotation
+
+
+@book_router.get("/search/{query}", response_model=CustomPage[Book], dependencies=[Depends(Params)])
+async def search_books(query: str):
+    return await BookTGRMSearchService.get(query)