
\[\                 @   s   d  Z  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 e d g  BZ Gd	 d
   d
 e j  Z d S)aj  Fixer 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^   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   s  | d } | d d } | d } |  j  } | j } | j d  } | j d  }	 | s^ |	 rq | d d   } n  | d k s t t |    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
 | rKd n d  | g  } n  | rt j | j | g |  } n  | j | _ | S)Nheadmethod    tailr   Zview   keysitemsvaluesc             S   s   g  |  ] } | j     q S )clone).0nr   r   ,/usr/lib/python3.4/lib2to3/fixes/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   	 prefixZparens list)zkeyszitemszvalues)symsvalue
startswithAssertionErrorreprin_special_contextr   ZNodeZtrailerr   r   r   r   Zpowerr   )selfnoderesultsr   r   r   r    Zmethod_nameisiterZisviewZspecialargsnewr   r   r   	transform7   s4    

				'zFixDict.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 Sn  | s d S|  j j | j  |  o | d | k S)NFr'   func)parentp1matchr!   iter_exemptr   consuming_callsp2)r&   r'   r)   r(   r   r   r   r%   [   s    zFixDict.in_special_contextN)__name__
__module____qualname__ZBM_compatibleZPATTERNr,   ZP1r   Zcompile_patternr/   ZP2r3   r%   r   r   r   r   r   *   s   
r   N)__doc__r   r   r   Zpgen2r   r   r   r   r   r   r	   r
   r   r2   setr1   ZBaseFixr   r   r   r   r   <module>   s   .