forked from VimPlug/jedi
35 lines
1.4 KiB
Python
35 lines
1.4 KiB
Python
"""
|
|
PEP 0484 ( https://www.python.org/dev/peps/pep-0484/ ) describes type hints
|
|
through function annotations. There is a strong suggestion in this document
|
|
that only the type of type hinting defined in PEP0484 should be allowed
|
|
as annotations in future python versions.
|
|
|
|
The (initial / probably incomplete) implementation todo list for pep-0484:
|
|
v Function parameter annotations with builtin/custom type classes
|
|
x Function returntype annotations with builtin/custom type classes
|
|
x Function parameter annotations with strings (forward reference)
|
|
x Function return type annotations with strings (forward reference)
|
|
x Local variable type hints
|
|
x Assigned types: `Url = str\ndef get(url:Url) -> str:`
|
|
x Type hints in `with` statements
|
|
x Stub files support
|
|
"""
|
|
|
|
from itertools import chain
|
|
|
|
from jedi.evaluate.cache import memoize_default
|
|
|
|
|
|
@memoize_default(None, evaluator_is_first_arg=True)
|
|
def follow_param(evaluator, param):
|
|
# annotation is in param.children[0] if present
|
|
# either this firstchild is a Name (if no annotation is present) or a Node
|
|
if hasattr(param.children[0], "children"):
|
|
assert len(param.children[0].children) == 3 and \
|
|
param.children[0].children[1] == ":"
|
|
definitions = evaluator.eval_element(param.children[0].children[2])
|
|
return list(chain.from_iterable(
|
|
evaluator.execute(d) for d in definitions))
|
|
else:
|
|
return []
|