] @ s d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z m Z m Z m
Z
m Z m Z d d l m Z e j
d h BZ Gd d d e j Z d
S) )pytree)patcomp)token)
fixer_base)NameCallLParenRParenArgListDot)
fixer_utiliterc @ s^ e Z d Z d Z d Z d d Z d Z e j e Z d Z
e j e
Z d d Z d S)
FixDictTa
power< head=any+
trailer< '.' method=('keys'|'items'|'values'|
'iterkeys'|'iteritems'|'itervalues'|
'viewkeys'|'viewitems'|'viewvalues') >
parens=trailer< '(' ')' >
tail=any*
>
c
C sn | d } | d d } | d } | j } | j } | j d } | j d } | s^ | rn | d d } d d | D } d
d | D } | o | j | | }
| t j | j t t | d | j g | d j
g } t j | j | } |
p | s<d
| _ t t | r*d n d | g } | r^t j | j | g | } | j | _ | S)Nheadmethod tailr
view c S s g | ] } | j q S )clone).0nr r -/opt/alt/python35/lib64/python3.5/fix_dict.py
<listcomp>B s z%FixDict.transform.<locals>.<listcomp>c S s g | ] } | j q Sr )r )r r r r r r C s prefixZparens list)
symsvalue
startswithin_special_contextr ZNodeZtrailerr r r r Zpowerr )
selfnoderesultsr r r r Z method_nameisiterZisviewZspecialargsnewr r r transform7 s2
$ zFixDict.transformz3power< func=NAME trailer< '(' node=any ')' > any* >zmfor_stmt< 'for' any 'in' node=any ':' any* >
| comp_for< 'for' any 'in' node=any any* >
c C s | j d k r d Si } | j j d k r | j j | j j | r | d | k r | rm | d j t k S| d j t j k S| s d S| j j | j | o | d | k S)NFr# func)parentp1matchr iter_exemptr consuming_callsp2)r" r# r% r$ r r r r! [ s zFixDict.in_special_contextN)
__name__
__module__ __qualname__Z
BM_compatibleZPATTERNr( ZP1r Zcompile_patternr+ ZP2r/ r! r r r r r * s
r N)r r r Zpgen2r r r r r r r r
r r. r- ZBaseFixr r r r r <module> s .