diff options
Diffstat (limited to 'pies/ast.py')
-rw-r--r-- | pies/ast.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/pies/ast.py b/pies/ast.py index 43739a1..3821960 100644 --- a/pies/ast.py +++ b/pies/ast.py @@ -6,5 +6,23 @@ from .version_info import PY2 if PY2: Try = TryExcept + + def argument_names(node): + return [isinstance(arg, Name) and arg.id or None for arg in node.args.args] + + def kw_only_argument_names(node): + return [] + + def kw_only_default_count(node): + return 0 else: TryFinally = () + + def argument_names(node): + return [arg.arg for arg in node.args.args] + + def kw_only_argument_names(self): + return [arg.arg for arg in node.args.kwonlyargs] + + def kw_only_default_count(self): + return sum(1 for n in node.args.kw_defaults if n is not None) |