djavu-asus/elpy/rpc-venv/lib/python3.11/site-packages/jedi/inference/lazy_value.py

   1 from jedi.inference.base_value import ValueSet, NO_VALUES
   2 from jedi.common import monkeypatch
   3
   4
   5 class AbstractLazyValue:
   6     def __init__(self, data, min=1, max=1):
   7         self.data = data
   8         self.min = min
   9         self.max = max
  10
  11     def __repr__(self):
  12         return '<%s: %s>' % (self.__class__.__name__, self.data)
  13
  14     def infer(self):
  15         raise NotImplementedError
  16
  17
  18 class LazyKnownValue(AbstractLazyValue):
  19     """data is a Value."""
  20     def infer(self):
  21         return ValueSet([self.data])
  22
  23
  24 class LazyKnownValues(AbstractLazyValue):
  25     """data is a ValueSet."""
  26     def infer(self):
  27         return self.data
  28
  29
  30 class LazyUnknownValue(AbstractLazyValue):
  31     def __init__(self, min=1, max=1):
  32         super().__init__(None, min, max)
  33
  34     def infer(self):
  35         return NO_VALUES
  36
  37
  38 class LazyTreeValue(AbstractLazyValue):
  39     def __init__(self, context, node, min=1, max=1):
  40         super().__init__(node, min, max)
  41         self.context = context
  42         # We need to save the predefined names. It's an unfortunate side effect
  43         # that needs to be tracked otherwise results will be wrong.
  44         self._predefined_names = dict(context.predefined_names)
  45
  46     def infer(self):
  47         with monkeypatch(self.context, 'predefined_names', self._predefined_names):
  48             return self.context.infer_node(self.data)
  49
  50
  51 def get_merged_lazy_value(lazy_values):
  52     if len(lazy_values) > 1:
  53         return MergedLazyValues(lazy_values)
  54     else:
  55         return lazy_values[0]
  56
  57
  58 class MergedLazyValues(AbstractLazyValue):
  59     """data is a list of lazy values."""
  60     def infer(self):
  61         return ValueSet.from_sets(l.infer() for l in self.data)