B

�5�a|�@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�
|st�
|j
}|
jj|jkr.|j�|
j�
r.dS|�d
�
}|d}|
j}dd�|dD�
}|d�	�}|�d�
}	|	r�dd�|	D�
}	|j|j
|j|j|j|j
|j|jfkr�t|�
}t|�
d	kr�|d
}nt�|j|�}d|_tddd
�}
|�
rtddd
�}t�|j||
f�}
t�|j
||
|f�}|	�
rBt|�
}t�|j|f
t|	�
�}|
jj|j
|j|j|j|j|j|j|j|jf	k�
r|t|�
}||_|S)N�negation�anchorc
Ssg|]}
|
���qS�)
�clone)�.0�
nr	r	�//usr/lib/python3.7/lib2to3/fixes/fix_has_key.py�
<listcomp>Rsz'FixHasKey.transform.<locals>.<listcomp>�before�arg�afterc
Ssg|]}
|
���qSr	)
r
)rrr	r	r
rVs�
��
 �in)
�prefix�not)�AssertionError�syms�parent�typeZnot_test�pattern�match�getrr
Z
comparisonZand_testZor_testZtestZlambdefZargumentr�lenrZNodeZpowerrZcomp_op�tuple�exprZxor_exprZand_exprZ
shift_exprZ
arith_exprZtermZfactor)
�selfZnodeZresultsrrrrrrrZn_opZn_not�newr	r	r
�	transformGsF



































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