mirror of
https://github.com/evgen-app/evg-app-back.git
synced 2024-11-25 10:13:45 +03:00
78 lines
2.0 KiB
Python
78 lines
2.0 KiB
Python
import graphene
|
|
|
|
from django.contrib.auth.models import User
|
|
|
|
from .models import TimeEvent
|
|
|
|
from .graphqlTypes import UserType, TimeEventType
|
|
|
|
from graphql_relay.node.node import from_global_id
|
|
|
|
class CreateTimeEvent(graphene.Mutation):
|
|
class Arguments:
|
|
user_id = graphene.ID()
|
|
deadline = graphene.DateTime()
|
|
|
|
ok = graphene.Boolean()
|
|
|
|
|
|
@classmethod
|
|
def mutate(cls, root, info, user_id, deadline):
|
|
user_id = from_global_id(user_id)[1]
|
|
user = User.objects.get(id=user_id)
|
|
|
|
TimeEvent.objects.create(user = user, deadline = deadline)
|
|
|
|
return CreateTimeEvent(ok=True)
|
|
|
|
|
|
class DeleteTimeEvent(graphene.Mutation):
|
|
class Arguments:
|
|
id = graphene.ID()
|
|
|
|
|
|
ok = graphene.Boolean()
|
|
|
|
|
|
@classmethod
|
|
def mutate(cls, root, info, id):
|
|
TimeEventId = from_global_id(id)[1]
|
|
|
|
TimeEvent.objects.get(id=TimeEventId).delete()
|
|
|
|
return DeleteTimeEvent(ok=True)
|
|
|
|
|
|
class ChangeTimeEventStatus(graphene.Mutation):
|
|
class Arguments:
|
|
id = graphene.ID()
|
|
status = graphene.Boolean(required=False)
|
|
deadline = graphene.DateTime(required = False)
|
|
|
|
timeEvent = graphene.Field(TimeEventType)
|
|
|
|
@classmethod
|
|
def mutate(cls, root, info, id, status=None, deadline=None):
|
|
TimeEventId = from_global_id(id)[1]
|
|
print(TimeEventId, TimeEvent.objects.all())
|
|
timeEvent = TimeEvent.objects.get(id=TimeEventId)
|
|
print(timeEvent)
|
|
timeEvent.deadline = deadline or timeEvent.deadline
|
|
timeEvent.checked = status or timeEvent.checked
|
|
|
|
timeEvent.save()
|
|
|
|
return ChangeTimeEventStatus(timeEvent=timeEvent)
|
|
|
|
class RegUser(graphene.Mutation):
|
|
class Arguments:
|
|
username = graphene.String()
|
|
password = graphene.String()
|
|
|
|
user = graphene.Field(UserType)
|
|
|
|
@classmethod
|
|
def mutate(cls, root, info, username, password):
|
|
user = User.objects.create_user(username=username, password=password)
|
|
|
|
return RegUser(user=user) |