3

��2a��@sfdd
lm
Z

ddlmZ
ddlmZ
ddlmZmZmZ
ddlmZ
ejdh
BZ	Gdd�dej
�Zd	S)
�)
�pytree)
�patcomp)
�
fixer_base)�Name�Call�Dot)
�
fixer_util�iterc@s@eZ
dZd
ZdZdd�ZdZeje�
Z	dZ
eje
�
Zdd�Zd	S)
�FixDictTa

    power< head=any+
         trailer< '.' method=('keys'|'items'|'values'|
                              'iterkeys'|'iteritems'|'itervalues'|
                              'viewkeys'|'viewitems'|'viewvalues') >
         parens=trailer< '(' ')' >
         tail=any*
    >
    c
Cs
|d
}|dd}|d}|j}|j
}|jd�
}|jd�
}	|sD|	rP|dd�}dd	�|D�
}d
d	�|D�
}|o||j|
|�}
|tj|jt�t||j	d�g�|dj
�g}tj|j|�}|
p�|	s�d
|_	tt|r�dnd�
|g
�}|r�tj|j|g
|�}|
j	|_	|S)N�head�method��tailr	Zview�c
Ssg|]}
|
j��qS�)
�clone)�.0�
nrr�-/opt/alt/python36/lib64/python3.6/fix_dict.py�
<listcomp>Asz%FixDict.transform.<locals>.<listcomp>c
Ssg|]}
|
j��qSr)
r)rrrrrrBs)
�prefixZparens��list)
�syms�value�
startswith�in_special_contextrZNodeZtrailerrrrrZpowerr)
�self�node�resultsrrrrZmethod_name�isiterZisviewZspecial�args�newrrr�	transform6s2


























zFixDict.transformz3power< func=NAME trailer< '(' node=any ')' > any* >zmfor_stmt< 'for' any 'in' node=any ':' any* >
            | comp_for< 'for' any 'in' node=any any* >
         cCs�|
jdkrd
Si}|
jjdk	r^|j
j|
jj|�r^|d|
kr^|rN|djtkS|djtjkS|sfd
S|jj|
j|�o�|d|
kS)NFr�func)�parent�p1�matchr�iter_exemptr�consuming_calls�p2)rrr rrrrrZs









zFixDict.in_special_contextN)
�__name__�
__module__�__qualname__Z
BM_compatibleZPATTERNr#ZP1rZcompile_patternr&ZP2r*rrrrrr
)s





r
N)rrrrrrrrr)r(ZBaseFixr
rrrr�<module>s