From d4f0424ddc1415dee7ded4dc9f7af044a6ae71ab Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Mon, 8 Jun 2020 00:58:38 +0200 Subject: [PATCH] Move py__getitem__ from Class to ClassMixin --- jedi/inference/value/klass.py | 32 ++++++++++++++++---------------- 1 file changed, 16 insertions(+), 16 deletions(-) diff --git a/jedi/inference/value/klass.py b/jedi/inference/value/klass.py index 9c2102ff..1a8cac2f 100644 --- a/jedi/inference/value/klass.py +++ b/jedi/inference/value/klass.py @@ -271,6 +271,22 @@ class ClassMixin(object): return True return False + def py__getitem__(self, index_value_set, contextualized_node): + from jedi.inference.gradual.base import GenericClass + if not index_value_set: + debug.warning('Class indexes inferred to nothing. Returning class instead') + return ValueSet([self]) + return ValueSet( + GenericClass( + self, + LazyGenericManager( + context_of_index=contextualized_node.context, + index_value=index_value, + ) + ) + for index_value in index_value_set + ) + def with_generics(self, generics_tuple): from jedi.inference.gradual.base import GenericClass return GenericClass( @@ -345,22 +361,6 @@ class ClassValue(use_metaclass(CachedMetaClass, ClassMixin, FunctionAndClassBase self.inference_state.builtins_module.py__getattribute__('object') )] - def py__getitem__(self, index_value_set, contextualized_node): - from jedi.inference.gradual.base import GenericClass - if not index_value_set: - debug.warning('Class indexes inferred to nothing. Returning class instead') - return ValueSet([self]) - return ValueSet( - GenericClass( - self, - LazyGenericManager( - context_of_index=contextualized_node.context, - index_value=index_value, - ) - ) - for index_value in index_value_set - ) - @plugin_manager.decorate() def get_metaclass_filters(self, metaclass, is_instance): debug.warning('Unprocessed metaclass %s', metaclass)