U
    A`|                     @   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t | j}|jj|jkr.| j|jr.d S |d}|d }|j}dd |d D }|d 	 }|d}	|	rdd |	D }	|j|j
|j|j|j|j|j|jfkrt|}t|d	kr|d
 }nt|j|}d|_tddd}
|rtddd}t|j||
f}
t|j
||
|f}|	rBt|}t|j|ft|	 }|jj|j
|j|j|j|j|j|j|j|jf	kr|t|}||_|S )Nnegationanchorc                 S   s   g | ]}|  qS  clone.0nr	   r	   //usr/lib/python3.8/lib2to3/fixes/fix_has_key.py
<listcomp>R   s     z'FixHasKey.transform.<locals>.<listcomp>beforeargafterc                 S   s   g | ]}|  qS r	   r
   r   r	   r	   r   r   V   s             in)prefixnot)AssertionErrorsymsparenttypeZnot_testpatternmatchgetr   r   Z
comparisonZand_testZor_testZtestZlambdefZargumentr   lenr   ZNodeZpowerr   Zcomp_optupleexprZxor_exprZand_exprZ
shift_exprZ
arith_exprZtermZfactor)selfZnodeZresultsr   r   r   r   r   r   r   Zn_opZn_notnewr	   r	   r   	transformG   sX    

   
   zFixHasKey.transformN)__name__
__module____qualname__ZBM_compatibleZPATTERNr&   r	   r	   r	   r   r   &   s   r   N)	__doc__ r   r   Z
fixer_utilr   r   ZBaseFixr   r	   r	   r	   r   <module>   s   