B

�5�a�@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
|��|
�
}|r4d
|kr4|d|dkr0|SdS|S)N�sortedZid1Zid2)�superr	�match)�self�node�
r)
�	__class__��./usr/lib/python3.7/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	|k�
r|r�|�d	�
d
|d
jf}	d	�	|	�
|d
_nH|j
s�t�
|jdks�t�
t
�}
|j
�|
�

|j|
ks�t�
|�d	�
d
|
_dS)N�sort�next�list�exprr
)
r"rzshould not have reached here�

�)�getr$rr"r!rr�remove�
rpartition�join�parent�AssertionErrorZnext_siblingrZappend_child)r
rrZ	sort_stmtZ	next_stmtZ	list_callZsimple_expr�newZbtwnZprefix_linesZend_linerrrrts0



















zFixIdioms.transform_sort)
�__name__�
__module__�__qualname__Zexplicit�TYPE�CMPZPATTERNrrrrr�
__classcell__rr)
rrr	%s
'
r	N)�__doc__rrZ
fixer_utilrrrrrrr6r5ZBaseFixr	rrrr�<module>s