37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
from typing import List, Optional
|
|
import strawberry
|
|
from database import models
|
|
from decorators.endpoints import authenticated_user_only
|
|
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
|
|
|
|
|
|
@strawberry.type
|
|
class CopilotQueries:
|
|
@strawberry.field()
|
|
@authenticated_user_only()
|
|
async def copilots(root, info) -> List[Copilot]:
|
|
return await BaseQueryResolver(Copilot, models.Copilot).get_list(info.context.user_id)
|
|
|
|
@strawberry.field()
|
|
async def copilot(root, info, id: int, pilot_username: Optional[str] = None) -> Copilot:
|
|
params = {}
|
|
return await BaseQueryResolver(Copilot, models.Copilot).get_one(id, user_id=info.context.user_id)
|
|
|
|
|
|
@strawberry.type
|
|
class CopilotMutation:
|
|
@strawberry.mutation
|
|
@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
|
|
)
|
|
|
|
@strawberry.mutation
|
|
@authenticated_user_only()
|
|
async def edit_copilot(root, info, id: int, input: EditCopilotInput) -> Copilot:
|
|
return await BaseMutationResolver(Copilot, models.Copilot).update(id, input, info.context.user_id)
|