diff --git a/server/objectives/schema.py b/server/objectives/schema.py index ca9e8488..cef4e1dd 100644 --- a/server/objectives/schema.py +++ b/server/objectives/schema.py @@ -31,6 +31,9 @@ class ObjectiveNode(DjangoObjectType): filter_fields = ['text'] interfaces = (relay.Node,) + def resolve_objective_progress(self, info, **kwargs): + return self.objective_progress.filter(user=info.context.user) + class ObjectiveProgressStatusNode(DjangoObjectType): pk = graphene.Int() @@ -45,4 +48,5 @@ class ObjectiveProgressStatusNode(DjangoObjectType): class ObjectivesQuery(object): + objective_group = relay.Node.Field(ObjectiveGroupNode) objective_groups = DjangoFilterConnectionField(ObjectiveGroupNode)