mirror of
https://github.com/flibusta-apps/book_library_server.git
synced 2025-12-06 15:15:36 +01:00
141 lines
4.1 KiB
Python
141 lines
4.1 KiB
Python
from fastapi import APIRouter, Depends, Request, HTTPException, status
|
|
|
|
from fastapi_pagination import Params
|
|
from fastapi_pagination.ext.ormar import paginate
|
|
|
|
from app.depends import check_token, get_allowed_langs
|
|
from app.models import Author as AuthorDB
|
|
from app.models import AuthorAnnotation as AuthorAnnotationDB
|
|
from app.models import 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, GetRandomAuthorService
|
|
from app.services.translator import TranslatorTGRMSearchService
|
|
from app.utils.pagination import CustomPage
|
|
|
|
|
|
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(allowed_langs: list[str] = Depends(get_allowed_langs)):
|
|
author_id = await GetRandomAuthorService.get_random_id(allowed_langs)
|
|
|
|
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, allowed_langs: list[str] = Depends(get_allowed_langs)
|
|
):
|
|
return await paginate(
|
|
BookDB.objects.select_related(["source", "annotations", "translators"])
|
|
.filter(authors__id=id, lang__in=allowed_langs, is_deleted=False)
|
|
.order_by("title")
|
|
)
|
|
|
|
|
|
@author_router.get(
|
|
"/search/{query}", response_model=CustomPage[Author], dependencies=[Depends(Params)]
|
|
)
|
|
async def search_authors(
|
|
query: str, request: Request, allowed_langs: list[str] = Depends(get_allowed_langs)
|
|
):
|
|
return await AuthorTGRMSearchService.get(
|
|
query, request.app.state.redis, allowed_langs
|
|
)
|
|
|
|
|
|
translator_router = APIRouter(
|
|
prefix="/api/v1/translators",
|
|
tags=["author"],
|
|
dependencies=[Depends(check_token)],
|
|
)
|
|
|
|
|
|
@translator_router.get("/{id}/books", response_model=CustomPage[TranslatedBook])
|
|
async def get_translated_books(
|
|
id: int, allowed_langs: list[str] = Depends(get_allowed_langs)
|
|
):
|
|
return await paginate(
|
|
BookDB.objects.select_related(["source", "annotations", "authors"]).filter(
|
|
translators__id=id,
|
|
lang__in=allowed_langs,
|
|
is_deleted=False,
|
|
)
|
|
)
|
|
|
|
|
|
@translator_router.get(
|
|
"/search/{query}", response_model=CustomPage[Author], dependencies=[Depends(Params)]
|
|
)
|
|
async def search_translators(
|
|
query: str, request: Request, allowed_langs: list[str] = Depends(get_allowed_langs)
|
|
):
|
|
return await TranslatorTGRMSearchService.get(
|
|
query, request.app.state.redis, allowed_langs
|
|
)
|