3
\� � @ s6 d dl mZmZ d dlmZmZ G dd� dej�ZdS )� )�
fixer_base�pytree)�token�symsc @ s e Zd ZdZdZdZdd� ZdS )�
FixSetLiteralTaj power< 'set' trailer< '('
(atom=atom< '[' (items=listmaker< any ((',' any)* [',']) >
|
single=any) ']' >
|
atom< '(' items=testlist_gexp< any ((',' any)* [',']) > ')' >
)
')' > >
c C s� |j d�}|r2tjtj|j� g�}|j|� |}n|d }tjtj d�g}|j
dd� |j D �� |j tjtj
d�� |jj|d
_tjtj|�}|j|_t|j �dkr�|j d }|j� |j|j d _|S ) N�single�items�{c s s | ]}|j � V qd S )N)�clone)�.0�n� r
�5/usr/lib64/python3.6/lib2to3/fixes/fix_set_literal.py� ' s z*FixSetLiteral.transform..�}� � � �����r )�getr ZNoder Z listmakerr
�replaceZLeafr �LBRACE�extendZchildren�append�RBRACEZ next_sibling�prefixZ dictsetmaker�len�remove) �selfZnodeZresultsr Zfaker �literalZmakerr r
r
r � transform s"
zFixSetLiteral.transformN)�__name__�
__module__� __qualname__Z
BM_compatibleZexplicitZPATTERNr r
r
r
r r s
r N)Zlib2to3r r Zlib2to3.fixer_utilr r ZBaseFixr r
r
r
r � s