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

Fix ForeignKey issues with invalid values

Browse Source
This commit is contained in:
Dave Halter
2020-04-22 00:27:06 +02:00
parent 1a89fafce4
commit ba4e3393d3
2 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
jedi/plugins/django.py
View File

@@ -48,10 +48,11 @@ mapping = {
def _infer_scalar_field(cls, field, field_tree_instance):
if field_tree_instance.name.string_name not in mapping:
try:
module_name, attribute_name = mapping[field_tree_instance.name.string_name]
except KeyError:
return None
module_name, attribute_name = mapping[field_tree_instance.name.string_name]
if module_name is None:
module = cls.inference_state.builtins_module
else:
@@ -69,6 +70,7 @@ def _infer_field(cls, field):
if field_tree_instance.name.string_name == 'ForeignKey':
if isinstance(field_tree_instance, TreeInstance):
# TODO private access..
argument_iterator = field_tree_instance._arguments.unpack()
key, lazy_values = next(argument_iterator, (None, None))
if key is None and lazy_values is not None:
@@ -77,7 +79,7 @@ def _infer_field(cls, field):
foreign_key_class_name = value.get_safe_value()
for v in cls.parent_context.py__getattribute__(foreign_key_class_name):
return DjangoModelField(v, field)
else:
elif value.is_class():
return DjangoModelField(value, field)
debug.dbg('django plugin: fail to infer `%s` from class `%s`',