Add allowed_langs filter

fastapi_book_server/app/views/author.py

@@ -3,7 +3,7 @@ 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
+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
@@ -44,8 +44,8 @@ async def create_author(data: CreateAuthor):
 
 
 @author_router.get("/random", response_model=Author)
-async def get_random_author():
-    author_id = await GetRandomAuthorService.get_random_id()
+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)
 
@@ -87,19 +87,25 @@ async def get_author_annotation(id: int):
 @author_router.get(
     "/{id}/books", response_model=CustomPage[AuthorBook], dependencies=[Depends(Params)]
 )
-async def get_author_books(id: int):
+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)
+        .filter(authors__id=id, lang__in=allowed_langs, is_deleted=False)
         .order_by("title")
     )
 
 
 @author_router.get("/{id}/translated_books", response_model=CustomPage[TranslatedBook])
-async def get_translated_books(id: int):
+async def get_translated_books(
+    id: int, allowed_langs: list[str] = Depends(get_allowed_langs)
+):
     return await paginate(
         BookDB.objects.select_related(["source", "annotations", "translators"]).filter(
-            translations__translator__id=id
+            translations__translator__id=id,
+            lang__in=allowed_langs,
+            is_deleted=False,
         )
     )
 
@@ -107,5 +113,9 @@ async def get_translated_books(id: int):
 @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)
+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
+    )

fastapi_book_server/app/views/book.py

@@ -5,7 +5,7 @@ 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
+from app.depends import check_token, get_allowed_langs
 from app.filters.book import get_book_filter
 from app.models import Author as AuthorDB
 from app.models import Book as BookDB
@@ -50,8 +50,8 @@ async def create_book(data: Union[CreateBook, CreateRemoteBook]):
 
 
 @book_router.get("/random", response_model=BookDetail)
-async def get_random_book():
-    book_id = await GetRandomBookService.get_random_id()
+async def get_random_book(allowed_langs: list[str] = Depends(get_allowed_langs)):
+    book_id = await GetRandomBookService.get_random_id(allowed_langs)
 
     return await BookDB.objects.select_related(SELECT_RELATED_FIELDS).get(id=book_id)
 
@@ -114,5 +114,9 @@ async def get_book_annotation(id: int):
 @book_router.get(
     "/search/{query}", response_model=CustomPage[Book], dependencies=[Depends(Params)]
 )
-async def search_books(query: str, request: Request):
-    return await BookTGRMSearchService.get(query, request.app.state.redis)
+async def search_books(
+    query: str, request: Request, allowed_langs: list[str] = Depends(get_allowed_langs)
+):
+    return await BookTGRMSearchService.get(
+        query, request.app.state.redis, allowed_langs
+    )

fastapi_book_server/app/views/sequence.py

@@ -3,7 +3,7 @@ from fastapi import APIRouter, Depends, Request
 from fastapi_pagination import Params
 from fastapi_pagination.ext.ormar import paginate
 
-from app.depends import check_token
+from app.depends import check_token, get_allowed_langs
 from app.models import Book as BookDB
 from app.models import Sequence as SequenceDB
 from app.serializers.sequence import Book as SequenceBook
@@ -27,8 +27,8 @@ async def get_sequences():
 
 
 @sequence_router.get("/random", response_model=Sequence)
-async def get_random_sequence():
-    sequence_id = await GetRandomSequenceService.get_random_id()
+async def get_random_sequence(allowed_langs: list[str] = Depends(get_allowed_langs)):
+    sequence_id = await GetRandomSequenceService.get_random_id(allowed_langs)
 
     return await SequenceDB.objects.get(id=sequence_id)
 
@@ -43,12 +43,14 @@ async def get_sequence(id: int):
     response_model=CustomPage[SequenceBook],
     dependencies=[Depends(Params)],
 )
-async def get_sequence_books(id: int):
+async def get_sequence_books(
+    id: int, allowed_langs: list[str] = Depends(get_allowed_langs)
+):
     return await paginate(
         BookDB.objects.select_related(
             ["source", "annotations", "authors", "translators"]
         )
-        .filter(sequences__id=id)
+        .filter(sequences__id=id, lang__in=allowed_langs, is_deleted=False)
         .order_by("sequences__booksequences__position")
     )
 
@@ -63,5 +65,9 @@ async def create_sequence(data: CreateSequence):
     response_model=CustomPage[Sequence],
     dependencies=[Depends(Params)],
 )
-async def search_sequences(query: str, request: Request):
-    return await SequenceTGRMSearchService.get(query, request.app.state.redis)
+async def search_sequences(
+    query: str, request: Request, allowed_langs: list[str] = Depends(get_allowed_langs)
+):
+    return await SequenceTGRMSearchService.get(
+        query, request.app.state.redis, allowed_langs
+    )