U

e5d�@sJdd
lm
Z

ddlmZmZmZmZmZmZ
dZ	dZ
Gdd�de
j�ZdS)�)
�
fixer_base)�Call�Comma�Name�Node�	BlankLine�symsz0(n='!=' | '==' | 'is' | n=comp_op< 'is' 'not' >)z(power< 'type' trailer< '(' x=any ')' > >csPeZ
dZd
ZdeeeefZ�f
dd�Zdd�Zdd�Z	d	d
�Z
dd�Z�ZS)
�	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*
        >
    cs8tt
|��|
�
}|r4d
|kr4|d|dkr0|SdS|S)N�sortedZid1Zid2)�superr	�match)�self�node�
r�
�	__class__��0/usr/lib64/python3.8/lib2to3/fixes/fix_idioms.pyrOs




zFixIdioms.matchcCsHd
|kr|�|
|�Sd|kr(|�
|
|�Sd|kr<|�|
|�Std�
�
dS)N�
isinstance�whiler
z
Invalid match)�transform_isinstance�transform_while�transform_sort�RuntimeError)r
r�resultsrrr�	transformZs





zFixIdioms.transformcCsh|d
��}|d��}d|_
d|_
ttd�
|t�|g�}d|kr\d|_
ttjtd�
|g�}|
j
|_
|S)N�
x�
T��
 r�
n�not)�clone�prefixrrrrrZnot_test)r
rrrrZtestrrrrds









zFixIdioms.transform_isinstancecCs |d
}|�t
d|jd��

dS)Nr�True�
r#)�replacerr#)r
rrZonerrrrps

zFixIdioms.transform_whilecCs�|d
}|d}|�d�
}|�d�
}|r>|�
td|jd��

n8|rn|��}d|_|�
ttd�
|g
|jd��

ntd�
�
|��
|j}d	|kr�|r�|�d	�
d
|d
jf}	d	�	|	�
|d
_n"t
�}
|j�|
�

|�d	�
d
|
_dS)N�sort�next�list�exprr
r%rzshould not have reached here�

�)
�getr&rr#r"rr�remove�
rpartition�joinr�parentZappend_child)r
rrZ	sort_stmtZ	next_stmtZ	list_callZsimple_expr�newZbtwnZprefix_linesZend_linerrrrts,











�




zFixIdioms.transform_sort)
�__name__�
__module__�__qualname__Zexplicit�TYPE�CMPZPATTERNrrrrr�
__classcell__rrrrr	%s
%
�'
r	N)
rrZ
fixer_utilrrrrrrr7r6ZBaseFixr	rrrr�<module>s