o

��bc�@sNdZd
dl
mZ
d
dlmZmZmZmZmZm	Z	
dZ
dZGdd�dej�Z
dS)	a�
Adjust some old Python 2 idioms to their modern counterparts.

* Change some type comparisons to isinstance() calls:
    type(x) == T -> isinstance(x, T)
    type(x) is T -> isinstance(x, T)
    type(x) != T -> not isinstance(x, T)
    type(x) is not T -> not isinstance(x, T)

* Change "while 1:" into "while True:".

* Change both

    v = list(EXPR)
    v.sort()
    foo(v)

and the more general

    v = EXPR
    v.sort()
    foo(v)

into

    v = sorted(EXPR)
    foo(v)
�)
�
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
|��|
�
}|rd
|vr|d|dkr|SdS|S)N�sorted�id1�id2)�superr	�match)�self�node�
r�
�	__class__��//usr/lib/python3.10/lib2to3/fixes/fix_idioms.pyrOs




zFixIdioms.matchcCsDd
|vr
|�|
|�Sd|vr|�
|
|�Sd|vr|�|
|�Std�
�
)N�
isinstance�whiler
z
Invalid match)�transform_isinstance�transform_while�transform_sort�RuntimeError)rr�resultsrrr�	transformZs





zFixIdioms.transformcCsh|d
��}|d��}d|_
d|_
ttd�
|t�|g�}d|vr.d|_
ttjtd�
|g�}|
j
|_
|S)N�
x�
T��
 r�
n�not)�clone�prefixrrrrr�not_test)rrrrr�testrrrrds









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

dS)Nr�True�
r%)�replacerr%)rrr�onerrrrps

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

n|r7|��}d|_|�
ttd�
|g
|jd��

ntd�
�
|��
|j}d	|vr�|r^|�d	�
d
|d
jf}	d	�	|	�
|d
_dS|j
scJ�
|jdusjJ�
t�}
|j
�
|
�

|j|
uszJ�
|�d	�
d
|
_dSdS)N�sort�next�list�exprr
r)r zshould not have reached here�

�)�getr*rr%r$rr�remove�
rpartition�join�parent�next_siblingr�append_child)rrr�	sort_stmt�	next_stmt�	list_call�simple_expr�new�btwn�prefix_lines�end_linerrrrts4












�






�zFixIdioms.transform_sort)
�__name__�
__module__�__qualname__�explicit�TYPE�CMP�PATTERNrrrrr�
__classcell__rrrrr	%s

%�'
r	N)�__doc__r r�
fixer_utilrrrrrrrFrE�BaseFixr	rrrr�<module>
s