Fix error when using generators with variable-length arguments

This commit is contained in:
micbou
2018-01-12 17:33:10 +01:00
committed by Dave Halter
parent d815470e54
commit 3a0ac37ee8
2 changed files with 18 additions and 1 deletions

View File

@@ -29,6 +29,8 @@ def try_iter_content(types, depth=0):
class AbstractArguments(object):
context = None
argument_node = None
trailer = None
def eval_argument_clinic(self, parameters):
"""Uses a list with argument clinic information (see PEP 436)."""
@@ -197,7 +199,11 @@ class TreeArguments(AbstractArguments):
arguments = param.var_args
break
return [arguments.argument_node or arguments.trailer]
if arguments.argument_node:
return [arguments.argument_node]
if arguments.trailer:
return [arguments.trailer]
return []
class ValuesArguments(AbstractArguments):

View File

@@ -211,6 +211,17 @@ class X():
#?
self.x()
def decorator_var_args(function, *args):
return function(*args)
@decorator_var_args
def function_var_args(param):
return param
#? int()
function_var_args(1)
# -----------------
# method decorators
# -----------------