VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-19 12:01:12 +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

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