o
    Š�bc8  ã                   @   sf   d Z ddlmZ ddlmZ ddlmZmZmZm	Z	m
Z
 ddlmZ ddlmZ G d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ÚArgListÚCallÚListCompÚin_special_context)Úpython_symbols)ÚNodec                   @   s    e Zd ZdZdZdZdd„ ZdS )Ú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*]
    >
    zfuture_builtins.mapc                 C   s–  |   |¡rd S g }d|v r|d D ]	}| | ¡ ¡ q|jjtjkr8|  |d¡ | ¡ }d|_t	t
dƒ|gƒ}n�d|v rZt|d  ¡ |d  ¡ |d  ¡ ƒ}ttj|g| dd	�}nkd
|v rh|d  ¡ }d|_nKd|v r­|d }|jtjkrž|jd jtjkrž|jd jd jtjkrž|jd jd jdkrž|  |d¡ d S ttjt
dƒ| ¡ gƒ}d|_t|ƒr³d S ttjt
dƒt|gƒg| ƒ}d|_|j|_|S )NÚextra_trailerszYou should use a for loop hereÚ ÚlistÚ
map_lambdaÚxpÚfpÚit)ÚprefixÚmap_noneÚargÚargsé   é    ÚNonezjcannot convert map(None, ...) with multiple arguments because map() now truncates to the shortest sequenceÚmap)Úshould_skipÚappendÚcloneÚparentÚtypeÚsymsÚsimple_stmtÚwarningr   r   r   r   r
   ÚpowerÚtrailerÚchildrenÚarglistr   ÚNAMEÚvaluer   r   )ÚselfÚnodeÚresultsÚtrailersÚtÚnewr   © r/   ú,/usr/lib/python3.10/lib2to3/fixes/fix_map.pyÚ	transform@   sH   


þzFixMap.transformN)Ú__name__Ú
__module__Ú__qualname__ÚBM_compatibleÚPATTERNÚskip_onr1   r/   r/   r/   r0   r      s
    r   N)Ú__doc__Úpgen2r   r   r   Ú
fixer_utilr   r   r   r   r   Úpygramr	   r    Úpytreer
   ÚConditionalFixr   r/   r/   r/   r0   Ú<module>   s   