Skip to content

Commit fadf366

Browse files
committed
Refactored django function to utils
1 parent bd0ec6d commit fadf366

File tree

2 files changed

+9
-8
lines changed

2 files changed

+9
-8
lines changed

graphene/contrib/django/types.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,14 @@
11
import six
2-
from django.db import models
32

43
from graphene.core.types import ObjectTypeMeta, BaseObjectType
54
from graphene.contrib.django.options import DjangoOptions
65
from graphene.contrib.django.converter import convert_django_field
6+
from graphene.contrib.django.utils import get_reverse_fields
77

88
from graphene.relay.types import BaseNode
99
from graphene.relay.fields import GlobalIDField
1010

1111

12-
def get_reverse_fields(model):
13-
for name, attr in model.__dict__.items():
14-
related = getattr(attr, 'related', None)
15-
if isinstance(related, models.ManyToOneRel):
16-
yield related
17-
18-
1912
class DjangoObjectTypeMeta(ObjectTypeMeta):
2013
options_cls = DjangoOptions
2114

graphene/contrib/django/utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from django.db import models
12
from django.db.models.query import QuerySet
23
from django.db.models.manager import Manager
34

@@ -15,6 +16,13 @@ def get_type_for_model(schema, model):
1516
return _type
1617

1718

19+
def get_reverse_fields(model):
20+
for name, attr in model.__dict__.items():
21+
related = getattr(attr, 'related', None)
22+
if isinstance(related, models.ManyToOneRel):
23+
yield related
24+
25+
1826
def lazy_map(value, func):
1927
if isinstance(value, Manager):
2028
value = value.get_queryset()

0 commit comments

Comments
 (0)