forked from VimPlug/jedi
92 lines
2.7 KiB
Python
92 lines
2.7 KiB
Python
"""
|
|
This module is about generics, like the `int` in `List[int]`. It's not about
|
|
the Generic class.
|
|
"""
|
|
|
|
from jedi.cache import memoize_method
|
|
from jedi.inference.utils import to_tuple
|
|
from jedi.inference.base_value import ValueSet
|
|
from jedi.inference.value.iterable import SequenceLiteralValue
|
|
from jedi.inference.helpers import is_string
|
|
|
|
|
|
def iter_over_arguments(maybe_tuple_value, defining_context):
|
|
def iterate():
|
|
if isinstance(maybe_tuple_value, SequenceLiteralValue):
|
|
for lazy_value in maybe_tuple_value.py__iter__(contextualized_node=None):
|
|
yield lazy_value.infer()
|
|
else:
|
|
yield ValueSet([maybe_tuple_value])
|
|
|
|
for value_set in iterate():
|
|
yield ValueSet(_resolve_forward_references(defining_context, value_set))
|
|
|
|
|
|
def _resolve_forward_references(context, value_set):
|
|
for value in value_set:
|
|
if is_string(value):
|
|
from jedi.inference.gradual.annotation import _get_forward_reference_node
|
|
node = _get_forward_reference_node(context, value.get_safe_value())
|
|
if node is not None:
|
|
for c in context.infer_node(node):
|
|
yield c
|
|
else:
|
|
yield value
|
|
|
|
|
|
class LazyGenericManager(object):
|
|
def __init__(self, context_of_index, index_value):
|
|
self._context_of_index = context_of_index
|
|
self._index_value = index_value
|
|
|
|
@memoize_method
|
|
def __getitem__(self, index):
|
|
return self._tuple()[index]()
|
|
|
|
def __len__(self):
|
|
return len(self._tuple())
|
|
|
|
@memoize_method
|
|
@to_tuple
|
|
def _tuple(self):
|
|
def lambda_scoping_in_for_loop_sucks(lazy_value):
|
|
return lambda: ValueSet(_resolve_forward_references(
|
|
self._context_of_index,
|
|
lazy_value.infer()
|
|
))
|
|
|
|
if isinstance(self._index_value, SequenceLiteralValue):
|
|
for lazy_value in self._index_value.py__iter__(contextualized_node=None):
|
|
yield lambda_scoping_in_for_loop_sucks(lazy_value)
|
|
else:
|
|
yield lambda: ValueSet(_resolve_forward_references(
|
|
self._context_of_index,
|
|
ValueSet([self._index_value])
|
|
))
|
|
|
|
@to_tuple
|
|
def to_tuple(self):
|
|
for callable_ in self._tuple():
|
|
yield callable_()
|
|
|
|
#def __iter__(self):
|
|
# return iter(self._iterate())
|
|
|
|
|
|
class TupleGenericManager(object):
|
|
def __init__(self, tup):
|
|
self._tuple = tup
|
|
|
|
def __getitem__(self, index):
|
|
return self._tuple[index]
|
|
|
|
def __len__(self):
|
|
return len(self._tuple)
|
|
|
|
#def __iter__(self):
|
|
# for value_set in self._tuple:
|
|
# yield lambda: value_set
|
|
|
|
def to_tuple(self):
|
|
return self._tuple
|