U
e5dg;
@ 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 symsZargumentr r EQUAL)keywordvalue r */usr/lib64/python3.8/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 )N r prefix)
isinstancelistr r r atomr r r )targetsourcer r r
Assign s
r Nc C s t tj| |dS Nr )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| | g}| r0|dt tj| |S )Nr )r r r# clone insert_childarglist)argsZlparenZrparennoder r r
ArgList4 s r2 c C s& t tj| t|g}|d k r"||_|S r" )r r powerr2 r )Z func_namer0 r r1 r r r
Call; s r4 c C s t tjdS )N
r r NEWLINEr r r r
NewlineB s r8 c C s t tjdS )N r6 r r r r
BlankLineF s r: c C s t tj| |dS 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
rB c C s t tj| |dS r )r r STRING)stringr r r r
StringS s rE c C s d| _ d|_ d|_ ttjd}d|_ ttjd}d|_ ||||g}|rtd|_ ttjd}d|_ |ttj||g ttj| ttj |g}ttj
ttj d|ttj dgS )Nr9 r forinifr> r? )
r r r r appendr r Zcomp_ifZ listmakerZcomp_forr r@ rA ) ZxpfpitZtestZfor_leafZin_leafZ
inner_argsZif_leafinnerr r r
ListCompW s(
rM c C sV |D ] }| qttjdttj| 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
rU c C s |d }|jtjkr"| }nttj| g}|d }|rNdd |D }ttjtt|d t|d ttj|d ||d gg | }| j |_ |S ) Nr% afterc S s g | ] }| qS r )r- ).0r< r r r
s z!ImportAndCall.. r ZlparZrpar)
r- typer r/ r r3 r' r! r# r )r1 resultsnamesr% Z
newarglistrV newr r r
ImportAndCall s*
r^ c C s t | tr | jt t gkr dS t | tot| jdkot | jd tot | jd tot | jd to| jd jdko| jd jdkS )NT rY r r r )r r rS r r lenr r r1 r r r
is_tuple s
rc c C sX t | toVt| jdkoVt | jd toVt | jd toV| jd jdkoV| jd jdkS )Nr rY r> r? )r r ra rS r r rb r r r
is_list s
re c C s t tjt | t gS r" )r r r r r rb r r r
parenthesize s rf sortedr setanyalltuplesumminmax enumeratec c s$ t | |}|r |V t ||}q
d S r" )getattr)r% r& nextr r r
attr_chain s
rr 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 sp t s&ttattattada tttg}t|t| dD ]*\}}i }|||r@|d | kr@ dS q@dS )NTparentr1 F)
pats_builtr Zcompile_patternp0p1p2ziprr match)r1 Zpatternspatternrs r[ r r r
in_special_context s
r{ c C s | j }|d k r |jtjkr dS | j}|jtjtjfkr:dS |jtjkrX|j d | krXdS |jtj
ks|jtj kr|d k r|jtj ks|j d | krdS dS )NFrY T)
Z prev_siblingrZ r r, rs r funcdefclassdef expr_stmtrS Z
parametersZ
typedargslistr) )r1 prevrs r r r
is_probably_builtin s&
r c C sJ | d k rF| j tjkr>t| jdkr>| jd }|j tjkr>|jS | j} q dS )Nr` r r9 ) rZ r suitera rS r INDENTr rs )r1 indentr r r
find_indentation s
r c C s> | j tjkr| S | } | jd }| _ttj| g}||_|S r" )rZ r r r- rs r )r1 rs r r r r
make_suite s r c C s$ | j tjkr | j} | s tdq | S )Nz,root found before file_input node was found.)rZ r Z
file_inputrs
ValueErrorrb r r r
find_root& s
r c C s t |t|| }t|S r" ) find_bindingr bool)packager r1 Zbindingr r r
does_tree_import/ s r c C s | j tjtjfkS r" )rZ r import_namerR rb r r r
is_import7 s r c C s. dd }t |}t| ||r d S d }}t|jD ]F\}}||sDq2t|j|d D ]\}}||sV qlqV|| } qzq2|dkrt|jD ]8\}}|jtjkr|jr|jd jtjkr|d } qq| d krt tj
t tj dt tj |ddg} nt
| t tj |ddg} | t g}
||t tj|
d S )Nc S s | j tjko | jo t| jd S )NrY )rZ r simple_stmtrS r rb r r r
is_import_stmt> s z$touch_import..is_import_stmtrY r rO r r )r r ro rS rZ r r r rC r r r r rU r8 r. ) r r r1 r rootZ
insert_posoffsetidxZnode2import_rS r r r
touch_import; s8
r c C s |j D ]}d }|jtjkrVt| |j d r4| S t| t|j d |}|rR|}n0|jtjtjfkrt| t|j d |}|r|} n|jtj krt| t|j d |}|r|}nTt
|j dd D ]@\}}|jt j kr|j
dkrt| t|j |d |}|r|}qnx|jtkr2|j d j
| kr2|}nTt|| |rF|}n@|jtjkrbt| ||}n$|jtjkrt| |j d r|}|r|s| S t|r| S qd S )Nr rd r` r_ : rY )rS rZ r Zfor_stmt_findr r Zif_stmtZ
while_stmtZtry_stmtro r COLONr _def_syms_is_import_bindingr r~ r )r r1 r childZretr< iZkidr r r
r i sH
r c C sT |g}|rP| }|jdkr4|jtkr4||j q|jtjkr|j| kr|S qd S )N )poprZ _block_symsextendrS r r r )r r1 Znodesr r r
r s r c C s | j tjkr|s| jd }|j tjkrx|jD ]H}|j tjkrV|jd j|krt| S q,|j tjkr,|j|kr,| S q,nL|j tjkr|jd }|j tjkr|j|kr| S n|j tjkr|j|kr| S n| j tj kr|rt
| jd |krd S | jd }|rt d|rd S |j tj
kr0t ||r0| S |j tjkrh|jd }|j tjkr|j|kr| S n6|j tjkr|j|kr| S |r|j tjkr| S d S )Nr r` rd r_ as)rZ r r rS Zdotted_as_namesZdotted_as_namer r r rR strstripr rQ Zimport_as_nameSTAR)r1 r r rT r Zlastr< r r r
r s@
r )N)NN)N)N)N)N)N)5Zpgen2r Zpytreer r Zpygramr r r9 r r r r r r! r' r* r$ r2 r4 r8 r: r= rB rE rM rU r^ rc re rf Zconsuming_callsrr ru rv rw rt r{ r r r r r r r r} r| r r r# r r r r r r r
s^
-
*