]. @ s d d l Td d d d Z d d Z d d d
d Z d d
Z d d Z d d d Z d d Z d d Z d d d Z d d Z
Gd d d e Z Gd d d e Z
d S) )*z execc C s t | | | t S)N)compileZ
PyCF_ONLY_AST)sourcefilenamemode r (/opt/alt/python35/lib64/python3.5/ast.pyparse s r
c sU t | t r! t | d d } t | t r9 | j } f d d | S)Nr evalc s t | t t f r | j St | t r2 | j St | t rW t t | j St | t
r| t t | j St | t r t
t | j St | t r t f d d t | j | j D St | t r | j St | t rbt | j t t f rbt | j t t t f rb | j } t | j t rZ|
S| Sn t | t rt | j t t f rt | j t t t f rt | j t t t f r | j } | j } t | j t r| | S| | St d t | d S)Nc 3 s- | ]# \ } } | | f Vq d S)Nr ).0kv)_convertr r = s z1literal_eval.._convert..zmalformed node or string: )!
isinstanceStrZBytessZNumnZTupletuplemapZeltsZListlistSetsetZDictdictzipkeysvaluesZ NameConstantvalueZUnaryOpopZUAddZUSuboperandZBinOpZAddZSubrightleft
ValueErrorrepr)noder r" r! )r r r r 1 s@ z literal_eval.._convert)r strr
Z
Expressionbody)Znode_or_stringr )r r literal_eval& s #r( TFc sG f d d t | t s= t d | j j | S)Nc s t t r f d d t D } d j j d j rY d d | D n d d | D f } r j r | | r d p d 7} | d j f d d j D 7} | d
St t r d d j f d d D St S)
Nc s( g | ] \ } } | | f q Sr r )r ab)_formatr r
b s z)dump.._format..z%s(%sz, c s s | ] } d | Vq d S)z%s=%sNr )r fieldr r r r d s z(dump.._format..c s s | ] \ } } | Vq d S)Nr )r r) r* r r r r f s c 3 s. | ]$ } d | t | f Vq d S)z%s=%sN)getattr)r r) )r+ r% r r r j s )z[%s]c 3 s | ] } | Vq d S)Nr )r x)r+ r r r n s ) r AST iter_fields __class____name__join _attributesr r$ )r% Zfieldsrv)r+ annotate_fieldsinclude_attributes)r% r r+ ` s $zdump.._formatzexpected AST, got %r)r r2 TypeErrorr4 r5 )r% r9 r: r )r+ r9 r: r dumpW s r< c C s[ xT d D]L } | | j k r | | j k r t | | r t | | t | | q W| S)Nlineno
col_offset)r= r> )r7 hasattrsetattrr/ )new_nodeZold_nodeattrr r r
copy_locationu s
rC c s& f d d | d d | S)Nc s d | j k r3 t | d s* | | _ n | j } d | j k rf t | d s] | | _ n | j } x$ t | D] } | | | qs Wd S)Nr= r> )r7 r? r= r> iter_child_nodes)r% r= r> child)_fixr r rF s z#fix_missing_locations.._fix r r )r% r )rF r fix_missing_locations s
rH rG c C sC x< t | D]. } d | j k r
t | d d | | _ q
W| S)Nr= r )walkr7 r/ r= )r% r rE r r r increment_lineno s rJ c c sE x> | j D]3 } y | t | | f VWq
t k
r< Yq
Xq
Wd S)N)_fieldsr/ AttributeError)r% r- r r r r3 s
r3 c c sl xe t | D]W \ } } t | t r0 | Vq
t | t r
x" | D] } t | t rF | VqF Wq
Wd S)N)r3 r r2 r )r% namer- itemr r r rD s
rD c C s t | t t t t f s1 t d | j j | j r t | j d t r t | j d j
t r | r d d l } | j
| j d j
j S| j d j
j Sd S)Nz%r can't have docstringsr )r ZAsyncFunctionDefZ FunctionDefZClassDefZModuler; r4 r5 r' ZExprr r inspectZcleandocr )r% ZcleanrO r r r
get_docstring s rP c c sT d d l m } | | g } x. | rO | j } | j t | | Vq" Wd S)Nr )deque) collectionsrQ popleftextendrD )r% rQ Ztodor r r rI s rI c @ s( e Z d Z d d Z d d Z d S) NodeVisitorc C s/ d | j j } t | | | j } | | S)NZvisit_)r4 r5 r/
generic_visit)selfr% methodZvisitorr r r visit s zNodeVisitor.visitc C s| xu t | D]g \ } } t | t rX xI | D]" } t | t r/ | j | q/ Wq
t | t r
| j | q
Wd S)N)r3 r r r2 rY )rW r% r- r rN r r r rV s
zNodeVisitor.generic_visitN)r5
__module__ __qualname__rY rV r r r r rU s rU c @ s e Z d Z d d Z d S)NodeTransformerc C s x t | D] \ } } t | t r g } xj | D]b } t | t r | j | } | d k rk q5 n t | t s | j | q5 | j | q5 W| | d d s
1 %