Fix otaceni fotek, drobny refaktoring

src/graphql_schema/entities/copilot.py

@@ -2,7 +2,6 @@ from typing import List
 import strawberry
 from database import models
 from decorators.endpoints import authenticated_user_only
-from database.transaction import get_session
 from graphql_schema.entities.resolvers.base import BaseQueryResolver, BaseMutationResolver
 from graphql_schema.entities.types.mutation_input import CreateCopilotInput, EditCopilotInput
 from graphql_schema.entities.types.types import Copilot
@@ -22,7 +21,7 @@ class CopilotQueries:
 
 
 @strawberry.type
-class CreateCopilotMutation:
+class CopilotMutation:
     @strawberry.mutation
     @authenticated_user_only()
     async def create_copilot(root, info, input: CreateCopilotInput) -> Copilot:
@@ -31,16 +30,7 @@ class CreateCopilotMutation:
             user_id=info.context.user_id
         )
 
-
-@strawberry.type
-class EditCopilotMutation:
     @strawberry.mutation
     @authenticated_user_only()
     async def edit_copilot(root, info, id: int, input: EditCopilotInput) -> Copilot:
-        async with get_session() as db:
-            copilot = (await db.scalars(
-                BaseQueryResolver(Copilot, models.Copilot).get_query(object_id=id, user_id=info.context.user_id)
-            )).one()
-
-            updated_copilot = await models.Copilot.update(db, obj=copilot, data=input.to_dict())
-            return Copilot(**updated_copilot.as_dict())
+        return await BaseMutationResolver(Copilot, models.Copilot).update(id, input, info.context.user_id)