a
i @ s d dl Z d dlT d dlmZmZ d dlmZmZ d_dddd d
Zd d Z d`ddddZ
dd Z dd Z daddZ
dd Zdd Zdbd d Zd d Zd d! Zdd"d#d$Zd%d& ZG d'd( d(eZG d)d* d*eZeed+sd,d- Zd.d/ Ze eee_ e eee_ G d0d1 d1e Z d2d3 Z!G d4d5 d5ee d6Z"G d7d8 d8ee d6Z#G d9d: d:ee d6Z$G d;d< d d>ee d6Z&e"e'e(e)fe#e*fe$e+fe%e de,fe&e d?fiZ-e"e,fiZ.e,diZ/G d@dA dAe0Z1G dBdC dCe1Z2G dDdE dEe1Z3ee4dFsBdGdH Z5dIdJ Z6e e5e6e4_7G dKdL dLe8Z9G dMdN dNe:Z;G dOdP dPe:Ze j?j@d ZAG dTdU dUeZBdVZCdWZDg eCeDR ZEG dXdY dYeZFdZd[ ZGd\d] ZHeId^kreH dS )c N)*)contextmanager nullcontext)IntEnumauto execF)
type_commentsfeature_versionc C sF t }|r|tO }t|tr(|\}}|}n |d u r4d}t| ||||dS )N)_feature_version)Z
PyCF_ONLY_ASTZPyCF_TYPE_COMMENTS
isinstancetuplecompile)sourcefilenamemoder r
flagsmajorminor r /usr/lib64/python3.9/ast.pyparse! s
r c s` t | trt| dd} t | tr&| j} dd fddfdd fd d
| S ) Nevalr c S s t d| d S )Nzmalformed node or string: )
ValueErrornoder r r _raise_malformed_nodeC s z+literal_eval.._raise_malformed_nodec s, t | tr t| jtttfvr& | | jS N)r
Constanttypevalueintfloatcomplexr )r r r _convert_numE s z"literal_eval.._convert_numc sD t | tr._convert_signed_numc sZ t | tr| jS t | tr*tt | jS t | trDtt | jS t | t r^t
t | jS t | t rt | j t
r| j jdkr| j| j krg krn nt
S t | trt| jt| jkrȈ| ttt | jt | jS t | trRt | jttfrR| j}| j }t |t t frRt |t rRt | jtrJ|| S || S | S )Nset) r
r r" Tupler mapeltsListlistSetr, ZCallfuncNameidargskeywordsDictlenkeysvaluesdictzipZBinOpr' AddSubleftrightr# r$ r% )r r@ rA _convertr& r+ r r r rC Q s<
z literal_eval.._convert)r
strr Z
Expressionbody)Znode_or_stringr rB r literal_eval6 s
rF T)indentc sT d fdd t | ts.td| jj d urHt tsHd | d S )Nr c s d ur* d7 d }d }nd}d}t | trt| }g }d}}| jD ]}zt| |} W n ty d}Y qXY n0 | d u rt||dd u rd}qX| \} }
|o|
}|r|d|| f qX|| qXrh| jrh| jD ]t}zt| |} W n ty Y qY n0 | d u r z(dump.._format..)r
ASTr! _fieldsgetattrAttributeErrorappend _attributesr9 __class____name__joinr1 repr) r rQ prefixsepclsr6 Z allsimpler7 namer" simplerP annotate_fieldsinclude_attributesrG )rQ r rP z sX
&zdump.._formatzexpected AST, got %r )r )r
rT TypeErrorrZ r[ rD )r rd re rG r rc r dumpn s 0
rh c C sV dD ]L}||j v r|| j v rt||d }|d usDt||r|drt| || q| S )N)lineno
col_offset
end_linenoend_col_offsetZend_)rY rV hasattr
startswithsetattr)new_nodeZold_nodeattrr" r r r
copy_location s rr c s fdd | dddd | S )Nc s d| j v r"t| ds || _n| j}d| j v rJt| dd d u rD|| _n| j}d| j v rlt| dsf|| _n| j}d| j v rt| dd d u r|| _n| j}t| D ]} ||||| qd S )Nri rk rj rl )rY rm ri rV rk rj rl iter_child_nodes)r ri rj rk rl child_fixr r rv s$
z#fix_missing_locations.._fixrH r r r r ru r fix_missing_locations s rw rH c C sV t | D ]H}d|jv r(t|dd| |_d|jv rt|dd }d ur|| |_q| S )Nri r rk )walkrY rV ri rk )r nrt rk r r r increment_lineno s
rz c c s8 | j D ],}z|t| |fV W q ty0 Y q0 qd S r )rU rV rW )r fieldr r r iter_fields s
r| c c sL t | D ]>\}}t|tr"|V qt|tr|D ]}t|tr0|V q0qd S r )r| r
rT r1 )r ra r{ itemr r r rs s
rs c C s t | ttttfs"td| jj | jr8t | jd t s|j d u s|jd u r W d S |jd }|j d }|j}|j}W n tyR Y d S 0 t| }||kr||| || S |rt|| d | }nd}||| |d } || d | }
||d | }| d| |
|
d |S )NrH rJ r ) rk rl ri rj rW r encodedecoder insertrX r\ ) r r r ri rk rj rl r Zpaddingfirstlastr r r get_source_segmentN s*
r c c s< ddl m} || g}|r8| } |t| | V qd S )Nr )deque) collectionsr popleftextendrs )r r Ztodor r r rx s s
rx c @ s$ e Zd Zdd Zdd Zdd ZdS ) NodeVisitorc C s" d|j j }t| || j}||S )Nvisit_)rZ r[ rV
generic_visit)selfr methodvisitorr r r visit s zNodeVisitor.visitc C sT t |D ]F\}}t|tr:|D ]}t|tr | | q qt|tr| | qd S r )r| r
r1 rT r )r r r{ r" r} r r r r s
zNodeVisitor.generic_visitc C s |j }tt|}|d u r@t D ]\}}t||r$|} q@q$|d urd| }zt| |}W n typ Y n&0 dd l}| | dt
d ||S | |S )Nr r z" is deprecated; add visit_Constant ) r" _const_node_type_namesgetr! itemsr
rV rW warningswarnDeprecationWarningr ) r r r" type_namer` ra r r r r r r visit_Constant s&
zNodeVisitor.visit_ConstantN)r[
__module__ __qualname__r r r r r r r r s
r c @ s e Zd Zdd ZdS )NodeTransformerc C s t |D ]\}}t|trvg }|D ]D}t|tr\| |}|d u rFq"nt|ts\|| q"|| q"||d d < qt|tr| |}|d u rt|| qt||| q|S r ) r| r
r1 rT r r rX delattrro )r r r{ old_valueZ
new_valuesr" rp r r r r s&
z NodeTransformer.generic_visitN)r[ r r r r r r r r s $r ry c C s | j S r r" r r r r _getter s r c C s
|| _ d S r r r r" r r r _setter s r c @ s e Zd Zdd Zdd ZdS )_ABCc G s
d| _ d S )Nz3Deprecated AST node class. Use ast.Constant instead)__doc__)r` r6 r r r __init__ s z
_ABC.__init__c C sd t |tsdS | tv rXz
|j}W n ty4 Y dS 0 t |t| oVt |t| d S t| |S )NFr ) r
r _const_typesr" rW _const_types_notr r! __instancecheck__)r` instr" r r r r
s
z_ABC.__instancecheck__N)r[ r r r r r r r r r s r c O sp |D ]<}|| j vrq| j |}|t|k rt| j d|q| tv rXt|i |S tj| g|R i |S )Nz" got multiple values for argument )rU indexr9 rg r[ r r __new__)r` r6 kwargskeyposr r r _new s
r c @ s e Zd ZdZeZdS )Num)ry Nr[ r r rU r r r r r r r % s r ) metaclassc @ s e Zd ZdZeZdS )r r Nr r r r r r ) s r c @ s e Zd ZdZeZdS )Bytesr Nr r r r r r - s r c @ s e Zd ZeZdS ) NameConstantN)r[ r r r r r r r r r 1 s r c @ s e Zd ZdZdd ZdS )Ellipsisr c O s6 | t u r tdg|R i |S tj| g|R i |S )N.)r r r )r` r6 r r r r r 7 s zEllipsis.__new__N)r[ r r rU r r r r r r 4 s r .c @ s e Zd ZdS )sliceNr[ r r r r r r r R s r c @ s e Zd Zdd ZdS )Indexc K s |S r r )r` r" r r r r r W s z
Index.__new__Nr[ r r r r r r r r U s r c @ s e Zd ZdddZdS )ExtSlicer c K s t t|t fi |S r )r- r1 ZLoad)r` dimsr r r r r \ s zExtSlice.__new__N)r r r r r r r Z s r r c C s | j S r r/ r r r r _dims_getterd s r c C s
|| _ d S r r r r r r _dims_setterh s r c @ s e Zd ZdS )SuiteNr r r r r r m s r c @ s e Zd ZdS )AugLoadNr r r r r r p s r c @ s e Zd ZdS )AugStoreNr r r r r r s s r c @ s e Zd ZdS )ParamNr r r r r r v s r Z1ec @ s~ e Zd Ze Ze Ze Ze Ze Ze Z e Z
e Z e Z e Z
e Ze Ze Ze Ze Ze Ze Ze Zdd ZdS ) _Precedencec C s, z| | d W S ty& | Y S 0 d S NrH )rZ r r r r r next s z_Precedence.nextN)r[ r r r TUPLEYIELDTESTORANDNOTCMPEXPRBORBXORBANDSHIFTARITHTERMFACTORPOWERAWAITATOMr r r r r r ~ s& r )'")z"""z'''c s e Zd ZddddZdd Zdd Zd d
Zdd d
Zdd Zdd Z e
dd Z e ddddZ
e dd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Z fd&d'Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Z d6d7 Z d8d9 Z d:d; Z dd? Z!d@dA Z"dBdC Z#dDdE Z$dFdG Z%dHdI Z&dJdK Z'dLdM Z(dNdO Z)dPdQ Z*dRdS Z+dTdU Z,dVdW Z-dXdY Z.dZd[ Z/d\d] Z0d^d_ Z1d`da Z2dbdc Z3ddde Z4dfdg Z5dhdi Z6djdk Z7dldm Z8dndo Z9e:ddpdqdrZ;e:dsdtduZdzd{ Z?d|d} Z@d~d ZAdd ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOdddddZPeQjReQjSeQjSeQjSdZTdd ZUdddddddddddddd
ZVeQjWeQjWeQjXeQjXeQjXeQjXeQjYeQjYeQjZeQj[eQj\eQjXeQj]d
Z^e_dZ`dd Zaddddddddddd
Zbdd ZcdddÜZdeQjeeQjfdĜZgddƄ ZhddȄ Ziddʄ Zjdd̄ Zkdd΄ ZlddЄ Zmdd҄ ZnddԄ Zoddք Zpdd Zqddڄ Zrdd܄ Zsddބ Zt ZuS ) _UnparserF_avoid_backslashesc C s( g | _ g | _i | _i | _d| _|| _d S )Nr )_source_buffer _precedences
_type_ignores_indentr )r r r r r r s z_Unparser.__init__c C sH t |}z|t| W n ty* Y n0 |D ]}| || q0d S r )iterr
StopIteration)r ZinterfseqrO r r r
interleave s z_Unparser.interleavec s> t |dkr$||d d n fdd|| d S )NrH r ,c s
dS NrK writer r r r rS z&_Unparser.items_view..)r9 r r )r Z traverserr r r r
items_view s z_Unparser.items_viewc C s | j r| d d S )NrI )r r r r r r
maybe_newline s z_Unparser.maybe_newlinerJ c C s | | d| j | d S )Nz )r r r r r r r r fill s z_Unparser.fillc C s | j | d S r )r rX r r r r r s z_Unparser.writec C s | j | d S r )r rX r r r r
buffer_writer s z_Unparser.buffer_writerc C s d | j}| j |S NrJ )r\ r clearr r r r buffer s
z_Unparser.bufferNextrac c s> | d |r| | | jd7 _d V | jd8 _d S )N:rH )r r )r r r r r block s
z_Unparser.blockc c s | | d V | | d S r r )r startendr r r delimit s
z_Unparser.delimitc C s |r| ||S t S d S r )r r )r r
r conditionr r r
delimit_if s z_Unparser.delimit_ifc C s | dd| ||kS N())r get_precedence)r
precedencer r r r require_parens s z_Unparser.require_parensc C s | j |tjS r )r r r r r r r r r r s z_Unparser.get_precedencec G s |D ]}|| j |< qd S r )r )r r Znodesr r r r set_precedence s z_Unparser.set_precedencec C s` t |ttttfr t|jdk r$d S |jd }t |ts | | }r.| | | |jdd n | |j d S r )r _write_docstringr rE )r r Z docstringr r r "_write_docstring_and_traverse_body& s
z,_Unparser._write_docstring_and_traverse_bodyc C s* dd |j D | _| | | j d S )Nc S s i | ]}|j d |j qS )ignore)ri tag)rN r r r r
. s z*_Unparser.visit_Module..) type_ignoresr r r r r r r visit_Module- s
z_Unparser.visit_Modulec s` dd* fdd j|j W d n1 s<0 Y d |j d S )Nr r c s
dS r r r r r r r 8 rS z._Unparser.visit_FunctionType.. -> )r r r argtypesr returnsr r r r visit_FunctionType5 s "
z _Unparser.visit_FunctionTypec C s( | | tj|j | |j d S r )r r r r r" r r r r r
visit_Expr> s z_Unparser.visit_Exprc C sh | tj|F | tj|j|j | |j | d | |j W d n1 sZ0 Y d S )Nz := ) r r r r r targetr" r r r r r r visit_NamedExprC s
z_Unparser.visit_NamedExprc s( d fdd j|j d S )Nzimport c s
dS r r r r r r r L rS z(_Unparser.visit_Import..)r r r namesr r r r visit_ImportJ s
z_Unparser.visit_Importc sT d d|j |jr, |j d fdd j|j d S )Nzfrom .z import c s
dS r r r r r r r T rS z,_Unparser.visit_ImportFrom..)r r rQ moduler r r+ r r r r visit_ImportFromN s
z_Unparser.visit_ImportFromc C sP | |jD ]}| | | d q| |j | | }rL| | d S )N = )r targetsr r r" r )r r r) r r r r visit_AssignV s
z_Unparser.visit_Assignc C sB | | |j | d| j|jjj d | |j d S )Nrf z= ) r r r) r binopr' rZ r[ r" r r r r visit_AugAssign_ s z_Unparser.visit_AugAssignc C s | | dd|j o"t|jt | |j W d n1 sH0 Y | d | |j |j r| d | |j d S )Nr r : r0 )
r r rb r
r) r4 r r
annotationr" r r r r visit_AnnAssigne s *
z_Unparser.visit_AnnAssignc C s* | d |jr&| d | |j d S )Nreturnrf )r r" r r r r r r visit_Returno s
z_Unparser.visit_Returnc C s | d d S )Npassr r r r r
visit_Passu s z_Unparser.visit_Passc C s | d d S )Nbreakr; r r r r visit_Breakx s z_Unparser.visit_Breakc C s | d d S )Ncontinuer; r r r r visit_Continue{ s z_Unparser.visit_Continuec s( d fdd j|j d S )Nzdel c s
dS r r r r r r r rS z(_Unparser.visit_Delete..)r r r r1 r r r r visit_Delete~ s
z_Unparser.visit_Deletec C s6 | d | |j |jr2| d | |j d S )Nzassert rK )r r testmsgr r r r r visit_Assert s
z_Unparser.visit_Assertc s( d fdd j|j d S )Nzglobal c s
dS r r r r r r r rS z(_Unparser.visit_Global..r r r r+ r r r r visit_Global s
z_Unparser.visit_Globalc s( d fdd j|j d S )Nz nonlocal c s
dS r r r r r r r rS z*_Unparser.visit_Nonlocal..rE r r r r visit_Nonlocal s
z_Unparser.visit_Nonlocalc C sh | tj|F | d |jrF| d | tj|j | |j W d n1 sZ0 Y d S )Nawaitrf )r r r r r" r r r r r r r visit_Await s
z_Unparser.visit_Awaitc C sh | tj|F | d |jrF| d | tj|j | |j W d n1 sZ0 Y d S )Nyieldrf )r r r r r" r r r r r r r visit_Yield s
z_Unparser.visit_Yieldc C sf | tj|D | d |js(td| tj|j | |j W d n1 sX0 Y d S )Nz yield from z-Node can't be used without a value attribute.) r r r r r" r r r r r r r r visit_YieldFrom s
z_Unparser.visit_YieldFromc C sX | d |js"|jr tdd S | d | |j |jrT| d | |j d S )Nraisez*Node can't use cause without an exception.rf z from )r exccauser r r r r r r visit_Raise s
z_Unparser.visit_Raisec C s | d | | |j W d n1 s40 Y |jD ]}| | qD|jr| d | | |j W d n1 s0 Y |jr| d | | |j W d n1 s0 Y d S )Ntryelsefinally)r r r rE handlersorelse finalbody)r r exr r r visit_Try s
*
*
z_Unparser.visit_Tryc C sz | d |jr&| d | |j |jrB| d | |j | | |j W d n1 sl0 Y d S )Nexceptrf as )r r! r r ra r rE r r r r visit_ExceptHandler s
z _Unparser.visit_ExceptHandlerc C s | |jD ]}| d | | q| d|j | jdd|jpJ|jdh d}|jD ]"}|rp| d nd}| | q\|jD ]"}|r| d nd}| | qW d n1 s0 Y | |
| W d n1 s0 Y d S ) N@zclass r r )r
FrK T) r decorator_listr r ra r basesr7 r r r )r r decocommaer r r visit_ClassDef s&
*
z_Unparser.visit_ClassDefc C s | |d d S )Ndef_function_helperr r r r visit_FunctionDef s z_Unparser.visit_FunctionDefc C s | |d d S )Nz async defrd r r r r visit_AsyncFunctionDef s z _Unparser.visit_AsyncFunctionDefc C s | |jD ]}| d | | q|d |j }| | | dd | |j W d n1 sn0 Y |jr| d | |j | j |
|d | | W d n1 s0 Y d S )Nr\ rf r r r$ r ) r r] r r ra r r6 r& r r r r )r r Z fill_suffixr_ Zdef_strr r r re s
*
z_Unparser._function_helperc C s | d| d S )Nzfor _for_helperr r r r visit_For s z_Unparser.visit_Forc C s | d| d S )Nz
async for rh r r r r visit_AsyncFor s z_Unparser.visit_AsyncForc C s | | | |j | d | |j | j| |d | |j W d n1 s`0 Y |jr| d | | |j W d n1 s0 Y d S )N in r rR ) r r r) r r r r rE rU )r r r r r r ri s
*
z_Unparser._for_helperc C s | d | |j | | |j W d n1 s@0 Y |jrt|jdkrt|jd tr|jd }| d | |j | | |j W d qJ1 s0 Y qJ|jr | d | | |j W d n1 s0 Y d S )Nzif rH r zelif rR ) r r rB r rE rU r9 r
ZIfr r r r visit_If s
*$
,
z_Unparser.visit_Ifc C s | d | |j | | |j W d n1 s@0 Y |jr| d | | |j W d n1 s0 Y d S )Nzwhile rR )r r rB r rE rU r r r r visit_While s
*
z_Unparser.visit_Whilec sf d fdd j|j j |d |j W d n1 sX0 Y d S )Nzwith c s
dS r r r r r r r , rS z&_Unparser.visit_With..r r r r r r r rE r r r r
visit_With* s
z_Unparser.visit_Withc sf d fdd j|j j |d |j W d n1 sX0 Y d S )Nz async with c s
dS r r r r r r r 2 rS z+_Unparser.visit_AsyncWith..r ro r r r r visit_AsyncWith0 s
z_Unparser.visit_AsyncWith quote_typesescape_special_whitespacec s fdd}d t||}dv r6dd |D }fdd|D }|sttfdd |D d
}d d |gfS r|jfd
dd |d
d
d krƈd d d d |fS )Nc s4 s| dv r| S | dks | s0| ddS | S )Nz
\Zunicode_escapeascii) isprintabler r )r )rt r r escape_char< s
z2_Unparser._str_literal_helper..escape_charrJ rI c S s g | ]}|t v r|qS r )
_MULTI_QUOTESrN qr r r
I rS z1_Unparser._str_literal_helper..c s g | ]}| vr|qS r r rz escaped_stringr r r| J rS c 3 s | ]} d |v r|V qdS rM r rz )stringr r rR P rS z0_Unparser._str_literal_helper..r rH r c s | d d kS )Nr r r )r{ r} r r r T rS z/_Unparser._str_literal_helper..)r ru )r\ r. r] r sort)r r rs rt rx Zpossible_quotesquoter )rt r~ r r _str_literal_helper6 s
z _Unparser._str_literal_helperrs c C s4 | j ||d\}}|d }| | | | d S )Nr r )r r )r r rs
quote_typer r r _write_str_avoiding_backslashes\ s z)_Unparser._write_str_avoiding_backslashesc C s | d | jr.| || j | | j d S g }|jD ]:}t| dt|j }||| j |
| jt |t f q8g }t
}|D ]&\}}| j|||d\}}|
| qd|}|d }| | | | d S )Nr _fstring_rr rJ r )r r _fstring_JoinedStrr r r r; rV r! r[ rX r
r _ALL_QUOTESr r\ ) r r r r" methZ
new_bufferrs Z is_constantr r r r visit_JoinedStrb s*
z_Unparser.visit_JoinedStrc C s( | d | || j | | j d S )Nr )r _fstring_FormattedValuer r r r r r r visit_FormattedValue s
z _Unparser.visit_FormattedValuec C s. |j D ]"}t| dt|j }||| qd S )Nr )r; rV r! r[ )r r r r" r r r r r s
z _Unparser._fstring_JoinedStrc C s6 t |jtstd|jdddd}|| d S )Nz.Constants inside JoinedStr should be a string.{z{{}z}})r
r" rD r replace)r r r r" r r r _fstring_Constant s z_Unparser._fstring_Constantc C s |d t | dd}|tj |j ||j}|drH|d d|v rXtd|| |j dkrt
|j }|dvrtd |d
| |j r|d t | d t |j j
}||j | |d
d S )Nr Tr rf ru z5Unable to avoid backslash in f-string expression partr Zsraz Unknown f-string conversion.!r r r )r! r r r r r" r rn r
conversionchr format_specrV r[ )r r r unparserexprr r r r r r s&
z!_Unparser._fstring_FormattedValuec C s | |j d S r )r r5 r r r r
visit_Name s z_Unparser.visit_Namec C s0 | |jdkr | d | j|jtd d S )Nur )r kindr r r" ry r r r r r s
z_Unparser._write_docstringc
C sj t |ttfr<| t|dtddt dt d n*| jrXt |trX| | n| t| d S )Ninfnanr -r )
r
r$ r% r r] r _INFSTRr rD r r r r r _write_constant s z_Unparser._write_constantc C s |j }t|trL| dd | | j| W d q1 s@0 Y n4|du r`| d n |jdkrt| d | |j d S )Nr r ....r )r" r
r r r r r r )r r r" r r r r s
.
z_Unparser.visit_Constantc sJ dd* fdd j|j W d n1 s<0 Y d S )N[]c s
dS r r r r r r r rS z&_Unparser.visit_List..)r r r r/ r r r r
visit_List s z_Unparser.visit_Listc C sR | dd2 | |j |jD ]}| | q W d n1 sD0 Y d S )Nr r r r elt
generatorsr r genr r r visit_ListComp s
z_Unparser.visit_ListCompc C sR | dd2 | |j |jD ]}| | q W d n1 sD0 Y d S r r r r r r visit_GeneratorExp s
z _Unparser.visit_GeneratorExpc C sR | dd2 | |j |jD ]}| | q W d n1 sD0 Y d S )Nr r r r r r r
visit_SetComp s
z_Unparser.visit_SetCompc C sh | ddH | |j | d | |j |jD ]}| | q6W d n1 sZ0 Y d S )Nr r r5 )r r r r r" r r r r r visit_DictComp s
z_Unparser.visit_DictCompc C s |j r| d n
| d | tj|j | |j | d | jtj |j g|j
R | |j |j
D ]}| d | | qrd S )Nz async for z for rl if ) is_asyncr r r r r) r r r r ifs)r r Z if_clauser r r visit_comprehension s
z _Unparser.visit_comprehensionc C s | tj|p | tj |j|j | |j | d | |j | d | tj|j | |j W d n1 s0 Y d S )Nr z else )
r r r r r rE rB r r rU r r r r visit_IfExp s
z_Unparser.visit_IfExpc s\ |j rN dd* fdd j|j W d qX1 sB0 Y n
d d S )Nr r c s
dS r r r r r r r rS z%_Unparser.visit_Set..z{*()})r/ r r r r r r r r visit_Set s :z_Unparser.visit_Setc sj fdd fdd} dd0 fdd|t|j|j W d n1 s\0 Y d S ) Nc s" | d | d S Nr5 )r r )kvr r r write_key_value_pair s
z2_Unparser.visit_Dict..write_key_value_pairc sB | \}}|d u r4 d tj| | n
|| d S )N**)r r r r r )r} r r r r r r
write_item s
z(_Unparser.visit_Dict..write_itemr r c s
dS r r r r r r r rS z&_Unparser.visit_Dict..)r r r= r: r; )r r r r r r
visit_Dict s z_Unparser.visit_Dictc C s@ | dd | | j|j W d n1 s20 Y d S r )r r r r/ r r r r visit_Tuple s z_Unparser.visit_Tuple~not+r )ZInvertZNotr( r) )r r r r c C s | j |jjj }| j| }| ||H | | |tjurF| d | ||j
| |j
W d n1 st0 Y d S Nrf ) unopr' rZ r[ unop_precedencer r r r r r* r )r r operatoroperator_precedencer r r
visit_UnaryOp* s
z_Unparser.visit_UnaryOpr r\ /%<<>>|^&//r )
r> r? ZMultZMatMultZDivZModZLShiftZRShiftZBitOrZBitXorZBitAndZFloorDivZPow)
r r r r\ r r r r r r r r r )r c C s | j |jjj }| j| }| ||z || jv r@| }|}n |}| }| ||j |
|j | d| d | ||j |
|j W d n1 s0 Y d S r )
r3 r' rZ r[ binop_precedencer binop_rassocr r r@ r r rA )r r r r Zleft_precedenceZright_precedencer r r visit_BinOpW s
z_Unparser.visit_BinOpz==z!=z>=iszis notinznot in)
ZEqZNotEqZLtZLtEZGtZGtEZIsZIsNotZInZNotInc C s | tj|x | jtj |jg|jR | |j t|j |jD ].\}}|
d| j |j j
d | | qHW d n1 s0 Y d S r )r r r r r r@ comparatorsr r= opsr cmpopsrZ r[ )r r ora r r r
visit_Compareu s z_Unparser.visit_Compareandor)ZAndZOr)r r c s~ j |jjj }j| fdd} |6 d| dfdd||j W d n1 sp0 Y d S )Nc s" | | d S r )r r r r )r r r r increasing_level_traverse s z9_Unparser.visit_BoolOp..increasing_level_traverserf c s
S r r r )r r r r r rS z(_Unparser.visit_BoolOp..)boolopsr' rZ r[ boolop_precedencer r r; )r r r r r )r r r r visit_BoolOp s
z_Unparser.visit_BoolOpc C sZ | tj|j | |j t|jtr@t|jjtr@| d | d | |j d S )Nrf r- )
r r r r" r r
r r# r rq r r r r visit_Attribute s
z_Unparser.visit_Attributec C s | tj|j | |j | ddh d}|jD ]"}|rH| d nd}| | q4|jD ]"}|rr| d nd}| | q^W d n1 s0 Y d S )Nr r FrK T) r r r r3 r r r6 r r7 )r r r` ra r r r
visit_Call s
z_Unparser.visit_Callc C s~ dd }| tj|j | |j | dd: ||jrP| | j|jj n | |j W d n1 sp0 Y d S )Nc S s& t | to$| jo$tdd | jD S )Nc s s | ]}t |tV qd S r )r
ZStarred)rN r r r r rR rS zE_Unparser.visit_Subscript..is_simple_tuple..)r
r- r/ any)Z slice_valuer r r is_simple_tuple s
z2_Unparser.visit_Subscript..is_simple_tupler r ) r r r r" r r r r r/ )r r r r r r visit_Subscript s
z_Unparser.visit_Subscriptc C s* | d | tj|j | |j d S )Nr )r r r r r" r r r r r
visit_Starred s
z_Unparser.visit_Starredc C s | d d S )Nr r r r r r visit_Ellipsis s z_Unparser.visit_Ellipsisc C sN |j r| |j | d |jr.| |j |jrJ| d | |j d S )Nr )lowerr r upperstepr r r r visit_Slice s
z_Unparser.visit_Slicec C s, | |j |jr(| d | |j d S r )r argr6 r r r r r visit_arg s
z_Unparser.visit_argc C s d}|j |j }d gt|t|j |j }tt||dD ]^\}}|\}}|rXd}n
| d | | |r| d | | |t|j kr>| d q>|js|j r|rd}n
| d | d |jr| |jj
|jj r| d | |jj |j rLt|j |j D ]8\}}| d | | |r| d | | q|j
r|r`d}n
| d | d |j
j
|j
j r| d | |j
j d S )
NTrH FrK =z, /r r5 r ) posonlyargsr6 r9 defaults enumerater= r r vararg
kwonlyargsr r6 kw_defaultskwarg) r r r Zall_argsr r elementsadr r r visit_arguments sN
z_Unparser.visit_argumentsc C s< |j d u r| d n| |j | d | |j d S )Nr r )r r r r" r r r r
visit_keyword s
z_Unparser.visit_keywordc C sn | tj|L | d | |j | d | tj|j | |j W d n1 s`0 Y d S )Nzlambda r5 )r r r r r r6 r rE r r r r visit_Lambda s
z_Unparser.visit_Lambdac C s& | |j |jr"| d|j d S NrZ )r ra asnamer r r r visit_alias s z_Unparser.visit_aliasc C s, | |j |jr(| d | |j d S r )r context_expr
optional_varsr r r r r visit_withitem s
z_Unparser.visit_withitem)rJ )vr[ r r r r r r r r r propertyr r r r r r r r r r r r r r# r' r( r* r, r/ r2 r4 r7 r9 r< r> r@ rA rD rF rG rI rK rL rP rX r[ rb rf rg re rj rk ri rm rn rp rq r r r r r r r r r r r r r r r r r r r r r r r r r r r r r3 r r r r r r r r frozensetr r r r r r r r r r r r r r r r r r r r r
__classcell__r r r r r s
&
3r c C s t }|| S r )r r )Zast_objr r r r unparse$ s r c C s dd l } | jdd}|jd| jddddd d
|jd d d
ddd |jddddd |jddddd |jdd td d d | }|j}| }W d n1 s0 Y t||jj |j
|j d }t t
||j|jd! d S )"Nr z
python -m ast)proginfilerbr ?r z$the file to parse; defaults to stdin)r! nargsdefaulthelpz-mz--moder )r Zsingler Z func_typez(specify what kind of code must be parsed)r choicesr z--no-type-commentsT store_falsez)don't add information about type comments)r actionr z-az--include-attributes
store_truez:include attributes such as line numbers and column offsets)r r z-iz--indentrL z'indentation of nodes (number of spaces))r! r r )r )re rG )argparseArgumentParser add_argumentFileTyper#
parse_argsr readr ra r Zno_type_commentsprintrh re rG )r parserr6 r r Ztreer r r main) s0
&r __main__)r r )TF)rH )T)JsysZ_ast
contextlibr r enumr r r rF rh rr rw rz r| rs r r r r rx objectr r rm r r r r ry r r! r r r r r r r r# r$ r% rD bytesboolr r r rT r r r r- r r r modr Z expr_contextr r r r]
float_info
max_10_expr r Z_SINGLE_QUOTESry r r r r r[ r r r r s 8C#
%:<