3
2ag;
@ s d dl mZ d dlmZmZ d dlmZ d dlm Z dd Z
dd Z d d
Z d d Z
dVddZdd Zdd Zdd Ze e fddZdWddZdd Zd d ZdXd d Zd d! ZdYd"d#ZdZd$d%Zd&d' Zd(d) Zd*d+ Z d,d- Z d.d/ Z d0d1d2d3d4d5d6d7d8d9h
Z d:d; Z da#d?a$d@dA Z%dBdC Z&dDdE Z'dFdG Z(dHdI Z)dJdK Z*dLdM Z+dNdO Z,ej-ej.hZ/d[dPdQZ0ej.ej-ej1hZ2dRdS Z3d\dTdUZ4d
S )] )token)LeafNode)python_symbols)patcompc C s t tj| ttjd|gS )N=)r symsargumentr r EQUAL)keywordvalue r
//opt/alt/python36/lib64/python3.6/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 )N r )prefix)
isinstancelistr r r atomr r r
)targetsourcer
r
r Assign s
r Nc C s t tj| |dS )N)r )r r NAME)namer r
r
r Name$ s r c C s | t tjt |ggS )N)r r trailerDot)objattrr
r
r Attr( s r% c C s t tjdS )N,)r r COMMAr
r
r
r Comma, s r( c C s t tjdS )N.)r r DOTr
r
r
r r" 0 s r" c C s4 t tj|j |j g}| r0|jdt tj| |S )Nr )r r r! clone insert_childarglist)argsZlparenZrparennoder
r
r ArgList4 s r0 c C s& t tj| t|g}|d k r"||_|S )N)r r powerr0 r )Z func_namer. r r/ r
r
r Call; s r2 c C s t tjdS )N
)r r NEWLINEr
r
r
r NewlineB s r5 c C s t tjdS )N )r r r4 r
r
r
r BlankLineF s r7 c C s t tj| |dS )N)r )r r NUMBER)nr r
r
r NumberJ s r: c C s" t tjttjd| ttjdgS )N[])r r r! r r LBRACERBRACE)Z
index_noder
r
r SubscriptM s r? c C s t tj| |dS )N)r )r r STRING)stringr r
r
r StringS s rB c C s d| _ d|_ d|_ ttjd}d|_ ttjd}d|_ ||||g}|rtd|_ ttjd}d|_ |jttj||g ttj| ttj |g}ttj
ttj d|ttj dgS )Nr6 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$
rJ c C sZ x|D ] }|j qW ttjdttj| ddttjdddttj|g}ttj|}|S )Nfromr )r import)remover r r r r import_as_names import_from)Z package_nameZ
name_leafsZleafchildrenimpr
r
r
FromImporto s
rR 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 ) Nr# afterc S s g | ] }|j qS r
)r+ ).0r9 r
r
r
s z!ImportAndCall.. r ZlparZrpar)
r+ typer r- r r1 r% r r! r )r/ resultsnamesr# Z
newarglistrS newr
r
r
ImportAndCall s
Dr[ c C s t | tr | jt t gkr dS t | tot| jdkot | jd tot | jd tot | jd to| jd jdko| jd jdkS )NT rV r r r )r r rP 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 )Nr rV r; r< r` )r r r^ rP r r )r/ r
r
r is_list s
ra c C s t tjt | t gS )N)r r r r r )r/ r
r
r parenthesize s rb sortedr setanyalltuplesumminmax enumeratec c s( t | |}x|r"|V t ||}q W d S )N)getattr)r# r$ nextr
r
r
attr_chain s
rn 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)rW r r} r+ ro r )r/ ro r} r
r
r
make_suite s r c C s( x"| j tjkr"| j} | stdqW | S )Nz,root found before file_input node was found.)rW r Z
file_inputro
ValueError)r/ r
r
r find_root& s
r c C s t |t|| }t|S )N) find_bindingr bool)packager r/ Zbindingr
r
r does_tree_import/ s r c C s | j tjtjfkS )N)rW r import_namerO )r/ r
r
r is_import7 s r c C s4 dd }t |}t| ||r d S d }}xTt|jD ]F\}}||sFq4x(t|j|d D ]\}}||sZP qZW || }P q4W |dkrxDt|jD ]6\}}|jtjkr|jr|jd jtjkr|d }P qW | d krt tj
t tj dt tj |ddg} nt
| t tj |ddg} | t g}
|j|t tj|
d S )Nc S s | j tjko | jo t| jd S )NrV )rW r simple_stmtrP r )r/ r
r
r is_import_stmt> s z$touch_import..is_import_stmtrV r rL r )r )r r rk rP rW r r r r@ r r r r rR r5 r, ) r r r/ r rootZ
insert_posoffsetidxZnode2import_rP 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 d S ) Nr r] r\ : rV r` r` )rP rW r Zfor_stmt_findr r Zif_stmtZ
while_stmtZtry_stmtrk r COLONr _def_syms_is_import_bindingr rz r )r r/ r childretr9 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 )poprW _block_symsextendrP 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 |krd S | jd }|rt d|rd S |j tj
kr.t ||r.| S |j tjkrf|jd }|j tjkr|j|kr| S n6|j tjkr|j|kr| S |r|j tjkr| S d S )Nr r] r\ asr` )rW r r rP Zdotted_as_namesZdotted_as_namer r r rO strstripr rN Zimport_as_nameSTAR)r/ r r rQ r Zlastr9 r
r
r r s@
r )N)NN)N)N)N)N)N)5Zpgen2r Zpytreer r Zpygramr r r6 r r r r r r r% r( r" r0 r2 r5 r7 r: r? rB rJ rR r[ r_ ra rb Zconsuming_callsrn rq rr rs rp rw r| r r r r r r ry rx r r r! r r r r
r
r
r sX
-
*