graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2024-11-26 19:43:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed many to many relations in Django model

Browse Source
This commit is contained in:
Syrus Akbary 2016-01-11 16:50:05 +01:00
parent 84dbc58115
commit f68763b822
4 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
graphene/contrib/django/converter.py
View File

@ -58,6 +58,7 @@ def convert_field_to_float(field):
@convert_django_field.register(models.ManyToManyField) @convert_django_field.register(models.ManyToManyField)
@convert_django_field.register(models.ManyToOneRel) @convert_django_field.register(models.ManyToOneRel)
@convert_django_field.register(models.ManyToManyRel)
def convert_field_to_list_or_connection(field): def convert_field_to_list_or_connection(field):
from .fields import DjangoModelField, ConnectionOrListField from .fields import DjangoModelField, ConnectionOrListField
model_field = DjangoModelField(get_related_model(field)) model_field = DjangoModelField(get_related_model(field))

5
graphene/contrib/django/tests/models.py
View File

@ -7,6 +7,11 @@ class Pet(models.Model):
name = models.CharField(max_length=30) name = models.CharField(max_length=30)
class Film(models.Model):
reporters = models.ManyToManyField('Reporter',
related_name='films')
class Reporter(models.Model): class Reporter(models.Model):
first_name = models.CharField(max_length=30) first_name = models.CharField(max_length=30)
last_name = models.CharField(max_length=30) last_name = models.CharField(max_length=30)

2
graphene/contrib/django/tests/test_schema.py
View File

@ -29,7 +29,7 @@ def test_should_map_fields_correctly():
model = Reporter model = Reporter
assert_equal_lists( assert_equal_lists(
ReporterType2._meta.fields_map.keys(), ReporterType2._meta.fields_map.keys(),
['articles', 'first_name', 'last_name', 'email', 'pets', 'id'] ['articles', 'first_name', 'last_name', 'email', 'pets', 'id', 'films']
) )

2
graphene/contrib/django/utils.py
View File

@ -35,6 +35,8 @@ def get_reverse_fields(model):
yield new_related yield new_related
elif isinstance(related, models.ManyToOneRel): elif isinstance(related, models.ManyToOneRel):
yield related yield related
elif isinstance(related, models.ManyToManyRel) and not related.symmetrical:
yield related
class WrappedQueryset(LazyList): class WrappedQueryset(LazyList):