Stahovani pocasi, editace kopilota, nahravani GPX k letu

src/graphql_schema/entities/copilot.py

@@ -3,13 +3,14 @@ import strawberry
 from sqlalchemy import select
 from database import models
 from graphql_schema.dataloaders.flight import flights_by_copilot_dataloader
-from graphql_schema.sqlalchemy_to_strawberry_type import strawberry_sqlalchemy_type
+from graphql_schema.sqlalchemy_to_strawberry_type import strawberry_sqlalchemy_type, strawberry_sqlalchemy_input
 
 if TYPE_CHECKING:
     from .flight import Flight
 
+
 @strawberry_sqlalchemy_type(models.Copilot)
-class CopilotType:
+class Copilot:
     async def load_flights(root):
         return await flights_by_copilot_dataloader.load(root.id)
 
@@ -28,14 +29,49 @@ def get_base_query(user_id: int):
 @strawberry.type
 class CopilotQueries:
     @strawberry.field
-    async def copilots(root, info) -> List[CopilotType]:
+    async def copilots(root, info) -> List[Copilot]:
         return (await info.context.db.scalars(
             get_base_query(info.context.user_id)
         )).all()
 
     @strawberry.field
-    async def copilot(root, info, id: int) -> CopilotType:
+    async def copilot(root, info, id: int) -> Copilot:
         return (await info.context.db.scalars(
             get_base_query(info.context.user_id)
             .filter(models.Copilot.id == id)
         )).one()
+
+
+@strawberry.type
+class CreateCopilotMutation:
+
+    @strawberry_sqlalchemy_input(model=models.Copilot, exclude_fields=["id"])
+    class CreateCopilotInput:
+        pass
+
+    @strawberry.mutation
+    async def create_copilot(root, info, input: CreateCopilotInput) -> Copilot:
+        input_data = input.to_dict()
+        return await models.Copilot.create(
+            info.context.db,
+            data=dict(
+                **input_data,
+                created_by_id=info.context.user_id,
+            )
+        )
+
+
+@strawberry.type
+class EditCopilotMutation:
+
+    @strawberry_sqlalchemy_input(model=models.Copilot, exclude_fields=["id"])
+    class EditCopilotInput:
+        pass
+
+    @strawberry.mutation
+    async def edit_copilot(root, info, id: int, input: EditCopilotInput) -> Copilot:
+        copilot = (await info.context.db.scalars(
+            get_base_query(info.context.user_id).filter(models.Copilot.id == id)
+        )).one()
+
+        return await models.Copilot.update(info.context.db, obj=copilot, data=input.to_dict())