� j f��@sudZddlmZddlmZddlmZmZmZm Z ddl mZGdd�dej �ZdS) 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�Call�ListComp�in_special_context)�python_symbolsc@s.eZdZdZdZdZdd�ZdS)�FixMapTa� map_none=power< 'map' trailer< '(' arglist< 'None' ',' arg=any [','] > ')' > > | map_lambda=power< 'map' trailer< '(' arglist< lambdef< 'lambda' (fp=NAME | vfpdef< '(' fp=NAME ')'> ) ':' xp=any > ',' it=any > ')' > > | power< 'map' trailer< '(' [arglist=any] ')' > > zfuture_builtins.mapcCs�|j|�rdS|jjtjkrh|j|d�|j�}d|_tt d�|g�}n d|kr�t |dj�|dj�|dj��}n�d|kr�|d j�}n�d |kr4|d }|jtjkr4|jdjt jkr4|jdjdkr4|j|d �dSnt|�rDdS|j�}d|_tt d�|g�}|j|_|S)NzYou should use a for loop here��listZ map_lambdaZxp�fp�itZmap_none�arg�arglist��Nonezjcannot convert map(None, ...) with multiple arguments because map() now truncates to the shortest sequence)Zshould_skip�parent�type�symsZsimple_stmtZwarningZclone�prefixrrrrZchildrenr�NAME�valuer)�selfZnodeZresults�new�args�r�:/opt/alt/python34/lib64/python3.4/lib2to3/fixes/fix_map.py� transform;s6 zFixMap.transformN)�__name__� __module__�__qualname__Z BM_compatibleZPATTERNZskip_onrrrrrr sr N)�__doc__Zpgen2rr rZ fixer_utilrrrrZpygramrrZConditionalFixr rrrr�s "