3
\| � @ 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| | j }|jj|jkr&| jj|j�r&d S |jd�}|d }|j}dd� |d D �}|d j� }|jd�} | rxdd� | D �} |j|j |j|j
|j |j |j
|jfkr�t|�}t|�d kr�|d
}ntj|j|�}d |_td d d
�}
|r�tdd d
�} tj|j| |
f�}
tj|j ||
|f�} | �r8t| �} tj|j| ft| � �} |jj|j |j|j|j|j|j|j |j |jf k�rrt| �} || _| S )N�negation�anchorc S s g | ] }|j � �qS � )�clone)�.0�nr r �1/usr/lib64/python3.6/lib2to3/fixes/fix_has_key.py�
R s z'FixHasKey.transform..�before�arg�afterc S s g | ] }|j � �qS r )r
)r r r r r
r V s � � � �in)�prefix�not) �syms�parent�typeZnot_test�pattern�match�getr r
Z
comparisonZand_testZor_testZtestZlambdefZargumentr �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 sD
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