Fix another issue with foreign keys

This commit is contained in:
Dave Halter committed 2020-04-22 00:33:51 +02:00
1 parent ba4e3393d3
commit 7756792bba
2 files changed
+6 -2

No files matched your search

+3 -2
View File
@@ -77,8 +77,9 @@ def _infer_field(cls, field):
for value in lazy_values.infer():
if value.name.string_name == 'str':
foreign_key_class_name = value.get_safe_value()
for v in cls.parent_context.py__getattribute__(foreign_key_class_name):
return DjangoModelField(v, field)
for v in cls.get_root_context().py__getattribute__(foreign_key_class_name):
if v.is_class():
return DjangoModelField(v, field)
elif value.is_class():
return DjangoModelField(value, field)