3
\g;
@ s d Z ddlmZ ddlmZmZ ddlmZ ddl m
Z
dd Z dd Z d
d Z
d d
ZdWddZdd Zdd Zdd Ze e
fddZdXddZdd Zd d ZdYd d Zd!d" ZdZd#d$Zd[d%d&Zd'd( Zd)d* Z d+d, Z d-d. Z d/d0 Z d1d2d3d4d5d6d7d8d9d:h
Z d;d< Z!d=a"d>a#d?a$d@a%dAdB Z&dCdD Z'dEdF Z(dGdH Z)dIdJ Z*dKdL Z+dMdN Z,dOdP Z-ej.ej/hZ0d\dQdRZ1ej/ej.ej2hZ3dSdT Z4d]dUdVZ5dS )^z1Utility functions, node construction macros, etc. )token)LeafNode)python_symbols)patcompc C s t tj| ttjd|gS )N=)r symsZargumentr r EQUAL)keywordvalue r */usr/lib64/python3.6/lib2to3/fixer_util.py
KeywordArg s r c C s t tjdS )N()r r LPARr r r r
LParen s r c C s t tjdS )N))r r RPARr r r r
RParen s r c C sH t | ts| g} t |ts&d|_|g}ttj| ttjdddg | S )z Build an assignment statement r )prefix)
isinstancelistr r r atomr r r )targetsourcer r r
Assign s
r Nc C s t tj| |dS )zReturn a NAME leaf)r )r r NAME)namer r r r
Name$ s r c C s | t tjt |ggS )zA node tuple for obj.attr)r r trailerDot)objattrr r r
Attr( s r$ c C s t tjdS )z A comma leaf,)r r COMMAr r r r
Comma, s r' c C s t tjdS )zA period (.) leaf.)r r DOTr r r r
r! 0 s r! c C s4 t tj|j |j g}| r0|jdt tj| |S )z-A parenthesised argument list, used by Call()r )r r r clone insert_childarglist)argsZlparenZrparennoder r r
ArgList4 s r/ c C s& t tj| t|g}|dk r"||_|S )zA function callN)r r powerr/ r )Z func_namer- r r. r r r
Call; s r1 c C s t tjdS )zA newline literal
)r r NEWLINEr r r r
NewlineB s r4 c C s t tjdS )z A blank line )r r r3 r r r r
BlankLineF s r6 c C s t tj| |dS )N)r )r r NUMBER)nr r r r
NumberJ s r9 c C s" t tjttjd| ttjdgS )z A numeric or string subscript[])r r r r r LBRACERBRACE)Z
index_noder r r
SubscriptM s r> c C s t tj| |dS )z
A string leaf)r )r r STRING)stringr r r r
StringS s rA c C s d| _ d|_ d|_ ttjd}d|_ ttjd}d|_ ||||g}|rtd|_ ttjd}d|_ |jttj||g ttj| ttj |g}ttj
ttj d|ttj dgS )zuA list comprehension of the form [xp for fp in it if test].
If test is None, the "if test" part is omitted.
r5 r forinifr: r; )
r r r r appendr r Zcomp_ifZ listmakerZcomp_forr r< r= ) ZxpfpitZtestZfor_leafZin_leafZ
inner_argsZif_leafinnerr r r
ListCompW s$
rI c C sZ x|D ] }|j qW ttjdttj| ddttjdddttj|g}ttj|}|S )zO Return an import statement in the form:
from package import name_leafsfromr )r import)remover r r r r import_as_names import_from)Z package_nameZ
name_leafsZleafchildrenimpr r r
FromImporto s
rQ c C s |d j }|jtjkr"|j }nttj|j g}|d }|rNdd |D }ttjtt|d t|d ttj|d j ||d j gg | }| j |_ |S ) zfReturns an import statement and calls a method
of the module:
import module
module.name()r" afterc S s g | ] }|j qS r )r* ).0r8 r r r
s z!ImportAndCall.. r ZlparZrpar)
r* typer r, r r0 r$ r r r )r. resultsnamesr" Z
newarglistrR newr r r
ImportAndCall s
DrZ c C s t | tr | jt t gkr dS t | tot| jdkot | jd tot | jd tot | jd to| jd jdko| jd jdkS )z(Does the node represent a tuple literal?T rU r r r )r r rO r r lenr r )r. r r r
is_tuple s
r^ c C sX t | toVt| jdkoVt | jd toVt | jd toV| jd jdkoV| jd jdkS )z'Does the node represent a list literal?r rU r: r; r_ )r r r] rO r r )r. r r r
is_list s
r` c C s t tjt | t gS )N)r r r r r )r. r r r
parenthesize s ra sortedr setanyalltuplesumminmax enumeratec c s( t | |}x|r"|V t ||}q W dS )al Follow an attribute chain.
If you have a chain of objects where a.foo -> b, b.foo-> c, etc,
use this to iterate over all objects in the chain. Iteration is
terminated by getattr(x, attr) is None.
Args:
obj: the starting object
attr: the name of the chaining attribute
Yields:
Each successive object in the chain.
N)getattr)r" r# nextr r r
attr_chain s
rm zefor_stmt< 'for' any 'in' node=any ':' any* >
| comp_for< 'for' any 'in' node=any any* >
z
power<
( 'iter' | 'list' | 'tuple' | 'sorted' | 'set' | 'sum' |
'any' | 'all' | 'enumerate' | (any* trailer< '.' 'join' >) )
trailer< '(' node=any ')' >
any*
>
z`
power<
( 'sorted' | 'enumerate' )
trailer< '(' arglist ')' >
any*
>
Fc C sr t s&tjtatjtatjtada tttg}x | j tjkr| S | j } | jd }| _ttj| g}||_|S )N)rV r r| r* rn r )r. rn r| r r r
make_suite s r c C s( x"| j tjkr"| j} | stdqW | S )z Find the top level namespace.z,root found before file_input node was found.)rV r Z
file_inputrn
ValueError)r. r r r
find_root& s
r c C s t |t|| }t|S )z Returns true if name is imported from package at the
top level of the tree which node belongs to.
To cover the case of an import like 'import foo', use
None for the package and 'foo' for the name. ) find_bindingr bool)packager r. Zbindingr r r
does_tree_import/ s r c C s | j tjtjfkS )z0Returns true if the node is an import statement.)rV r import_namerN )r. r r r
is_import7 s r c C s4 dd }t |}t| ||r dS d }}xTt|jD ]F\}}||sFq4x(t|j|d D ]\}}||sZP qZW || }P q4W |dkrxDt|jD ]6\}}|jtjkr|jr|jd jtjkr|d }P qW | dkrt tj
t tj dt tj |ddg} nt
| t tj |ddg} | t g}
|j|t tj|
dS ) z\ Works like `does_tree_import` but adds an import statement
if it was not imported. c S s | j tjko | jo t| jd S )NrU )rV r simple_stmtrO r )r. r r r
is_import_stmt> s z$touch_import..is_import_stmtNrU r rK r )r )r r rj rO rV r r r r? r r r r rQ r4 r+ ) r r r. r rootZ
insert_posoffsetidxZnode2import_rO r r r
touch_import; s4
r c C s x|j D ]}d}|jtjkrVt| |j d r4|S t| t|j d |}|rR|}n4|jtjtjfkrt| t|j d |}|r|}n |jtj krt| t|j d |}|r|}nXxt
|j dd D ]@\}}|jt j ko|j
dkrt| t|j |d |}|r|}qW nx|jtkr6|j d j
| kr6|}nTt|| |rJ|}n@|jtjkrft| ||}n$|jtjkrt| |j d r|}|r
|s|S t|r
|S q
W dS )
z Returns the node which binds variable name, otherwise None.
If optional argument package is supplied, only imports will
be returned.
See test cases for examples.Nr r\ r[ : rU r_ r_ )rO rV r Zfor_stmt_findr r Zif_stmtZ
while_stmtZtry_stmtrj r COLONr _def_syms_is_import_bindingr ry r )r r. r childZretr8 iZkidr r r
r i sH
r c C sX |g}xL|rR|j }|jdkr6|jtkr6|j|j q|jtjkr|j| kr|S qW d S )N )poprV _block_symsextendrO r r r )r r. Znodesr r r
r s r c C s | j tjkr| r| jd }|j tjkrvx|jD ]@}|j tjkrV|jd j|krp| S q0|j tjkr0|j|kr0| S q0W nL|j tjkr|jd }|j tjkr|j|kr| S n|j tjkr|j|kr| S n| j tj kr|rt
| jd j |krdS | jd }|rt d|rdS |j tj
kr.t ||r.| S |j tjkrf|jd }|j tjkr|j|kr| S n6|j tjkr|j|kr| S |r|j tjkr| S dS )z Will reuturn node if node will import name, or node
will import * from package. None is returned otherwise.
See test cases for examples. r r\ Nr[ asr_ )rV r r rO Zdotted_as_namesZdotted_as_namer r r rN strstripr rM Zimport_as_nameSTAR)r. r r rP r Zlastr8 r r r
r s@
r )N)NN)N)N)N)N)N)6__doc__Zpgen2r Zpytreer r Zpygramr r r5 r r r r r r r$ r' r! r/ r1 r4 r6 r9 r> rA rI rQ rZ r^ r` ra Zconsuming_callsrm rp rq rr ro rv r{ r r r r r r rx rw r r r r r r r r r r
sZ
-
*