34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
import graphene
|
|
from graphene import relay
|
|
from api.utils import get_object
|
|
from objectives.models import ObjectiveProgressStatus, Objective
|
|
from objectives.schema import ObjectiveNode
|
|
|
|
|
|
class UpdateObjectiveProgress(relay.ClientIDMutation):
|
|
class Input:
|
|
id = graphene.ID(required=True, description="The ID of the objective")
|
|
done = graphene.Boolean(required=True)
|
|
|
|
errors = graphene.List(graphene.String)
|
|
objective = graphene.Field(ObjectiveNode)
|
|
|
|
@classmethod
|
|
def mutate_and_get_payload(cls, root, info, **kwargs):
|
|
objective_id = kwargs.get('id')
|
|
done = kwargs.get('done')
|
|
objective = get_object(Objective, objective_id) # info.context.user = django user
|
|
objective_progress, created = ObjectiveProgressStatus.objects.get_or_create(
|
|
objective=objective,
|
|
user=info.context.user
|
|
)
|
|
|
|
objective_progress.done = done
|
|
objective_progress.save()
|
|
|
|
return cls(objective=objective)
|
|
|
|
|
|
class ObjectiveMutations:
|
|
update_objective_progress = UpdateObjectiveProgress.Field()
|