3
\| � @ s> d dl mZ d dl mZ d dlmZmZ G dd� dej�ZdS )� )�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)� r r Z
fixer_utilr r ZBaseFixr r r r r
�! s