a

�0`��@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�|�d
�
}|r2t
�tj|��g
�}|�|�

|}n|d}t
�tj	d�g
}|�
dd�|jD�
�

|�t
�tj
d��

|jj|d_t
�tj|�}|
j|_t|j�
dkr�|jd	}|��
|j|jd_|S)
N�single�items�
{c
ss|]}
|
��V
qdS)
N)
�clone)�.0�
n�r
�3/usr/lib/python3.9/lib2to3/fixes/fix_set_literal.py�	<genexpr>'�z*FixSetLiteral.transform.<locals>.<genexpr>�
}�����)�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