U i�f��@sjdZddlmZddlmZddlmZddlmZmZmZddlmZej dhBZ Gdd �d ej�Zd 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@eZdZdZdZdd�ZdZe�e�Z dZ e�e �Zdd�Zd 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�}dd �|D�}d d �|D�}|o||�||�} |t�|jt�t||j d�g�|d� �g}t�|j|�}| s�| s�d |_ tt|r�dnd�|g�}|r�t�|j|g|�}|j |_ |S)N�head�method��tailr Zview�cSsg|]}|��qS��clone��.0�nrr�;/opt/alt/python38/lib64/python3.8/lib2to3/fixes/fix_dict.py� Asz%FixDict.transform..cSsg|]}|��qSrrrrrrrBs)�prefixZparens��list) �syms�value� startswith�in_special_contextrZNodeZtrailerrrrrZpowerr) �self�node�resultsrrrrZmethod_name�isiterZisviewZspecial�args�newrrr� transform6s: �� 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^|j�|jj|�r^|d|kr^|rN|djtkS|djtjkS|sfdS|j�|j|�o�|d|kS)NFr �func)�parent�p1�matchr�iter_exemptr�consuming_calls�p2)rr r"r!rrrrZs � �zFixDict.in_special_contextN) �__name__� __module__�__qualname__Z BM_compatibleZPATTERNr%ZP1rZcompile_patternr(ZP2r,rrrrrr )s r N) �__doc__rrrrrrrrr+r*ZBaseFixr rrrr�s