diff --git a/fastapi_book_server/app/serializers/author.py b/fastapi_book_server/app/serializers/author.py index f3c88c1..a7cceb5 100644 --- a/fastapi_book_server/app/serializers/author.py +++ b/fastapi_book_server/app/serializers/author.py @@ -3,6 +3,8 @@ from typing import Optional from pydantic import BaseModel +from app.serializers.sequence import Sequence + class Author(BaseModel): id: int @@ -22,6 +24,7 @@ class AuthorBook(BaseModel): available_types: list[str] uploaded: date translators: list[Author] + sequences: list[Sequence] annotation_exists: bool @@ -32,4 +35,5 @@ class TranslatedBook(BaseModel): file_type: str available_types: list[str] authors: list[Author] + sequences: list[Sequence] annotation_exists: bool diff --git a/fastapi_book_server/app/views/author.py b/fastapi_book_server/app/views/author.py index 70ee4c0..68438a6 100644 --- a/fastapi_book_server/app/views/author.py +++ b/fastapi_book_server/app/views/author.py @@ -87,7 +87,7 @@ async def get_author_books( ): return await paginate( BookDB.objects.prefetch_related(["source"]) - .select_related(["annotations", "translators"]) + .select_related(["annotations", "translators", "sequences"]) .filter(authors__id=id, lang__in=allowed_langs, is_deleted=False) .order_by("title") ) @@ -120,7 +120,7 @@ async def get_translated_books( ): return await paginate( BookDB.objects.prefetch_related(["source"]) - .select_related(["annotations", "authors"]) + .select_related(["annotations", "authors", "sequences"]) .filter( translators__id=id, lang__in=allowed_langs,