forked from VimPlug/jedi
Basic django model fields are infered as builtin types.
This commit is contained in:
@@ -787,6 +787,69 @@ _implemented = {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def new_django_dict_filter(cls):
|
||||||
|
filter_ = ParserTreeFilter(parent_context=cls.as_context())
|
||||||
|
field, = filter_.values()
|
||||||
|
field_tree_instance, = field.infer()
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name in ('CharField', 'TextField', 'EmailField'):
|
||||||
|
builtin_str, = cls.inference_state.builtins_module.py__getattribute__('str')
|
||||||
|
return [DictFilter({
|
||||||
|
field.string_name: DjangoModelField(builtin_str, field).name
|
||||||
|
})]
|
||||||
|
|
||||||
|
integer_field_classes = ('IntegerField', 'BigIntegerField', 'PositiveIntegerField', 'SmallIntegerField')
|
||||||
|
if field_tree_instance.name.string_name in integer_field_classes:
|
||||||
|
builtin_str, = cls.inference_state.builtins_module.py__getattribute__('int')
|
||||||
|
return [DictFilter({
|
||||||
|
field.string_name: DjangoModelField(builtin_str, field).name
|
||||||
|
})]
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'FloatField':
|
||||||
|
builtin_str, = cls.inference_state.builtins_module.py__getattribute__('float')
|
||||||
|
return [DictFilter({
|
||||||
|
field.string_name: DjangoModelField(builtin_str, field).name
|
||||||
|
})]
|
||||||
|
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'BinaryField':
|
||||||
|
builtin_str, = cls.inference_state.builtins_module.py__getattribute__('bytes')
|
||||||
|
return [DictFilter({
|
||||||
|
field.string_name: DjangoModelField(builtin_str, field).name
|
||||||
|
})]
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'BooleanField':
|
||||||
|
builtin_str, = cls.inference_state.builtins_module.py__getattribute__('bool')
|
||||||
|
return [DictFilter({
|
||||||
|
field.string_name: DjangoModelField(builtin_str, field).name
|
||||||
|
})]
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'DecimalField':
|
||||||
|
# TODO: make decimal.Decimal filter
|
||||||
|
return
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'ForeignKey':
|
||||||
|
# TODO: infer related object class and make a filter for that class
|
||||||
|
return
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'TimeField':
|
||||||
|
# TODO: make time.time filter
|
||||||
|
return
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'DurationField':
|
||||||
|
# TODO: make datetime.timedelta filter
|
||||||
|
return
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'DateField':
|
||||||
|
# TODO: make datetime.date filter
|
||||||
|
return
|
||||||
|
|
||||||
|
if field_tree_instance.name.string_name == 'DatetimeField':
|
||||||
|
# TODO: make datetime.datetime filter
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def get_metaclass_filters(func):
|
def get_metaclass_filters(func):
|
||||||
def wrapper(cls, metaclasses):
|
def wrapper(cls, metaclasses):
|
||||||
for metaclass in metaclasses:
|
for metaclass in metaclasses:
|
||||||
@@ -796,10 +859,37 @@ def get_metaclass_filters(func):
|
|||||||
return [DictFilter({
|
return [DictFilter({
|
||||||
name.string_name: EnumInstance(cls, name).name for name in filter_.values()
|
name.string_name: EnumInstance(cls, name).name for name in filter_.values()
|
||||||
})]
|
})]
|
||||||
|
|
||||||
|
if metaclass.py__name__() == 'ModelBase' \
|
||||||
|
and metaclass.get_root_context().py__name__() == 'django.db.models.base':
|
||||||
|
django_dict_filter = new_django_dict_filter(cls)
|
||||||
|
if django_dict_filter is not None:
|
||||||
|
return django_dict_filter
|
||||||
|
|
||||||
return func(cls, metaclasses)
|
return func(cls, metaclasses)
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
|
class DjangoModelField(LazyValueWrapper):
|
||||||
|
def __init__(self, cls, name):
|
||||||
|
self.inference_state = cls.inference_state
|
||||||
|
self._cls = cls # Corresponds to super().__self__
|
||||||
|
self._name = name
|
||||||
|
self.tree_node = self._name.tree_name
|
||||||
|
|
||||||
|
@safe_property
|
||||||
|
def name(self):
|
||||||
|
return ValueName(self, self._name.tree_name)
|
||||||
|
|
||||||
|
def _get_wrapped_value(self):
|
||||||
|
obj, = self._cls.execute_with_values()
|
||||||
|
return obj
|
||||||
|
|
||||||
|
def get_filters(self, origin_scope=None):
|
||||||
|
for f in self._get_wrapped_value().get_filters():
|
||||||
|
yield f
|
||||||
|
|
||||||
|
|
||||||
class EnumInstance(LazyValueWrapper):
|
class EnumInstance(LazyValueWrapper):
|
||||||
def __init__(self, cls, name):
|
def __init__(self, cls, name):
|
||||||
self.inference_state = cls.inference_state
|
self.inference_state = cls.inference_state
|
||||||
|
|||||||
Reference in New Issue
Block a user