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}nd|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|rd 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   