Refactored django function to utils

This commit is contained in:
Syrus Akbary 2015-10-18 19:43:58 -07:00
parent bd0ec6dc14
commit fadf3664c0
2 changed files with 9 additions and 8 deletions

View File

@ -1,21 +1,14 @@
import six
from django.db import models
from graphene.core.types import ObjectTypeMeta, BaseObjectType
from graphene.contrib.django.options import DjangoOptions
from graphene.contrib.django.converter import convert_django_field
from graphene.contrib.django.utils import get_reverse_fields
from graphene.relay.types import BaseNode
from graphene.relay.fields import GlobalIDField
def get_reverse_fields(model):
for name, attr in model.__dict__.items():
related = getattr(attr, 'related', None)
if isinstance(related, models.ManyToOneRel):
yield related
class DjangoObjectTypeMeta(ObjectTypeMeta):
options_cls = DjangoOptions

View File

@ -1,3 +1,4 @@
from django.db import models
from django.db.models.query import QuerySet
from django.db.models.manager import Manager
@ -15,6 +16,13 @@ def get_type_for_model(schema, model):
return _type
def get_reverse_fields(model):
for name, attr in model.__dict__.items():
related = getattr(attr, 'related', None)
if isinstance(related, models.ManyToOneRel):
yield related
def lazy_map(value, func):
if isinstance(value, Manager):
value = value.get_queryset()