HEX

File: //lib/python3.10/lib2to3/fixes/__pycache__/fix_map.cpython-310.pyc
o

��bc8�@sfdZddlmZddlmZddlmZmZmZm	Z	m
Z
ddlmZ
ddlmZGdd�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 eZdZdZdZdZdd�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|vr|dD]	}|�|���q|jjtjkr8|�|d�|��}d|_t	t
d�|g�}n�d|vrZt|d��|d��|d���}ttj
|g|dd	�}nkd
|vrh|d��}d|_nKd|vr�|d}|jtjkr�|jd
jtjkr�|jd
jdjtjkr�|jd
jdjdkr�|�|d�dSttj
t
d�|��g�}d|_t|�r�dSttj
t
d�t|g�g|�}d|_|j|_|S)N�extra_trailerszYou 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�warningrrrrr
�power�trailer�children�arglistr�NAME�valuerr)�self�node�results�trailers�t�newr�r/�,/usr/lib/python3.10/lib2to3/fixes/fix_map.py�	transform@sH


�zFixMap.transformN)�__name__�
__module__�__qualname__�
BM_compatible�PATTERN�skip_onr1r/r/r/r0rs
rN)�__doc__�pgen2rr
r�
fixer_utilrrrrr�pygramr	r �pytreer
�ConditionalFixrr/r/r/r0�<module>s