U
��,a8 � @ sf d Z ddlmZ ddlmZ ddlmZmZmZm Z m
Z
ddl m Z
ddlmZ G dd� 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)�Name�ArgList�Call�ListComp�in_special_context)�python_symbols)�Nodec @ s e Zd ZdZdZdZdd� 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 C s� | � |�rd S g }d|kr6|d D ]}|�|�� � q"|jjtjkrr| �|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�rf|d }|jtjk�rH|jd
jtjk�rH|jd
jd jtjk�rH|jd
jd jdk�rH| �|d� d S t tj
t
d�|�� g�}d|_t|��rtd S t tj
t
d�t|g�g| �}d|_|j|_|S )NZextra_trailersz You 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)Z should_skip�appendZclone�parent�type�symsZ simple_stmtZwarningr r r r r
ZpowerZtrailerZchildrenZarglistr �NAME�valuer r )�selfZnodeZresultsZtrailers�t�newr � r �:/opt/alt/python38/lib64/python3.8/lib2to3/fixes/fix_map.py� transform@ sN
�
���
zFixMap.transformN)�__name__�
__module__� __qualname__Z
BM_compatibleZPATTERNZskip_onr" r r r r! r s r N)�__doc__Zpgen2r r r Z
fixer_utilr r r r r Zpygramr r Zpytreer
ZConditionalFixr r r r r! �<module> s