mirror of
https://github.com/flibusta-apps/book_library_server.git
synced 2025-12-06 15:15:36 +01:00
97 lines
2.7 KiB
Python
97 lines
2.7 KiB
Python
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
|
|
from app.depends import check_token
|
|
|
|
|
|
book_router = APIRouter(
|
|
prefix="/api/v1/books",
|
|
tags=["book"],
|
|
dependencies=[Depends(check_token)],
|
|
)
|
|
|
|
|
|
@book_router.get("/", response_model=CustomPage[Book], dependencies=[Depends(Params)])
|
|
async def get_books():
|
|
return await paginate(
|
|
BookDB.objects.select_related(["source", "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(["source", "authors"]).get(id=book.id)
|
|
|
|
|
|
@book_router.get("/{id}", response_model=Book)
|
|
async def get_book(id: int):
|
|
book = await BookDB.objects.select_related(["source", "authors"]).get_or_none(id=id)
|
|
|
|
if book is None:
|
|
raise HTTPException(status.HTTP_404_NOT_FOUND)
|
|
|
|
return book
|
|
|
|
|
|
@book_router.get("/remote/{source_id}/{remote_id}", response_model=Book)
|
|
async def get_remote_book(source_id: int, remote_id: int):
|
|
book = await BookDB.objects.select_related(["source", "authors"]).get_or_none(
|
|
source=source_id,
|
|
remote_id=remote_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(["source", "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)
|