3
�
�^ c @ sJ d dl mZ d dlmZmZmZmZmZmZ dZ dZ
G dd� dej�ZdS )i )�
fixer_base)�Call�Comma�Name�Node� BlankLine�symsz0(n='!=' | '==' | 'is' | n=comp_op< 'is' 'not' >)z(power< 'type' trailer< '(' x=any ')' > >c sP e Zd ZdZdeeeef Z� fdd�Zdd� Zdd� Z d d
� Z
dd� Z� ZS )
� FixIdiomsTa�
isinstance=comparison< %s %s T=any >
|
isinstance=comparison< T=any %s %s >
|
while_stmt< 'while' while='1' ':' any+ >
|
sorted=any<
any*
simple_stmt<
expr_stmt< id1=any '='
power< list='list' trailer< '(' (not arglist<any+>) any ')' > >
>
'\n'
>
sort=
simple_stmt<
power< id2=any
trailer< '.' 'sort' > trailer< '(' ')' >
>
'\n'
>
next=any*
>
|
sorted=any<
any*
simple_stmt< expr_stmt< id1=any '=' expr=any > '\n' >
sort=
simple_stmt<
power< id2=any
trailer< '.' 'sort' > trailer< '(' ')' >
>
'\n'
>
next=any*
>
c s8 t t| �j|�}|r4d|kr4|d |d kr0|S d S |S )N�sortedZid1Zid2)Zsuperr �match)�self�nodeZr�Z __class__� �0/usr/lib64/python3.6/lib2to3/fixes/fix_idioms.pyr O s zFixIdioms.matchc C sH d|kr| j ||�S d|kr(| j||�S d|kr<| j||�S td��d S )N�
isinstance�whiler z
Invalid match)�transform_isinstance�transform_while�transform_sort�RuntimeError)r
r �resultsr
r
r � transformZ s zFixIdioms.transformc C sh |d j � }|d j � }d|_d|_ttd�|t� |g�}d|kr\d|_ttjtd�|g�}|j|_|S )N�x�T� z r ZnZnot)�clone�prefixr r r r r Znot_test)r
r r r r Ztestr
r
r r d s zFixIdioms.transform_isinstancec C s |d }|j td|jd�� d S )Nr ZTrue�r )�replacer r )r
r r Zoner
r
r r p s zFixIdioms.transform_whilec C s� |d }|d }|j d�}|j d�}|r>|jtd|jd�� n8|rn|j� }d|_|jttd�|g|jd�� ntd��|j� |j}d |kr�|r�|jd �d
|d
jf} d j | �|d
_n"t
� }
|jj|
� |jd �d
|
_d S )NZsortZnextZlistZexprr r r zshould not have reached hereZ
i )
Zgetr r r r r r ZremoveZ
rpartitionZjoinr ZparentZappend_child)r
r r Z sort_stmtZ next_stmtZ list_callZsimple_exprZnewZbtwnZprefix_linesZend_liner
r
r r t s*
zFixIdioms.transform_sort)
Z__name__Z
__module__Z__qualname__Zexplicit�TYPE�CMPZPATTERNr r r r r Z
__classcell__r
r
r r r % s '
r N)
r r Z
fixer_utilr r r r r r r r ZBaseFixr r
r
r
r Z<module> s