draft: crud pro planovac

This commit is contained in:
Michal Kváček
2024-03-25 19:04:24 +01:00
parent 01f8a616bf
commit 63e2eed8c2
8 changed files with 150 additions and 21 deletions
@@ -0,0 +1,30 @@
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.types import FlightPlan
class FlightPlanMutationResolver(BaseMutationResolver):
def __init__(self):
super().__init__(graphql_type=FlightPlan, model=models.FlightPlan)
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)
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
})
return flight_plan