��Yf��@sIdZd
dl
mZ
d
dlmZmZ
Gdd�dej�ZdS)zuFixer that addes parentheses where they are required

This converts ``[x for x in 1, 2]`` to ``[x for x in (1, 2)]``.�)
�
fixer_base)�LParen�RParenc@s(eZ
dZd
ZdZdd�ZdS)�FixParenTa
        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]
                >
            >)
        (']' | ')') >
    cCsL|d
}t�}|j
|_
d|_
|jd|�
|jt��

dS)N�target��)r�prefixZinsert_childZappend_childr)�selfZnodeZresultsrZlparen�r�</opt/alt/python35/lib64/python3.5/lib2to3/fixes/fix_paren.py�	transform%s

	

	

zFixParen.transformN)�__name__�
__module__�__qualname__Z
BM_compatibleZPATTERNr
rrrrrs
rN)�__doc__rrZ
fixer_utilrrZBaseFixrrrrr�<module>s