This commit is contained in:
Mardanov Timur Rustemovich 2018-11-20 04:23:54 +03:00
parent e801fcaa3d
commit 8df0a44d65
2 changed files with 2 additions and 13 deletions

View File

@ -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)

View File

@ -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