B

�5�a8�@sfdZd
dl
mZ
d
dlmZ
d
dlmZmZmZm	Z	m
Z

d
dlmZ

d
dlmZ
Gdd�dej�Zd	S)
aFixer 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�ArgList�Call�ListComp�in_special_context)
�python_symbols)
�Nodec@s eZ
dZd
ZdZdZdd�ZdS)�FixMapTaL
    map_none=power<
        'map'
        trailer< '(' arglist< 'None' ',' arg=any [','] > ')' >
        [extra_trailers=trailer*]
    >
    |
    map_lambda=power<
        'map'
        trailer<
            '('
            arglist<
                lambdef< 'lambda'
                         (fp=NAME | vfpdef< '(' fp=NAME ')'> ) ':' xp=any
                >
                ','
                it=any
            >
            ')'
        >
        [extra_trailers=trailer*]
    >
    |
    power<
        'map' args=trailer< '(' [any] ')' >
        [extra_trailers=trailer*]
    >
    zfuture_builtins.mapcCs�
|�|
�
rdSg}d
|kr:x|d
D]}|�
|���

q$W|
jjtjkrv|�|
d�
|
��}d|_t	t
d�
|g
�}�
n&d|kr�t|d��|d��|d���}ttj
|g
|dd	�}n�d
|kr�|d��}d|_n�d|k�
rj|d}|jtjk�
rL|jd
jtjk�
rL|jd
jdjtjk�
rL|jd
jdjdk�
rL|�|
d�
dSttj
t
d�
|��g�}d|_t|
�
�
rxdSttj
t
d�
t|g
�
g|�}d|_|
j|_|S)NZextra_trailerszYou should use a for loop here��listZ
map_lambdaZxp�fp�it)
�prefixZmap_none�arg�args�
��Nonezjcannot convert map(None, ...) with multiple arguments because map() now truncates to the shortest sequence�map)Zshould_skip�appendZclone�parent�type�symsZsimple_stmtZwarningrrrrr
ZpowerZtrailerZchildrenZarglistr�NAME�valuerr)�selfZnodeZresultsZtrailers�
t�newr�r �+/usr/lib/python3.7/lib2to3/fixes/fix_map.py�	transform@sF





























zFixMap.transformN)�__name__�
__module__�__qualname__Z
BM_compatibleZPATTERNZskip_onr"r r r r!rs
rN)�__doc__Zpgen2rrrZ
fixer_utilrrrrrZpygramr	rZpytreer
ZConditionalFixrr r r r!�<module>s