ó
è¾bc           @   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 d e j f d „  ƒ  YZ d S(	   s  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.
i   (   t   token(   t
   fixer_base(   t   Namet   Callt   ListCompt   in_special_context(   t   python_symbolst   FixMapc           B   s#   e  Z e Z d  Z d Z d „  Z RS(   sé  
    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] ')' >
    >
    s   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(   Ns   You should use a for loop hereu    u   listt
   map_lambdat   xpt   fpt   itt   map_nonet   argt   arglisti    t   Nonesj   cannot convert map(None, ...) with multiple arguments because map() now truncates to the shortest sequence(   t   should_skipt   parentt   typet   symst   simple_stmtt   warningt   clonet   prefixR   R   R   R   t   childrenR    t   NAMEt   valueR   R   (   t   selft   nodet   resultst   newt   args(    (    s+   /usr/lib/python2.7/lib2to3/fixes/fix_map.pyt	   transform;   s6    	
	(   t   __name__t
   __module__t   Truet   BM_compatiblet   PATTERNt   skip_onR    (    (    (    s+   /usr/lib/python2.7/lib2to3/fixes/fix_map.pyR      s   N(   t   __doc__t   pgen2R    t    R   t
   fixer_utilR   R   R   R   t   pygramR   R   t   ConditionalFixR   (    (    (    s+   /usr/lib/python2.7/lib2to3/fixes/fix_map.pyt   <module>   s
   "