VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2026-05-18 06:19:39 +08:00
Code Issues Actions Packages Projects Releases Wiki 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
jedi/plugins/django.py
+3 -2
View File
@@ -77,8 +77,9 @@ def _infer_field(cls, field):
for value in lazy_values.infer(): for value in lazy_values.infer():
if value.name.string_name == 'str': if value.name.string_name == 'str':
foreign_key_class_name = value.get_safe_value() foreign_key_class_name = value.get_safe_value()
for v in cls.parent_context.py__getattribute__(foreign_key_class_name): for v in cls.get_root_context().py__getattribute__(foreign_key_class_name):
return DjangoModelField(v, field) if v.is_class():
return DjangoModelField(v, field)
elif value.is_class(): elif value.is_class():
return DjangoModelField(value, field) return DjangoModelField(value, field)
test/completion/django.py
+3
View File
@@ -16,6 +16,7 @@ class BusinessModel(models.Model):
category_fk = models.ForeignKey(Category) category_fk = models.ForeignKey(Category)
category_fk2 = models.ForeignKey('Category') category_fk2 = models.ForeignKey('Category')
category_fk3 = models.ForeignKey(1) category_fk3 = models.ForeignKey(1)
category_fk4 = models.ForeignKey('models')
integer_field = models.IntegerField() integer_field = models.IntegerField()
big_integer_field = models.BigIntegerField() big_integer_field = models.BigIntegerField()
positive_integer_field = models.PositiveIntegerField() positive_integer_field = models.PositiveIntegerField()
@@ -76,6 +77,8 @@ model_instance.category_fk2
model_instance.category_fk2.category_name model_instance.category_fk2.category_name
#? models.ForeignKey() #? models.ForeignKey()
model_instance.category_fk3 model_instance.category_fk3
#? models.ForeignKey()
model_instance.category_fk4
#? models.ManyToManyField() #? models.ManyToManyField()
model_instance.tags_m2m model_instance.tags_m2m