3
\H � @ s2 d Z ddlmZ ddlmZ G dd� dej�ZdS )a, Fixer that cleans up a tuple argument to isinstance after the tokens
in it were fixed. This is mainly used to remove double occurrences of
tokens as a leftover of the long -> int / unicode -> str conversion.
eg. isinstance(x, (int, long)) -> isinstance(x, (int, int))
-> isinstance(x, int)
� )�
fixer_base)�tokenc @ s e Zd ZdZdZdZdd� ZdS )�
FixIsinstanceTz�
power<
'isinstance'
trailer< '(' arglist< any ',' atom< '('
args=testlist_gexp< any+ >
')' > > ')' >
>
� c C s� t � }|d }|j}g }t|�}xx|D ]p\}} | jtjkrt| j|krt|t|�d k r�||d jtjkr�t |� q&q&|j
| � | jtjkr&|j | j� q&W |r�|d jtjkr�|d= t|�dkr�|j }
|
j
|d _
|
j|d � n||d d �< |j� d S )N�args� � ���r )�setZchildren� enumerate�typer �NAME�value�len�COMMA�next�append�add�parent�prefix�replaceZchanged) �selfZnodeZresultsZnames_insertedZtestlistr Znew_args�iterator�idx�argZatom� r �4/usr/lib64/python3.6/lib2to3/fixes/fix_isinstance.py� transform s* $
zFixIsinstance.transformN)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNZ run_orderr r r r r r s r N)�__doc__� r Z
fixer_utilr ZBaseFixr r r r r �<module>
s