Add genres api

fastapi_book_server/app/views/genre.py

@@ -0,0 +1,54 @@
+from fastapi import APIRouter, Depends, HTTPException, status
+
+from fastapi_pagination import Params, Page
+from fastapi_pagination.ext.ormar import paginate
+
+from app.depends import check_token
+from app.models import Genre as GenreDB
+from app.serializers.genre import Genre, CreateGenre, UpdateGenre
+
+
+genre_router = APIRouter(
+    prefix="/api/v1/genres", tags=["genres"], dependencies=[Depends(check_token)]
+)
+
+
+PREFETCH_RELATED_FIELDS = ["source"]
+
+
+@genre_router.get("/", response_model=Page[Genre], dependencies=[Depends(Params)])
+async def get_genres():
+    return await paginate(GenreDB.objects.prefetch_related(PREFETCH_RELATED_FIELDS))
+
+
+@genre_router.get("/{id}", response_model=Genre)
+async def get_genre(id: int):
+    genre = await GenreDB.objects.prefetch_related(PREFETCH_RELATED_FIELDS).get_or_none(
+        id=id
+    )
+
+    if genre is None:
+        raise HTTPException(status.HTTP_404_NOT_FOUND)
+
+    return genre
+
+
+@genre_router.post("/", response_model=Genre)
+async def create_genre(data: CreateGenre):
+    return await GenreDB.objects.prefetch_related(PREFETCH_RELATED_FIELDS).create(
+        **data.dict()
+    )
+
+
+@genre_router.put("/{id}", response_model=Genre)
+async def update_genre(id: int, data: UpdateGenre):
+    genre = await GenreDB.objects.prefetch_related(PREFETCH_RELATED_FIELDS).get_or_none(
+        id=id
+    )
+
+    if genre is None:
+        raise HTTPException(status.HTTP_404_NOT_FOUND)
+
+    genre.update_from_dict(data.dict())
+
+    return await genre.save()