3 \8�@sfdZddlmZddlmZddlmZmZmZm Z m Z ddl m Z 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�|j|�rdSg}d|kr:x|dD]}|j|j��q$W|jjtjkrv|j|d�|j�}d|_t t d�|g�}�n&d|kr�t |dj�|dj�|dj��}t tj |g|dd �}n�d |kr�|d j�}d|_n�d |k�rj|d }|jtjk�rL|jd jtjk�rL|jd jdjtjk�rL|jd jdjdk�rL|j|d�dSt tj t d�|j�g�}d|_t|��rxdSt tj t d�t|g�g|�}d|_|j|_|S)NZextra_trailerszYou 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_stmtZwarningrrrrr ZpowerZtrailerZchildrenZarglistr�NAME�valuerr)�selfZnodeZresultsZtrailers�t�newr�r �-/usr/lib64/python3.6/lib2to3/fixes/fix_map.py� transform@sF        zFixMap.transformN)�__name__� __module__� __qualname__Z BM_compatibleZPATTERNZskip_onr"r r r r!r sr N)�__doc__Zpgen2rr rZ fixer_utilrrrrrZpygramr rZpytreer ZConditionalFixr r r r r!�s