o
    bc|                     @   sB   d Z ddlmZ ddlmZ ddlmZmZ G dd dejZdS )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 Zd ZdZdZdd ZdS )	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              
   C   s  |sJ | j }|jj|jkr| j|jrd S |d}|d }|j}dd |d D }|d  }|d}	|	r@dd |	D }	|j|j	|j|j
|j|j|j|jfv rWt|}t|d	krb|d
 }nt|j|}d|_tddd}
|rtddd}t|j||
f}
t|j	||
|f}|	rt|}t|j|ft|	 }|jj|j	|j|j|j|j|j|j|j|jf	v rt|}||_|S )Nnegationanchorc                 S      g | ]}|  qS  clone.0nr
   r
   0/usr/lib/python3.10/lib2to3/fixes/fix_has_key.py
<listcomp>R       z'FixHasKey.transform.<locals>.<listcomp>beforeargafterc                 S   r	   r
   r   r   r
   r
   r   r   V   r           in)prefixnot)symsparenttypenot_testpatternmatchgetr   r   
comparisonand_testor_testtestlambdefargumentr   lenr   Nodepowerr   comp_optupleexprxor_exprand_expr
shift_expr
arith_exprtermfactor)selfnoderesultsr   r   r   r   r   r   r   n_opn_notnewr
   r
   r   	transformG   sL   


zFixHasKey.transformN)__name__
__module____qualname__BM_compatiblePATTERNr;   r
   r
   r
   r   r   &   s    r   N)	__doc__ r   r   
fixer_utilr   r   BaseFixr   r
   r
   r
   r   <module>   s
   