mirror of
https://github.com/flibusta-apps/book_library_server.git
synced 2025-12-06 07:05:36 +01:00
101 lines
3.2 KiB
Python
101 lines
3.2 KiB
Python
from random import choice as random_choice
|
|
|
|
from fastapi import APIRouter, Depends, Request, HTTPException, status
|
|
|
|
from fastapi_pagination import Params
|
|
from fastapi_pagination.ext.ormar import paginate
|
|
from app.utils.pagination import CustomPage
|
|
|
|
from app.models import Author as AuthorDB, AuthorAnnotation as AuthorAnnotationDB, Book as BookDB
|
|
from app.serializers.author import Author, CreateAuthor, UpdateAuthor, AuthorBook, TranslatedBook
|
|
from app.serializers.author_annotation import AuthorAnnotation
|
|
from app.services.author import AuthorTGRMSearchService
|
|
from app.depends import check_token
|
|
|
|
|
|
author_router = APIRouter(
|
|
prefix="/api/v1/authors",
|
|
tags=["author"],
|
|
dependencies=[Depends(check_token)],
|
|
)
|
|
|
|
|
|
|
|
PREFETCH_RELATED = ["source", "annotations"]
|
|
|
|
|
|
@author_router.get("/", response_model=CustomPage[Author], dependencies=[Depends(Params)])
|
|
async def get_authors():
|
|
return await paginate(
|
|
AuthorDB.objects.prefetch_related(PREFETCH_RELATED)
|
|
)
|
|
|
|
|
|
@author_router.post("/", response_model=Author, dependencies=[Depends(Params)])
|
|
async def create_author(data: CreateAuthor):
|
|
author = await AuthorDB.objects.create(
|
|
**data.dict()
|
|
)
|
|
|
|
return await AuthorDB.objects.prefetch_related(PREFETCH_RELATED).get(id=author.id)
|
|
|
|
|
|
@author_router.get("/random", response_model=Author)
|
|
async def get_random_author():
|
|
author_ids: list[int] = await AuthorDB.objects.values_list("id", flatten=True)
|
|
|
|
author_id = random_choice(author_ids)
|
|
|
|
return await AuthorDB.objects.prefetch_related(PREFETCH_RELATED).get(id=author_id)
|
|
|
|
|
|
@author_router.get("/{id}", response_model=Author)
|
|
async def get_author(id: int):
|
|
author = await AuthorDB.objects.prefetch_related(PREFETCH_RELATED).get_or_none(id=id)
|
|
|
|
if author is None:
|
|
raise HTTPException(status.HTTP_404_NOT_FOUND)
|
|
|
|
return author
|
|
|
|
|
|
@author_router.put("/{id}", response_model=Author)
|
|
async def update_author(id: int, data: UpdateAuthor):
|
|
author = await AuthorDB.objects.get_or_none(id=id)
|
|
|
|
if author is None:
|
|
raise HTTPException(status.HTTP_404_NOT_FOUND)
|
|
|
|
author.update_from_dict(data.dict())
|
|
|
|
return await author.save()
|
|
|
|
|
|
@author_router.get("/{id}/annotation", response_model=AuthorAnnotation)
|
|
async def get_author_annotation(id: int):
|
|
annotation = await AuthorAnnotationDB.objects.get_or_none(author__id=id)
|
|
|
|
if annotation is None:
|
|
raise HTTPException(status.HTTP_404_NOT_FOUND)
|
|
|
|
return annotation
|
|
|
|
|
|
@author_router.get("/{id}/books", response_model=CustomPage[AuthorBook], dependencies=[Depends(Params)])
|
|
async def get_author_books(id: int):
|
|
return await paginate(
|
|
BookDB.objects.select_related(["source", "annotations", "translators"]).filter(authors__id=id).order_by('title')
|
|
)
|
|
|
|
|
|
@author_router.get("/{id}/translated_books", response_model=CustomPage[TranslatedBook])
|
|
async def get_translated_books(id: int):
|
|
return await paginate(
|
|
BookDB.objects.select_related(["source", "annotations", "translators"]).filter(translations__translator__id=id)
|
|
)
|
|
|
|
|
|
@author_router.get("/search/{query}", response_model=CustomPage[Author], dependencies=[Depends(Params)])
|
|
async def search_authors(query: str, request: Request):
|
|
return await AuthorTGRMSearchService.get(query, request.app.state.redis)
|