VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

Fix another issue with foreign keys

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

5
jedi/plugins/django.py
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)

3
test/completion/django.py
View File

@@ -16,6 +16,7 @@ class BusinessModel(models.Model):
category_fk = models.ForeignKey(Category)
category_fk2 = models.ForeignKey('Category')
category_fk3 = models.ForeignKey(1)
category_fk4 = models.ForeignKey('models')
integer_field = models.IntegerField()
big_integer_field = models.BigIntegerField()
positive_integer_field = models.PositiveIntegerField()
@@ -76,6 +77,8 @@ model_instance.category_fk2
model_instance.category_fk2.category_name
#? models.ForeignKey()
model_instance.category_fk3
#? models.ForeignKey()
model_instance.category_fk4
#? models.ManyToManyField()
model_instance.tags_m2m