VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

better separation pep0484 code and py__annotation__() function

Browse Source
This commit is contained in:
Claude
2015-12-14 12:10:00 +01:00
parent be399c81c3
commit 576fdf8106
2 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jedi/evaluate/pep0484.py
+14 -4
View File
@@ -21,12 +21,22 @@ 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 = param.annotation()
if annotation:
def _evaluate_for_annotation(evaluator, annotation):
if annotation is not None:
definitions = evaluator.eval_element(annotation)
return list(chain.from_iterable(
evaluator.execute(d) for d in definitions))
else:
return []
@memoize_default(None, evaluator_is_first_arg=True)
def follow_param(evaluator, param):
annotation = param.annotation()
return _evaluate_for_annotation(evaluator, annotation)
@memoize_default(None, evaluator_is_first_arg=True)
def find_return_types(evaluator, func):
annotation = func.py__annotations__().get("return", None)
return _evaluate_for_annotation(evaluator, annotation)