VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2026-05-17 05:49:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

cleaned up evaluate.param

Browse Source
This commit is contained in:
Dave Halter
2014-01-07 14:44:39 +01:00
parent 1881a24e73
commit 8a9453872f
1 changed files with 56 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jedi/evaluate/param.py
+56 -66
View File
@@ -53,9 +53,8 @@ def get_params(evaluator, func, var_args):
param_dict = {} param_dict = {}
for param in func.params: for param in func.params:
param_dict[str(param.get_name())] = param param_dict[str(param.get_name())] = param
# There may be calls, which don't fit all the params, this just ignores # There may be calls, which don't fit all the params, this just ignores it.
# it. var_arg_iterator = common.PushBackIterator(_var_args_iterator(evaluator, var_args))
var_arg_iterator = _get_var_args_iterator(evaluator, var_args)
non_matching_keys = [] non_matching_keys = []
keys_used = set() keys_used = set()
@@ -63,9 +62,8 @@ def get_params(evaluator, func, var_args):
for param in func.params[start_offset:]: for param in func.params[start_offset:]:
# The value and key can both be null. There, the defaults apply. # The value and key can both be null. There, the defaults apply.
# args / kwargs will just be empty arrays / dicts, respectively. # args / kwargs will just be empty arrays / dicts, respectively.
# Wrong value count is just ignored. If you try to test cases that # Wrong value count is just ignored. If you try to test cases that are
# are not allowed in Python, Jedi will maybe not show any # not allowed in Python, Jedi will maybe not show any completions.
# completions.
key, value = next(var_arg_iterator, (None, None)) key, value = next(var_arg_iterator, (None, None))
while key: while key:
keys_only = True keys_only = True
@@ -75,8 +73,7 @@ def get_params(evaluator, func, var_args):
non_matching_keys.append((key, value)) non_matching_keys.append((key, value))
else: else:
keys_used.add(str(key)) keys_used.add(str(key))
result.append(gen_param_name_copy(key_param, result.append(gen_param_name_copy(key_param, values=[value]))
values=[value]))
key, value = next(var_arg_iterator, (None, None)) key, value = next(var_arg_iterator, (None, None))
expression_list = param.expression_list() expression_list = param.expression_list()
@@ -111,17 +108,17 @@ def get_params(evaluator, func, var_args):
result.append(param.get_name()) result.append(param.get_name())
param.is_generated = True param.is_generated = True
else: else:
# If there is no assignment detail, that means there is # If there is no assignment detail, that means there is no
# no assignment, just the result. Therefore nothing has # assignment, just the result. Therefore nothing has to be
# to be returned. # returned.
values = [] values = []
# Just ignore all the params that are without a key, after one # Just ignore all the params that are without a key, after one keyword
# keyword argument was set. # argument was set.
if not ignore_creation and (not keys_only or expression_list[0] == '**'): if not ignore_creation and (not keys_only or expression_list[0] == '**'):
keys_used.add(str(key)) keys_used.add(str(key))
result.append(gen_param_name_copy(param, keys=keys, result.append(gen_param_name_copy(param, keys=keys, values=values,
values=values, array_type=array_type)) array_type=array_type))
if keys_only: if keys_only:
# sometimes param arguments are not completely written (which would # sometimes param arguments are not completely written (which would
@@ -131,61 +128,55 @@ def get_params(evaluator, func, var_args):
return result return result
def _get_var_args_iterator(evaluator, var_args): def _var_args_iterator(evaluator, var_args):
""" """
Yields a key/value pair, the key is None, if its not a named arg. Yields a key/value pair, the key is None, if its not a named arg.
""" """
def iterate(): # `var_args` is typically an Array, and not a list.
# `var_args` is typically an Array, and not a list. for stmt in var_args:
for stmt in var_args: if not isinstance(stmt, pr.Statement):
if not isinstance(stmt, pr.Statement): if stmt is None:
if stmt is None: yield None, None
yield None, None
continue
old = stmt
# generate a statement if it's not already one.
module = builtin.Builtin.scope
stmt = pr.Statement(module, [], (0, 0), None)
stmt._expression_list = [old]
# *args
expression_list = stmt.expression_list()
if not len(expression_list):
continue continue
if expression_list[0] == '*': old = stmt
arrays = evaluator.eval_expression_list(expression_list[1:]) # generate a statement if it's not already one.
# *args must be some sort of an array, otherwise -> ignore module = builtin.Builtin.scope
stmt = pr.Statement(module, [], (0, 0), None)
stmt._expression_list = [old]
for array in arrays: # *args
if isinstance(array, iterable.Array): expression_list = stmt.expression_list()
for field_stmt in array: # yield from plz! if not len(expression_list):
yield None, field_stmt continue
elif isinstance(array, iterable.Generator): if expression_list[0] == '*':
for field_stmt in array.iter_content(): # *args must be some sort of an array, otherwise -> ignore
yield None, _FakeStatement(field_stmt) for array in evaluator.eval_expression_list(expression_list[1:]):
# **kwargs if isinstance(array, iterable.Array):
elif expression_list[0] == '**': for field_stmt in array: # yield from plz!
arrays = evaluator.eval_expression_list(expression_list[1:]) yield None, field_stmt
for array in arrays: elif isinstance(array, iterable.Generator):
if isinstance(array, iterable.Array): for field_stmt in array.iter_content():
for key_stmt, value_stmt in array.items(): yield None, _FakeStatement(field_stmt)
# first index, is the key if syntactically correct # **kwargs
call = key_stmt.expression_list()[0] elif expression_list[0] == '**':
if isinstance(call, pr.Name): for array in evaluator.eval_expression_list(expression_list[1:]):
yield call, value_stmt if isinstance(array, iterable.Array):
elif isinstance(call, pr.Call): for key_stmt, value_stmt in array.items():
yield call.name, value_stmt # first index, is the key if syntactically correct
# Normal arguments (including key arguments). call = key_stmt.expression_list()[0]
if isinstance(call, pr.Name):
yield call, value_stmt
elif isinstance(call, pr.Call):
yield call.name, value_stmt
# Normal arguments (including key arguments).
else:
if stmt.assignment_details:
key_arr, op = stmt.assignment_details[0]
# named parameter
if key_arr and isinstance(key_arr[0], pr.Call):
yield key_arr[0].name, stmt
else: else:
if stmt.assignment_details: yield None, stmt
key_arr, op = stmt.assignment_details[0]
# named parameter
if key_arr and isinstance(key_arr[0], pr.Call):
yield key_arr[0].name, stmt
else:
yield None, stmt
return iter(common.PushBackIterator(iterate()))
class _FakeSubModule(): class _FakeSubModule():
@@ -194,6 +185,5 @@ class _FakeSubModule():
class _FakeStatement(pr.Statement): class _FakeStatement(pr.Statement):
def __init__(self, content): def __init__(self, content):
cls = type(self)
p = 0, 0 p = 0, 0
super(cls, self).__init__(_FakeSubModule, [content], p, p) super(_FakeStatement, self).__init__(_FakeSubModule, [content], p, p)