Slugy, webp, hromada zmen

src/graphql_schema/entities/copilot.py

@@ -4,6 +4,7 @@ from strawberry.types import Info
 from database import models
 from decorators.error_logging import error_logging
 from decorators.endpoints import authenticated_user_only, allow_public
+from graphql_schema.entities.helpers.detail import get_detail_filters
 from graphql_schema.entities.resolvers.base import BaseMutationResolver
 from graphql_schema.entities.resolvers.copilot import CopilotQueryResolver
 from graphql_schema.entities.types.mutation_input import CreateCopilotInput, EditCopilotInput
@@ -21,11 +22,22 @@ class CopilotQueries:
     @strawberry.field()
     @error_logging
     @allow_public
-    async def copilot(root, info: Info, id: int, pilot_username: Optional[str] = None) -> Copilot:
+    async def copilot(
+            root, info: Info,
+            id: Optional[int] = None,
+            url_slug: Optional[str] = None,
+            pilot_username: Optional[str] = None,
+            public: Optional[bool] = False
+    ) -> Copilot:
+
+        filter_params = get_detail_filters(id, url_slug)
+        if pilot_username:
+            filter_params['pilot_username'] = pilot_username
+
         return await CopilotQueryResolver().get_one(
-            id,
             user_id=info.context.user_id,
-            pilot_username=pilot_username
+            only_public=public,
+            **filter_params
         )
 
 
@@ -35,10 +47,7 @@ class CopilotMutation:
     @error_logging
     @authenticated_user_only()
     async def create_copilot(root, info, input: CreateCopilotInput) -> Copilot:
-        return await BaseMutationResolver(Copilot, models.Copilot).create(
-            data=input.to_dict(),
-            user_id=info.context.user_id
-        )
+        return await BaseMutationResolver(Copilot, models.Copilot).create(info.context, data=input)
 
     @strawberry.mutation
     @error_logging