VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-06 14:04:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Some more code to a function

Browse Source
This commit is contained in:
Dave Halter
2018-03-14 21:49:17 +01:00
parent ce0aa224f1
commit 053618edd0
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
jedi/evaluate/pep0484.py
View File

@@ -43,12 +43,17 @@ def _evaluate_for_annotation(context, annotation, index=None):
and we're interested in that index
"""
context_set = context.eval_node(_fix_forward_reference(context, annotation))
if index is not None:
context_set = context_set.filter(
lambda context: context.array_type == u'tuple'
and len(list(context.py__iter__())) >= index
).py__getitem__(index)
return context_set.execute_evaluated()
return _check_index(context_set, index).execute_evaluated()
def _check_index(context_set, index):
if index is None:
return context_set
return context_set.filter(
lambda context: context.array_type == u'tuple'
and len(list(context.py__iter__())) >= index
).py__getitem__(index)
def _evaluate_annotation_string(context, string, index=None):
@@ -57,12 +62,7 @@ def _evaluate_annotation_string(context, string, index=None):
return NO_CONTEXTS
context_set = context.eval_node(node)
if index is not None:
context_set = context_set.filter(
lambda context: context.array_type == u'tuple'
and len(list(context.py__iter__())) >= index
).py__getitem__(index)
return context_set.execute_evaluated()
return _check_index(context_set, index).execute_evaluated()
def _fix_forward_reference(context, node):