diff options
Diffstat (limited to 'Lib/lib2to3/fixes/fix_paren.py')
-rw-r--r-- | Lib/lib2to3/fixes/fix_paren.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/Lib/lib2to3/fixes/fix_paren.py b/Lib/lib2to3/fixes/fix_paren.py new file mode 100644 index 0000000..0b72bb1 --- /dev/null +++ b/Lib/lib2to3/fixes/fix_paren.py @@ -0,0 +1,42 @@ +"""Fixer that addes parentheses where they are required + +This converts ``[x for x in 1, 2]`` to ``[x for x in (1, 2)]``.""" + +# By Taek Joo Kim and Benjamin Peterson + +# Local imports +from .. import fixer_base +from ..fixer_util import LParen, RParen + +# XXX This doesn't support nested for loops like [x for x in 1, 2 for x in 1, 2] +class FixParen(fixer_base.BaseFix): + PATTERN = """ + atom< ('[' | '(') + (listmaker< any + comp_for< + 'for' NAME 'in' + target=testlist_safe< any (',' any)+ [','] + > + [any] + > + > + | + testlist_gexp< any + comp_for< + 'for' NAME 'in' + target=testlist_safe< any (',' any)+ [','] + > + [any] + > + >) + (']' | ')') > + """ + + def transform(self, node, results): + target = results["target"] + + lparen = LParen() + lparen.set_prefix(target.get_prefix()) + target.set_prefix("") # Make it hug the parentheses + target.insert_child(0, lparen) + target.append_child(RParen()) |