3 \8@sfdZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddlmZGdddejZd 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)NameArgListCallListCompin_special_context)python_symbols)Nodec@s eZdZdZdZdZddZdS)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|jq$W|jjtjkrv|j|d|j}d|_t t d|g}n&d|krt |dj|dj|dj}t tj |g|dd }nd |kr|d j}d|_nd |krj|d }|jtjkrL|jd jtjkrL|jd jdjtjkrL|jd jdjdkrL|j|ddSt tj t d|jg}d|_t|rxdSt tj t dt|gg|}d|_|j|_|S)NZextra_trailerszYou should use a for loop herelistZ map_lambdaZxpfpit)prefixZmap_noneargargsNonezjcannot convert map(None, ...) with multiple arguments because map() now truncates to the shortest sequencemap)Z should_skipappendZcloneparenttypesymsZ simple_stmtZwarningrrrrr ZpowerZtrailerZchildrenZarglistrNAMEvaluerr)selfZnoderesultsZtrailerstnewrr!/usr/lib64/python3.6/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