Prepis strankovani na extension, sjednoceni volani resolveru

src/graphql_schema/entities/resolvers/aircraft.py

@@ -3,6 +3,7 @@ from typing import Set, Optional
 from sqlalchemy import and_
 from database import models
 from database.transaction import get_session
+from graphql_schema.context import GraphQLContext
 from graphql_schema.entities.helpers.combobox import handle_combobox_save
 from graphql_schema.entities.resolvers.base import BaseMutationResolver, BaseQueryResolver
 from graphql_schema.entities.types.mutation_input import EditAircraftInput, CreateAircraftInput
@@ -60,7 +61,7 @@ class AircraftMutationResolver(BaseMutationResolver):
     def __init__(self):
         super().__init__(graphql_type=Aircraft, model=models.Aircraft)
 
-    async def create(self, context, data: CreateAircraftInput) -> Aircraft:
+    async def create(self, context: GraphQLContext, data: CreateAircraftInput) -> Aircraft:
         input_data = data.to_dict()
 
         async with get_session() as db:
@@ -75,7 +76,7 @@ class AircraftMutationResolver(BaseMutationResolver):
 
             return await self._do_create(db, data=input_data)
 
-    async def update(self, context, id: int, data: EditAircraftInput) -> Aircraft:
+    async def update(self, context: GraphQLContext, id: int, data: EditAircraftInput) -> Aircraft:
         update_data = data.to_dict()
         async with get_session() as db:
             if data.organization: