Add some logging

This commit is contained in:
Ramon Wenger 2018-11-13 19:04:34 +01:00
parent ee512dc7c2
commit 7a09621a01
1 changed files with 14 additions and 7 deletions

View File

@ -1,5 +1,6 @@
# mysite/api/graphene_wagtail.py # mysite/api/graphene_wagtail.py
# Taken from https://github.com/patrick91/wagtail-ql/blob/master/backend/graphene_utils/converter.py and slightly adjusted # 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.types import Scalar
from graphene_django.converter import convert_django_field from graphene_django.converter import convert_django_field
@ -10,6 +11,8 @@ from wagtail.images.models import Image
from assignments.models import Assignment from assignments.models import Assignment
from basicknowledge.models import BasicKnowledge from basicknowledge.models import BasicKnowledge
logger = logging.getLogger(__name__)
class GenericStreamFieldType(Scalar): class GenericStreamFieldType(Scalar):
@staticmethod @staticmethod
@ -29,13 +32,17 @@ class GenericStreamFieldType(Scalar):
d['value'] = value d['value'] = value
if _type == 'assignment': if _type == 'assignment':
_value = d['value'] _value = d['value']
assignment = Assignment.objects.get(pk=_value['assignment_id']) assignment_id = _value['assignment_id']
value = { try:
'title': assignment.title, assignment = Assignment.objects.get(pk=assignment_id)
'assignment': assignment.assignment, value = {
'id': to_global_id('AssignmentNode', assignment.pk) 'title': assignment.title,
} 'assignment': assignment.assignment,
d['value'] = value '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': if _type == 'basic_knowledge':
_value = d['value'] _value = d['value']
basic_knowledge = BasicKnowledge.objects.get(pk=_value['basic_knowledge']) basic_knowledge = BasicKnowledge.objects.get(pk=_value['basic_knowledge'])