o
    Š�bc  ã                   @   sN   d Z ddlmZ ddlmZmZmZmZmZm	Z	 dZ
dZG dd„ dejƒZdS )	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 ')' > >c                       sP   e Zd ZdZdeeeef Z‡ fdd„Zdd„ Zdd„ Z	d	d
„ Z
dd„ 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*
        >
    c                    s8   t t| ƒ |¡}|rd|v r|d |d kr|S d S |S )NÚsortedÚid1Úid2)Úsuperr	   Úmatch)ÚselfÚnodeÚr©Ú	__class__© ú//usr/lib/python3.10/lib2to3/fixes/fix_idioms.pyr   O   s   zFixIdioms.matchc                 C   sD   d|v r
|   ||¡S d|v r|  ||¡S d|v r|  ||¡S tdƒ‚)NÚ
isinstanceÚwhiler
   zInvalid match)Útransform_isinstanceÚtransform_whileÚtransform_sortÚRuntimeError)r   r   Úresultsr   r   r   Ú	transformZ   s   zFixIdioms.transformc                 C   sh   |d   ¡ }|d   ¡ }d|_d|_ttdƒ|tƒ |gƒ}d|v r.d|_ttjtdƒ|gƒ}|j|_|S )NÚxÚTÚ ú r   ÚnÚnot)ÚcloneÚprefixr   r   r   r   r   Únot_test)r   r   r   r   r   Útestr   r   r   r   d   s   zFixIdioms.transform_isinstancec                 C   s    |d }|  td|jd�¡ d S )Nr   ÚTrue©r%   )Úreplacer   r%   )r   r   r   Úoner   r   r   r   p   s   zFixIdioms.transform_whilec                 C   s  |d }|d }|  d¡}|  d¡}|r| td|jd�¡ n|r7| ¡ }d|_| ttdƒ|g|jd�¡ ntdƒ‚| ¡  |j}d	|v r„|r^| d	¡d
 |d
 jf}	d	 	|	¡|d
 _d S |j
scJ ‚|jd u sjJ ‚tƒ }
|j
 |
¡ |j|
u szJ ‚| d	¡d
 |
_d S d S )NÚsortÚnextÚlistÚexprr
   r)   r    zshould not have reached hereÚ
é    )Úgetr*   r   r%   r$   r   r   ÚremoveÚ
rpartitionÚjoinÚparentÚnext_siblingr   Úappend_child)r   r   r   Ú	sort_stmtÚ	next_stmtÚ	list_callÚsimple_exprÚnewÚbtwnÚprefix_linesÚend_liner   r   r   r   t   s4   


ÿ
ðzFixIdioms.transform_sort)Ú__name__Ú
__module__Ú__qualname__ÚexplicitÚTYPEÚCMPÚPATTERNr   r   r   r   r   Ú__classcell__r   r   r   r   r	   %   s    
%Û'
r	   N)Ú__doc__r    r   Ú
fixer_utilr   r   r   r   r   r   rF   rE   ÚBaseFixr	   r   r   r   r   Ú<module>   s     