44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
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 CreateEventInput, EditEventInput
|
|
from graphql_schema.entities.types.types import Event
|
|
|
|
|
|
@strawberry.type
|
|
class EventQueries:
|
|
@strawberry.field()
|
|
@authenticated_user_only()
|
|
async def events(root, info) -> List[Event]:
|
|
return await BaseQueryResolver(Event, models.Event).get_list(
|
|
info.context.user_id,
|
|
order_by=[models.Event.date_from.desc(), models.Event.id.desc()]
|
|
)
|
|
|
|
@strawberry.field()
|
|
@authenticated_user_only()
|
|
async def event(root, info, id: int) -> Event:
|
|
return await BaseQueryResolver(Event, models.Event).get_one(id, user_id=info.context.user_id)
|
|
|
|
|
|
@strawberry.type
|
|
class EventMutation:
|
|
@strawberry.mutation
|
|
@authenticated_user_only()
|
|
async def create_event(root, info, input: CreateEventInput) -> Event:
|
|
return await BaseMutationResolver(Event, models.Event).create(input.to_dict(), info.context.user_id)
|
|
|
|
@strawberry.mutation
|
|
@authenticated_user_only()
|
|
async def edit_event(root, info, id: int, input: EditEventInput) -> Event:
|
|
async with get_session() as db:
|
|
event = (await db.scalars(
|
|
BaseQueryResolver(Event, models.Event).get_query(user_id=info.context.user_id, object_id=id)
|
|
)).one()
|
|
|
|
updated_event = await models.Event.update(db, obj=event, data=input.to_dict())
|
|
return Event(**updated_event.as_dict())
|