a
XC?h � @ sN d Z ddlmZ ddlmZmZmZmZmZm Z dZ
dZ G dd� 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 ')' > >c sP e Zd ZdZdeeeef Z� fdd�Zdd� Zdd� 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 ')' > >
>
'\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| ��|�}|r4d|v r4|d |d kr0|S d S |S )N�sortedZid1Zid2)�superr �match)�self�node�r�� __class__� �=/opt/alt/python39/lib64/python3.9/lib2to3/fixes/fix_idioms.pyr O s zFixIdioms.matchc C sH d|v r| � ||�S d|v r(| �||�S d|v r<| �||�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 � � }|d � � }d|_d|_ttd�|t� |g�}d|v r\d|_ttjtd�|g�}|j|_|S )N�x�T� � r �n�not)�clone�prefixr r r r r Znot_test)r
r r r r �testr r r r d s z FixIdioms.transform_isinstancec C s |d }|� td|jd�� d S )Nr �True�r# )�replacer r# )r
r r Zoner r r r p s zFixIdioms.transform_whilec C s |d }|d }|� d�}|� d�}|r>|�td|jd�� n8|rn|�� }d|_|�ttd�|g|jd�� ntd��|�� |j}d |v �r|r�|�d �d
|d
jf} d � | �|d
_nH|j
s�J �|j d u s�J �t � }
|j
�
|
� |j |
u s�J �|�d �d
|
_d S ) N�sort�next�list�exprr
r& r z should not have reached here�
� )�getr' r r# r" r r �remove�
rpartition�join�parentZ next_siblingr Z append_child) r
r r Z sort_stmtZ next_stmtZ list_callZ simple_expr�newZbtwnZ prefix_linesZend_liner r r r t s2
�
zFixIdioms.transform_sort)
�__name__�
__module__� __qualname__Zexplicit�TYPE�CMPZPATTERNr r r r r �
__classcell__r r r r r % s %
�'
r N)�__doc__r r Z
fixer_utilr r r r r r r8 r7 ZBaseFixr r r r r � s