Fix id lookup to handle cases where the primary key of the Django model is not called ID

This commit is contained in:
Morgante Pell 2016-10-13 02:52:58 -04:00
parent 31a49841ca
commit 900354837e
No known key found for this signature in database
GPG Key ID: A38BA77A4985F471

View File

@ -98,6 +98,9 @@ class DjangoObjectTypeMeta(ObjectTypeMeta):
class DjangoObjectType(six.with_metaclass(DjangoObjectTypeMeta, ObjectType)):
def resolve_id(self, args, context, info):
return self.pk
@classmethod
def is_type_of(cls, root, context, info):
if isinstance(root, cls):
@ -112,6 +115,6 @@ class DjangoObjectType(six.with_metaclass(DjangoObjectTypeMeta, ObjectType)):
@classmethod
def get_node(cls, id, context, info):
try:
return cls._meta.model.objects.get(id=id)
return cls._meta.model.objects.get(pk=id)
except cls._meta.model.DoesNotExist:
return None