Zakladni verze planovace

src/graphql_schema/entities/resolvers/flight_plan.py

@@ -1,7 +1,8 @@
+from sqlalchemy import delete
 from database import models
 from database.transaction import get_session
 from graphql_schema.entities.resolvers.base import BaseMutationResolver
-from graphql_schema.entities.types.mutation_input import CreateFlightPlanInput
+from graphql_schema.entities.types.mutation_input import CreateFlightPlanInput, EditFlightPlanInput
 from graphql_schema.entities.types.types import FlightPlan
 
 
@@ -9,22 +10,44 @@ class FlightPlanMutationResolver(BaseMutationResolver):
     def __init__(self):
         super().__init__(graphql_type=FlightPlan, model=models.FlightPlan)
 
+    @staticmethod
+    async def save_markers(db, flight_plan: models.FlightPlan, markers: list):
+        position = 0
+
+        for marker in markers:
+            await models.FlightPlanMarker.create(db, data={
+                "position": position,
+                "flight_plan_id": flight_plan.id,
+                "type": marker.type,
+                "name": marker.name,
+                "gps_latitude": marker.gps_latitude,
+                "gps_longitude": marker.gps_longitude
+            })
+            position += 1
+
+    @staticmethod
+    async def reset_plan_markers(db, flight_plan: models.FlightPlan):
+        await db.execute(
+            delete(models.FlightPlanMarker)
+            .filter(models.FlightPlanMarker.flight_plan_id == flight_plan.id)
+        )
+
     async def create(self, context, data: CreateFlightPlanInput) -> FlightPlan:
         input_data = data.to_dict()
         input_data['created_by_id'] = context.user_id
 
         async with get_session() as db:
             flight_plan = await self._do_create(db, data=input_data)
+            await self.save_markers(db, flight_plan, data.markers)
+            return flight_plan
 
-            position = 0
-            for marker in data.markers:
-                await models.FlightPlanMarker.create(db, data={
-                    "position": position,
-                    "flight_plan_id": flight_plan.id,
-                    "type": marker.type,
-                    "name": marker.name,
-                    "gps_latitude": marker.gps_latitude,
-                    "gps_longitude": marker.gps_longitude
-                })
+    async def update(self, id: int, data: EditFlightPlanInput, user_id: int) -> FlightPlan:
+        input_data = data.to_dict()
+
+        async with get_session() as db:
+            flight_plan = await self._do_update(db, id, input_data)
+
+            await self.reset_plan_markers(db, flight_plan)
+            await self.save_markers(db, flight_plan, data.markers)
 
             return flight_plan