Slugy, webp, hromada zmen

src/graphql_schema/entities/aircraft.py

@@ -29,12 +29,25 @@ class AircraftQueries:
     @strawberry.field()
     @error_logging
     @allow_public
-    async def aircraft(root, info, id: int, public: Optional[bool] = False) -> Aircraft:
+    async def aircraft(
+            root, info,
+            id: Optional[int] = None,
+            call_sign: Optional[str] = None,
+            public: Optional[bool] = False
+    ) -> Aircraft:
+        filter_params = {}
+
+        if id:
+            filter_params['object_id'] = id
+
+        if call_sign:
+            filter_params['call_sign'] = call_sign
+
         return await AircraftQueryResolver().get_one(
-            id,
             user_id=info.context.user_id,
             organization_ids=info.context.organization_ids,
-            public=public
+            only_public=public,
+            **filter_params
         )
 
 
@@ -44,13 +57,13 @@ class AircraftMutation:
     @error_logging
     @authenticated_user_only()
     async def create_aircraft(root, info, input: CreateAircraftInput) -> Aircraft:
-        return await AircraftMutationResolver().create_new(input, info.context.user_id)
+        return await AircraftMutationResolver().create(info.context, input)
 
     @strawberry.mutation
     @error_logging
     @authenticated_user_only()
     async def edit_aircraft(root, info, id: int, input: EditAircraftInput) -> Aircraft:
-        return await AircraftMutationResolver().edit(id, user_id=info.context.user_id, data=input)
+        return await AircraftMutationResolver().update(id, user_id=info.context.user_id, data=input)
 
     @strawberry.mutation
     @authenticated_user_only()