� j f��@s�dZddlmZddlmZddlmZddlmZddlmZm Z m Z mZmZm Z ddlmZejdhBZGd d �d ej�ZdS)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)�token)� fixer_base)�Name�Call�LParen�RParen�ArgList�Dot)� fixer_util�iterc@s^eZdZdZdZdd�ZdZeje�Z dZ eje �Zdd�Zd S) �FixDictTa power< head=any+ trailer< '.' method=('keys'|'items'|'values'| 'iterkeys'|'iteritems'|'itervalues'| 'viewkeys'|'viewitems'|'viewvalues') > parens=trailer< '(' ')' > tail=any* > c Csw|d}|dd}|d}|j}|j}|jd�}|jd�} |s^| rq|dd�}ndd �|D�}d d �|D�}|o�|j||�} |tj|jt�t|d|j �g�|dj �g}tj|j|�}| p| sBd |_ tt|r-dnd�|g�}n|rgtj|j|g|�}n|j |_ |S)N�head�method��tailr Zview�cSsg|]}|j��qS�)�clone)�.0�nrr�;/opt/alt/python34/lib64/python3.4/lib2to3/fixes/fix_dict.py� Bs z%FixDict.transform..cSsg|]}|j��qSr)r)rrrrrrCs �prefixZparens��list) �syms�value� startswith�in_special_contextrZNodeZtrailerrrrrZpowerr) �self�node�resultsrrrrZmethod_name�isiterZisviewZspecial�args�newrrr� transform7s2 '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�|jdkrdSi}|jjdk r�|jj|jj|�r�|d|kr�|rm|djtkS|djtjkSn|s�dS|jj|j|�o�|d|kS)NFr"�func)�parent�p1�matchr�iter_exemptr�consuming_calls�p2)r!r"r$r#rrrr [szFixDict.in_special_contextN) �__name__� __module__�__qualname__Z BM_compatibleZPATTERNr'ZP1rZcompile_patternr*ZP2r.r rrrrr*s rN)�__doc__rrrZpgen2rrrrrrr r rr-r,ZBaseFixrrrrr�s.