Fix error for user without existing submission
This commit is contained in:
parent
f4fb15e428
commit
0612daa05f
|
|
@ -21,7 +21,7 @@ class AssignmentNode(DjangoObjectType):
|
||||||
interfaces = (relay.Node,)
|
interfaces = (relay.Node,)
|
||||||
|
|
||||||
def resolve_submission(self, info, **kwargs):
|
def resolve_submission(self, info, **kwargs):
|
||||||
return self.submissions.get(student=info.context.user)
|
return self.submissions.filter(student=info.context.user).first() # returns None if it doesn't exist yet
|
||||||
|
|
||||||
def resolve_submissions(self, info, **kwargs):
|
def resolve_submissions(self, info, **kwargs):
|
||||||
return self.submissions.all()
|
return self.submissions.all()
|
||||||
Loading…
Reference in New Issue