VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-18 03:25:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Functions infers type of Django model field is refactored.

Browse Source
This commit is contained in:
ANtlord
2020-01-21 21:12:38 +02:00
parent c61ca0d27b
commit 7287d67e7a
1 changed files with 28 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
jedi/plugins/django.py
View File

@@ -30,34 +30,38 @@ class DjangoModelField(LazyValueWrapper):
obj, = self._cls.execute_with_values() obj, = self._cls.execute_with_values()
return obj return obj
mapping = {
'IntegerField': (None, 'int'),
'BigIntegerField': (None, 'int'),
'PositiveIntegerField': (None, 'int'),
'SmallIntegerField': (None, 'int'),
'CharField': (None, 'str'),
'TextField': (None, 'str'),
'EmailField': (None, 'str'),
'FloatField': (None, 'float'),
'BinaryField': (None, 'bytes'),
'BooleanField': (None, 'bool'),
'DecimalField': ('decimal', 'Decimal'),
'TimeField': ('datetime', 'time'),
'DurationField': ('datetime', 'timedelta'),
'DateField': ('datetime', 'date'),
'DateTimeField': ('datetime', 'datetime'),
}
def _infer_field(cls, field): def _infer_field(cls, field):
field_tree_instance, = field.infer() field_tree_instance, = field.infer()
if field_tree_instance.name.string_name in ('CharField', 'TextField', 'EmailField'): try:
model_instance_field_type, = cls.inference_state.builtins_module.py__getattribute__('str') module_name, attribute_name = mapping[field_tree_instance.name.string_name]
return DjangoModelField(model_instance_field_type, field).name except KeyError:
pass
integer_field_classes = ('IntegerField', 'BigIntegerField', 'PositiveIntegerField', 'SmallIntegerField') else:
if field_tree_instance.name.string_name in integer_field_classes: if module_name is None:
model_instance_field_type, = cls.inference_state.builtins_module.py__getattribute__('int') module = cls.inference_state.builtins_module
return DjangoModelField(model_instance_field_type, field).name else:
module = cls.inference_state.import_module((module_name,))
if field_tree_instance.name.string_name == 'FloatField': attribute, = module.py__getattribute__(attribute_name)
model_instance_field_type, = cls.inference_state.builtins_module.py__getattribute__('float') return DjangoModelField(attribute, field).name
return DjangoModelField(model_instance_field_type, field).name
if field_tree_instance.name.string_name == 'BinaryField':
model_instance_field_type, = cls.inference_state.builtins_module.py__getattribute__('bytes')
return DjangoModelField(model_instance_field_type, field).name
if field_tree_instance.name.string_name == 'BooleanField':
model_instance_field_type, = cls.inference_state.builtins_module.py__getattribute__('bool')
return DjangoModelField(model_instance_field_type, field).name
if field_tree_instance.name.string_name == 'DecimalField':
model_instance_field_type, = cls.inference_state.import_module(('decimal',)).py__getattribute__('Decimal')
return DjangoModelField(model_instance_field_type, field).name
if field_tree_instance.name.string_name == 'ForeignKey': if field_tree_instance.name.string_name == 'ForeignKey':
if isinstance(field_tree_instance, TreeInstance): if isinstance(field_tree_instance, TreeInstance):
@@ -76,20 +80,3 @@ def _infer_field(cls, field):
return DjangoModelField(value, field).name return DjangoModelField(value, field).name
raise Exception('Should be handled') raise Exception('Should be handled')
if field_tree_instance.name.string_name == 'TimeField':
model_instance_field_type, = cls.inference_state.import_module(('datetime',)).py__getattribute__('time')
return DjangoModelField(model_instance_field_type, field).name
if field_tree_instance.name.string_name == 'DurationField':
model_instance_field_type, = cls.inference_state.import_module(('datetime',)).py__getattribute__('timedelta')
return DjangoModelField(model_instance_field_type, field).name
if field_tree_instance.name.string_name == 'DateField':
model_instance_field_type, = cls.inference_state.import_module(('datetime',)).py__getattribute__('date')
return DjangoModelField(model_instance_field_type, field).name
if field_tree_instance.name.string_name == 'DateTimeField':
model_instance_field_type, = cls.inference_state.import_module(('datetime',)).py__getattribute__('datetime')
return DjangoModelField(model_instance_field_type, field).name