a
XC?h| � @ sB d Z ddlmZ ddlmZ ddlmZmZ G dd� dej�ZdS )a& Fixer for has_key().
Calls to .has_key() methods are expressed in terms of the 'in'
operator:
d.has_key(k) -> k in d
CAVEATS:
1) While the primary target of this fixer is dict.has_key(), the
fixer will change any has_key() method call, regardless of its
class.
2) Cases like this will not be converted:
m = d.has_key
if m(k):
...
Only *calls* to has_key() are converted. While it is possible to
convert the above to something like
m = d.__contains__
if m(k):
...
this is currently not done.
� )�pytree)�
fixer_base)�Name� parenthesizec @ s e Zd ZdZdZdd� ZdS )� FixHasKeyTa�
anchor=power<
before=any+
trailer< '.' 'has_key' >
trailer<
'('
( not(arglist | argument) arg=any ','>
)
')'
>
after=any*
>
|
negation=not_test<
'not'
anchor=power<
before=any+
trailer< '.' 'has_key' >
trailer<
'('
( not(arglist | argument) arg=any ','>
)
')'
>
>
>
c
C s� |sJ �| j }|jj|jkr.| j�|j�r.d S |�d�}|d }|j}dd� |d D �}|d �� }|�d�} | r�dd� | D �} |j|j |j|j
|j |j |j
|jfv r�t|�}t|�d kr�|d
}nt�|j|�}d |_td d d
�}
|�rtdd d
�} t�|j| |
f�}
t�|j ||
|f�} | �rBt| �} t�|j| ft| � �} |jj|j |j|j|j|j|j|j |j |jf v �r|t| �} || _| S )N�negation�anchorc S s g | ] }|� � �qS � ��clone��.0�nr r �>/opt/alt/python39/lib64/python3.9/lib2to3/fixes/fix_has_key.py�
R � z'FixHasKey.transform..�before�arg�afterc S s g | ] }|� � �qS r r
r r r r r V r � � � �in)�prefix�not) �syms�parent�typeZnot_test�pattern�match�getr r Z
comparisonZand_testZor_test�testZlambdefZargumentr �lenr ZNodeZpowerr Zcomp_op�tuple�exprZxor_exprZand_exprZ
shift_exprZ
arith_exprZtermZfactor)
�selfZnodeZresultsr r r r r r r Zn_opZn_not�newr r r � transformG sL �
�
�zFixHasKey.transformN)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNr' r r r r r & s r N) �__doc__� r r Z
fixer_utilr r ZBaseFixr r r r r � s