3


 \��@s:dZd
dl
mZmZ
d
dlmZmZ
Gdd�dej�ZdS)z:
Optional fixer to transform set() calls to set literals.
�)�
fixer_base�pytree)�token�symsc@s eZ
dZd
Zd
ZdZdd�ZdS)�
FixSetLiteralTaj
power< 'set' trailer< '('
                     (atom=atom< '[' (items=listmaker< any ((',' any)* [',']) >
                                |
                                single=any) ']' >
                     |
                     atom< '(' items=testlist_gexp< any ((',' any)* [',']) > ')' >
                     )
                     ')' > >
              c	Cs�|jd
�
}|r2t
jtj|j�g
�}|j|�

|}n|d}t
jtj	d�g
}|j
dd�|jD�
�

|jt
jtj
d��

|jj|d
_t
jtj|�}|
j|_t|j�
dkr�|jd	}|j�
|j|jd_|S)N�single�items�
{c
ss|]}
|
j�V
qdS)
N)
�clone)�.0�
n�r
�5/usr/lib64/python3.6/lib2to3/fixes/fix_set_literal.py�	<genexpr>'sz*FixSetLiteral.transform.<locals>.<genexpr>�
}�
�����r)�getrZNoderZ	listmakerr
�replaceZLeafr�LBRACE�extendZchildren�append�RBRACEZnext_sibling�prefixZdictsetmaker�len�remove)	�selfZnodeZresultsrZfaker�literalZmakerrr
r
r�	transforms"













zFixSetLiteral.transformN)�__name__�
__module__�__qualname__Z
BM_compatibleZexplicitZPATTERNr r
r
r
rrs

rN)	�__doc__Zlib2to3rrZlib2to3.fixer_utilrrZBaseFixrr
r
r
r�<module>s