feat(fastapi): cover CreateRelation

Author: emptybutton

src/tgdb/presentation/fastapi/common/error_handling.py

@@ -1,8 +1,14 @@
 from fastapi import FastAPI, Response, status
 from fastapi.responses import JSONResponse
 
-from tgdb.application.common.ports.relations import NoRelationError
-from tgdb.presentation.fastapi.common.schemas.error import NoRelationSchema
+from tgdb.application.common.ports.relations import (
+    NoRelationError,
+    NotUniqueRelationNumberError,
+)
+from tgdb.presentation.fastapi.common.schemas.error import (
+    NoRelationSchema,
+    NotUniqueRelationNumberSchema,
+)
 from tgdb.presentation.fastapi.horizon.error_handling import (
     add_horizon_error_handling,
 )
@@ -22,3 +28,12 @@ def _(_: object) -> Response:
             schema.model_dump(mode="json", by_alias=True),
             status_code=status.HTTP_404_NOT_FOUND,
         )
+
+    @app.exception_handler(NotUniqueRelationNumberError)
+    def _(_: object) -> Response:
+        schema = NotUniqueRelationNumberSchema()
+
+        return JSONResponse(
+            schema.model_dump(mode="json", by_alias=True),
+            status_code=status.HTTP_409_CONFLICT,
+        )

src/tgdb/presentation/fastapi/common/schemas/error.py

@@ -8,4 +8,8 @@ class NoRelationSchema(BaseModel):
     Relation was not created.
     """
 
-    type: Literal["NoRelation"] = "NoRelation"
+    type: Literal["noRelation"] = "noRelation"
+
+
+class NotUniqueRelationNumberSchema(BaseModel):
+    type: Literal["notUniqueRelationNumber"] = "notUniqueRelationNumber"

src/tgdb/presentation/fastapi/relation/routers.py

@@ -1,3 +1,6 @@
+from tgdb.presentation.fastapi.relation.routes.create_relation import (
+    create_relation_router,
+)
 from tgdb.presentation.fastapi.relation.routes.view_all_relations import (
     view_all_relations_router,
 )
@@ -7,6 +10,7 @@
 
 
 relation_routers = (
+    create_relation_router,
     view_all_relations_router,
     view_relation_router,
 )

src/tgdb/presentation/fastapi/relation/routes/create_relation.py

@@ -0,0 +1,46 @@
+from typing import Annotated
+
+from dishka.integrations.fastapi import FromDishka, inject
+from fastapi import APIRouter, Query, status
+from fastapi.responses import Response
+from pydantic import BaseModel, Field, PositiveInt
+
+from tgdb.application.relation.create_relation import CreateRelation
+from tgdb.entities.numeration.number import Number
+from tgdb.presentation.fastapi.common.schemas.error import (
+    NotUniqueRelationNumberSchema,
+)
+from tgdb.presentation.fastapi.common.tags import Tag
+from tgdb.presentation.fastapi.relation.schemas.schema import SchemaSchema
+
+
+create_relation_router = APIRouter()
+
+
+class CreateRelationSchema(BaseModel):
+    schema_: SchemaSchema = Field(alias="schema")
+
+
+@create_relation_router.post(
+    "/relations/{relation_number}",
+    status_code=status.HTTP_201_CREATED,
+    responses={
+        status.HTTP_201_CREATED: {"content": None},
+        status.HTTP_409_CONFLICT: {"model": NotUniqueRelationNumberSchema},
+    },
+    summary="Create relation",
+    description="Create relation with unique number.",
+    tags=[Tag.relation],
+)
+@inject
+async def _(
+    create_relation: FromDishka[CreateRelation],
+    relation_number: Annotated[PositiveInt, Query(alias="relationNumber")],
+    request_body: CreateRelationSchema,
+) -> Response:
+    await create_relation(
+        Number(relation_number),
+        request_body.schema_.decoded(),
+    )
+
+    return Response(status_code=status.HTTP_201_CREATED)