a

�DOg|�@sBdZd
dl
mZ
d
dl
mZ
d
dlmZmZ
Gdd�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@seZ
dZd
ZdZdd�ZdS)�	FixHasKeyTa�
    anchor=power<
        before=any+
        trailer< '.' 'has_key' >
        trailer<
            '('
            ( not(arglist | argument<any '=' any>) arg=any
            | arglist<(not argument<any '=' any>) arg=any ','>
            )
            ')'
        >
        after=any*
    >
    |
    negation=not_test<
        'not'
        anchor=power<
            before=any+
            trailer< '.' 'has_key' >
            trailer<
                '('
                ( not(arglist | argument<any '=' any>) arg=any
                | arglist<(not argument<any '=' any>) arg=any ','>
                )
                ')'
            >
        >
    >
    c

Cs�
|sJ�
|j}|
j
j|jkr.|j�|
j
�
r.dS|�d
�
}|d}|
j}dd�|dD�
}|d��}|�d�
}	|	r�dd�|	D�
}	|j|j	|j|j
|j|j|j
|jfvr�t|�
}t|�
d	kr�|d
}nt�|j|�}d|_tddd
�}
|�
rtddd
�}t�|j||
f�}
t�|j	||
|f�}|	�
rBt|�
}t�|j|f
t|	�
�}|
j
j|j	|j|j|j|j|j|j|j|jf	v�
r|t|�
}||_|S)N�negation�anchorc
Ssg|]}
|
���qS��
�clone��.0�
nr	r	�1/usr/lib64/python3.9/lib2to3/fixes/fix_has_key.py�
<listcomp>R�z'FixHasKey.transform.<locals>.<listcomp>�before�arg�afterc
Ssg|]}
|
���qSr	r
rr	r	rrVr�
��
 �in)
�prefix�not)�syms�parent�typeZnot_test�pattern�match�getrrZ
comparisonZand_testZor_test�testZlambdef�argumentr�lenrZNodeZpowerrZcomp_op�tuple�exprZxor_exprZand_exprZ
shift_exprZ
arith_exprZtermZfactor)
�selfZnode�resultsrrrrrrrZn_opZn_not�newr	r	r�	transformGsL



�











�















�

zFixHasKey.transformN)�__name__�
__module__�__qualname__Z
BM_compatibleZPATTERNr)r	r	r	rr&s
rN)	�__doc__�rrZ
fixer_utilrrZBaseFixrr	r	r	r�<module>s