f8 | d Z ddlmZ ddlmZ ddlmZmZmZm Z m
Z
ddl m Z
ddlmZ G d dej Zd 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)NameArgListCallListCompin_special_context)python_symbols)Nodec e Zd ZdZdZdZd 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.mapc N | | rd S g }d|v r2|d D ])}| | *|j j t
j k rQ| |d | }d|_ t t d |g }nd|v r{t |d |d |d }t t
j
|g|z d }n_d
|v r"|d }d|_ nd |v r|d }|j t
j k r|j d
j t
j k rd|j d
j d j t" j k r9|j d
j d j dk r| |d d S t t
j
t d | g }d|_ t) | rd S t t
j
t d t+ |g g|z }d|_ |j |_ |S )Nextra_trailersz You should use a for loop here list
map_lambdaxpfpit)prefixmap_noneargargs Nonezjcannot convert map(None, ...) with multiple arguments because map() now truncates to the shortest sequencemap) should_skipappendcloneparenttypesyms simple_stmtwarningr r r r r powertrailerchildrenarglistr NAMEvaluer r )selfnoderesultstrailerstnewr s rB s & J J J J J J J J J J J J J J + + + + + + P P P P PZ
& P P P P Pr4