From 7a09621a014f22b2e08108e1ab7fe532ebdcf67b Mon Sep 17 00:00:00 2001 From: Ramon Wenger Date: Tue, 13 Nov 2018 19:04:34 +0100 Subject: [PATCH] Add some logging --- server/api/graphene_wagtail.py | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/server/api/graphene_wagtail.py b/server/api/graphene_wagtail.py index 4002fb65..01700a39 100644 --- a/server/api/graphene_wagtail.py +++ b/server/api/graphene_wagtail.py @@ -1,5 +1,6 @@ # mysite/api/graphene_wagtail.py # Taken from https://github.com/patrick91/wagtail-ql/blob/master/backend/graphene_utils/converter.py and slightly adjusted +import logging from graphene.types import Scalar from graphene_django.converter import convert_django_field @@ -10,6 +11,8 @@ from wagtail.images.models import Image from assignments.models import Assignment from basicknowledge.models import BasicKnowledge +logger = logging.getLogger(__name__) + class GenericStreamFieldType(Scalar): @staticmethod @@ -29,13 +32,17 @@ class GenericStreamFieldType(Scalar): d['value'] = value if _type == 'assignment': _value = d['value'] - assignment = Assignment.objects.get(pk=_value['assignment_id']) - value = { - 'title': assignment.title, - 'assignment': assignment.assignment, - 'id': to_global_id('AssignmentNode', assignment.pk) - } - d['value'] = value + assignment_id = _value['assignment_id'] + try: + assignment = Assignment.objects.get(pk=assignment_id) + value = { + 'title': assignment.title, + 'assignment': assignment.assignment, + 'id': to_global_id('AssignmentNode', assignment.pk) + } + d['value'] = value + except Assignment.DoesNotExist: + logger.error('Assignment {} does not exist'.format(assignment_id)) if _type == 'basic_knowledge': _value = d['value'] basic_knowledge = BasicKnowledge.objects.get(pk=_value['basic_knowledge'])