
\[\                 @   su   d  Z  d d l m Z d d l m Z d d l m Z m Z m Z m	 Z	 d d l
 m Z Gd d   d e j  Z d S)	a  Fixer that changes map(F, ...) into list(map(F, ...)) unless there
exists a 'from future_builtins import map' statement in the top-level
namespace.

As a special case, map(None, X) is changed into list(X).  (This is
necessary because the semantics are changed in this case -- the new
map(None, X) is equivalent to [(x,) for x in X].)

We avoid the transformation (except for the special case mentioned
above) if the map() call is directly contained in iter(<>), list(<>),
tuple(<>), sorted(<>), ...join(<>), or for V in <>:.

NOTE: This is still not correct if the original code was depending on
map(F, X, Y, ...) to go on until the longest argument is exhausted,
substituting None for missing values -- like zip(), it now stops as
soon as the shortest argument is exhausted.
   )token)
fixer_base)NameCallListCompin_special_context)python_symbolsc               @   s.   e  Z d  Z d Z d Z d Z d d   Z d S)FixMapTa  
    map_none=power<
        'map'
        trailer< '(' arglist< 'None' ',' arg=any [','] > ')' >
    >
    |
    map_lambda=power<
        'map'
        trailer<
            '('
            arglist<
                lambdef< 'lambda'
                         (fp=NAME | vfpdef< '(' fp=NAME ')'> ) ':' xp=any
                >
                ','
                it=any
            >
            ')'
        >
    >
    |
    power<
        'map' trailer< '(' [arglist=any] ')' >
    >
    zfuture_builtins.mapc             C   s  |  j  |  r d  S| j j t j k rh |  j | d  | j   } d | _ t t	 d  | g  } n	d | k r t
 | d j   | d j   | d j    } n d | k r | d	 j   } n d
 | k r4| d
 } | j t j k r4| j d j t j k r4| j d j d k r4|  j | d  d  Sn  t |  rDd  S| j   } d | _ t t	 d  | g  } | j | _ | S)NzYou should use a for loop here listZ
map_lambdaZxpfpitZmap_noneargarglist    Nonezjcannot convert map(None, ...) with multiple arguments because map() now truncates to the shortest sequence)Zshould_skipparenttypesymsZsimple_stmtZwarningZcloneprefixr   r   r   r   Zchildrenr   NAMEvaluer   )selfZnodeZresultsnewargs r   +/usr/lib/python3.4/lib2to3/fixes/fix_map.py	transform;   s6    	
	zFixMap.transformN)__name__
__module____qualname__ZBM_compatibleZPATTERNZskip_onr   r   r   r   r   r	      s   r	   N)__doc__Zpgen2r   r
   r   Z
fixer_utilr   r   r   r   Zpygramr   r   ZConditionalFixr	   r   r   r   r   <module>   s
   "