U AT^c@sjdZddlmZddlmZddlmZddlmZmZmZddlmZej dhBZ 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). i)pytree)patcomp) fixer_base)NameCallDot) fixer_utiliterc@s@eZdZdZdZddZdZeeZ dZ ee Z ddZ 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}d d |D}| o|||} |t|jt t ||j d g|d  g} t|j | } | s| sd| _ tt |rdnd| g} |rt|j | g|} |j | _ | S)NheadmethoditailrZviewi)ZkeysZitemsZvaluescSg|] }|qScloneZ.0Znr r ./usr/lib64/python3.8/lib2to3/fixes/fix_dict.py Az%FixDict.transform..cSr r rrr r rrBr)prefixZparensZlist)symsvalueZ startswithZAssertionErrorZreprin_special_contextrZNodeZtrailerrrrrZpowerr) selfnoderesultsr r r rZ method_nameisiterZisviewZspecialZargsZnewr r r 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)NFrZfunc)Zparentp1Zmatchr iter_exemptrconsuming_callsp2)rrrrr r rrZs   zFixDict.in_special_contextN) Z__name__Z __module__Z __qualname__Z BM_compatibleZPATTERNrZP1rZcompile_patternrZP2r!rr r r rr)s   rN) Z__doc__rrrrrrrrr rZBaseFixrr r r rZs