Fix property access of vararg and kwarg.

This commit is contained in:
Toshihiko Yanase 2020-02-19 22:21:34 +09:00
parent 78aac416a4
commit 0ee64fa40c

View File

@ -55,7 +55,7 @@ def signature_from_ast(node: ast.FunctionDef, bound_method: bool,
if node.args.vararg: if node.args.vararg:
param = Parameter(node.args.vararg.arg, Parameter.VAR_POSITIONAL, param = Parameter(node.args.vararg.arg, Parameter.VAR_POSITIONAL,
annotation=arg.type_comment or Parameter.empty) annotation=node.args.vararg.type_comment or Parameter.empty)
params.append(param) params.append(param)
for arg in node.args.kwonlyargs: for arg in node.args.kwonlyargs:
@ -65,7 +65,7 @@ def signature_from_ast(node: ast.FunctionDef, bound_method: bool,
if node.args.kwarg: if node.args.kwarg:
param = Parameter(node.args.kwarg.arg, Parameter.VAR_KEYWORD, param = Parameter(node.args.kwarg.arg, Parameter.VAR_KEYWORD,
annotation=arg.type_comment or Parameter.empty) annotation=node.args.kwarg.type_comment or Parameter.empty)
params.append(param) params.append(param)
# Remove first parameter when *obj* is bound_method # Remove first parameter when *obj* is bound_method