B

�5�a��@sjdZd
dl
mZ
d
dl
mZ
d
dl
mZ
d
dlmZmZmZ
d
dl
mZ
ej	dh
BZ
Gdd	�d	ej�Zd
S)ajFixer for dict methods.

d.keys() -> list(d.keys())
d.items() -> list(d.items())
d.values() -> list(d.values())

d.iterkeys() -> iter(d.keys())
d.iteritems() -> iter(d.items())
d.itervalues() -> iter(d.values())

d.viewkeys() -> d.keys()
d.viewitems() -> d.items()
d.viewvalues() -> d.values()

Except in certain very specific contexts: the iter() can be dropped
when the context is list(), sorted(), iter() or for...in; the list()
can be dropped when the context is list() or sorted() (but not iter()
or for...in!). Special contexts that apply to both: list(), sorted(), tuple()
set(), any(), all(), sum().

Note: iter(d.keys()) could be written as iter(d) but since the
original d.iterkeys() was also redundant we don't fix this.  And there
are (rare) contexts where it makes a difference (e.g. when passing it
as an argument to a function that introspects the argument).
�)
�pytree)
�patcomp)
�
fixer_base)�Name�Call�Dot)
�
fixer_util�iterc@s@eZ
dZd
ZdZdd�ZdZe�e�
Z	dZ
e�e
�
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
}|�d�
}|�d�
}	|sD|	rP|dd�}|dksdtt|�
�
�
d	d
�|D�
}dd
�|D�
}|o�|�|
|�}
|t�|jt	�t
||jd�g�|d
��g}t�|j
|�}|
s�|	s�d|_tt
|r�dnd�
|g
�}|�
rt�|j
|g
|�}|
j|_|S)N�head�method��tailr	Zview�)�keys�items�valuesc
Ssg|]}
|
���qS�)
�clone)�.0�
nrr�,/usr/lib/python3.7/lib2to3/fixes/fix_dict.py�
<listcomp>Asz%FixDict.transform.<locals>.<listcomp>c
Ssg|]}
|
���qSr)
r)rrrrrrBs)
�prefixZparens��list)�syms�value�
startswith�AssertionError�repr�in_special_contextrZNodeZtrailerrrrrZpowerr)
�self�node�resultsrrrrZmethod_name�isiterZisviewZspecial�args�newrrr�	transform6s4




























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
�|
jj|�r^|d|
kr^|rN|djtkS|djtjkS|sfd
S|j�|
j|�o�|d|
kS)NFr#�func)�parent�p1�matchr�iter_exemptr�consuming_calls�p2)r"r#r%r$rrrr!Zs









zFixDict.in_special_contextN)
�__name__�
__module__�__qualname__Z
BM_compatibleZPATTERNr(ZP1rZcompile_patternr+ZP2r/r!rrrrr
)s





r
N)
�__doc__rrrrrrrrr.r-ZBaseFixr
rrrr�<module>s