1
0
forked from VimPlug/jedi

Fix another issue with foreign keys

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

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)