From 8df0a44d653c936c9e121029137624f7dd2f66fb Mon Sep 17 00:00:00 2001 From: Mardanov Timur Rustemovich Date: Tue, 20 Nov 2018 04:23:54 +0300 Subject: [PATCH] fix --- graphene_django/converter.py | 4 ++-- graphene_django/registry.py | 11 ----------- 2 files changed, 2 insertions(+), 13 deletions(-) diff --git a/graphene_django/converter.py b/graphene_django/converter.py index 5cb1b15..ec1e1e5 100644 --- a/graphene_django/converter.py +++ b/graphene_django/converter.py @@ -166,7 +166,7 @@ def convert_date_to_string(field, registry=None): @convert_django_field.register(RelationshipDefinition) def convert_onetoone_field_to_djangomodel(field, registry=None): - model = registry.get_model_by_string(field._raw_class) + model = field._raw_class def dynamic_type(): _type = registry.get_type_for_model(model) @@ -186,7 +186,7 @@ def convert_onetoone_field_to_djangomodel(field, registry=None): @convert_django_field.register(models.ManyToManyRel) @convert_django_field.register(models.ManyToOneRel) def convert_field_to_list_or_connection(field, registry=None): - model = registry.get_model_by_string(field._raw_class) + model = field.related_model def dynamic_type(): _type = registry.get_type_for_model(model) diff --git a/graphene_django/registry.py b/graphene_django/registry.py index 2c0702c..50a8ae5 100644 --- a/graphene_django/registry.py +++ b/graphene_django/registry.py @@ -19,19 +19,8 @@ class Registry(object): self._registry[cls._meta.model] = cls def get_type_for_model(self, model): - print(model) - if isinstance(model, str): - model = self.get_model_by_string(model) return self._registry.get(model) - def get_model_by_string(self, model): - print(self.__dict__) - print(self._registry.__dict__) - print(self._field_registry) - print(self._registry.keys()) - return list(filter(lambda x: model == x.__name__, self._registry.keys()))[0] - - def register_converted_field(self, field, converted): self._field_registry[field] = converted