diff options
Diffstat (limited to 'pies/ast.py')
-rw-r--r-- | pies/ast.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/pies/ast.py b/pies/ast.py index 015e24e..ef576eb 100644 --- a/pies/ast.py +++ b/pies/ast.py @@ -1,12 +1,16 @@ from __future__ import absolute_import +import sys from ast import * from .version_info import PY2 -if PY2: +if PY2 or sys.version_info[1] <= 2: Try = TryExcept +else: + TryFinally = () +if PY2: def argument_names(node): return [isinstance(arg, Name) and arg.id or None for arg in node.args.args] @@ -16,8 +20,6 @@ if PY2: def kw_only_default_count(node): return 0 else: - TryFinally = () - def argument_names(node): return [arg.arg for arg in node.args.args] |