BVh&%dZddlZddlZddlZddlmZddlmZddlmZddlmZGddejZ Gd d ejZ d Z d Z d Zy)z1AST conversion templates. Adapted from Tangent. N)anno)ast_util)parser) qual_namescdeZdZdZdZfdZdZdZdZdZ dZ d Z d Z d Z d Zd ZxZS)ContextAdjusterzAdjusts the ctx field of nodes to ensure consistency. This transformer can change the ctx fields of a variable, tuple and other AST elements that allow one, based on whether the element is being read or written. c||_yN) _ctx_override)selfoverride_values Z/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/autograph/pyct/templates.py__init__zContextAdjuster.__init__'s 'Dc|j}tt||}t |dr"|j Jdj |||_|S)Nctxznode {} has ctx unset)r superrvisithasattrrformat)r nodeoriginal_override __class__s rrzContextAdjuster.visit*sX** $ -d 3DtU XX !G#:#A#A$#GG !*D KrcH|j|j|_yyr )r rr rs r_apply_overridezContextAdjuster._apply_override2s# %##%dh&rct|j|tj|_|j |}|Sr )rgastLoadr generic_visitrs rvisit_AttributezContextAdjuster.visit_Attribute6s1D   d #D KrcF|j||j|Sr rr rs r visit_TuplezContextAdjuster.visit_Tuple<!   d ##rcF|j||j|Sr r#rs r visit_ListzContextAdjuster.visit_List@r%rcF|j||j|Sr r#rs r visit_NamezContextAdjuster.visit_NameDr%rcT|j|d|_|j|Sr )rr r rs r visit_CallzContextAdjuster.visit_CallHs+D   d ##rc2d|_|j|Sr r r rs r visit_DictzContextAdjuster.visit_DictOD   d ##rc|j|tj|_|j |j |_|j |Sr )rrrr rvaluer rs rvisit_SubscriptzContextAdjuster.visit_SubscriptUsADDJJ'DJ   d ##rc2d|_|j|Sr r-rs rvisit_comprehensionz#ContextAdjuster.visit_comprehension[r/rc2d|_|j|Sr r-rs r visit_LambdazContextAdjuster.visit_Lambdaar/r)__name__ __module__ __qualname____doc__rrrr!r$r'r)r+r.r2r4r6 __classcell__)rs@rrrsD(& $$$$$ $ $ $rrc:eZdZdZdZdZdZdZdZdZ dZ y ) ReplaceTransformerzReplace AST nodes.c*||_d|_tjjtjj tjj tjjtjjdh|_ y)zCreate a new ReplaceTransformer. Args: replacements: A mapping from placeholder names to (lists of) AST nodes that these placeholders will be replaced by. Ffunction_context_nameN) replacementsin_replacementsrBasic DIRECTIVESEXTRA_LOOP_TESTORIGINSKIP_PROCESSINGStaticORIG_DEFINITIONSpreserved_annos)r r@s rrzReplaceTransformer.__init__ksb%D D  ""  "" $$ Drc|j|}tj||j}t |t j r|g}|S)zPrepares a replacement AST that's safe to swap in for a node. Args: replaced: ast.AST, the node being replaced key: Hashable, the key of the replacement AST Returns: ast.AST, the replacement AST )preserve_annos)r@r copy_cleanrI isinstancerAST)r replacedkeyrepl new_nodess r_prepare_replacementz'ReplaceTransformer._prepare_replacement}sG   S !D##D9M9MNI)TXX&+i rc\|j|j}||jur|S|Sr )rr1)r r new_values r visit_ExprzReplaceTransformer.visit_Exprs- 4::&IDJJ k rcf|j|jvr|j|S|j||j}t |t j r|S|r*t |ttfrtd|Dr|Stdj||j)Nc3PK|]}t|tj ywr )rMrkeyword).0rs r z3ReplaceTransformer.visit_keyword..s 8ajDLL) 8s$&zta keyword argument may only be replaced by another keyword or a non-empty list of keywords. Found: {} for keyword {}) argr@r rSrMrrYlisttupleall ValueErrorrr rrQs r visit_keywordz ReplaceTransformer.visit_keywords xxt(((    %%  $ $T488 4D$ % k :dT5M2 84 8 8 k  ??Ev $((@ rc |j|}|j|jvr|S|j|j}t|tj t j fstd|z|j|_|S)Nz>a function name can only be replaced by a Name node. Found: %s) r namer@rMrNameastraidrbs rvisit_FunctionDefz$ReplaceTransformer.visit_FunctionDefs~   d #D yy))) k   TYY 'D dTYY1 2  J   DI Krc|j|}|j|jvr|S|j|j}t|tj st d|z|j|_|S)Nz;An attribute can only be replaced by a Name node. Found: %s)r attrr@rMrrfrarhrbs rr!z"ReplaceTransformer.visit_Attributesr   d #D yy))) k   TYY 'D dDII &  G$ N PPDI Krc$|j|jvr|S|j||j}|s|Stt |j }|D] }t |ds|j|"t|dk(r|\}|S)Nr) rhr@rSrtyperrrlen)r rrRadjusterns rr)zReplaceTransformer.visit_Names wwd''' k))$8I  tDHH~.H  E q 9~ji rN) r7r8r9r:rrSrVrcrir!r)rrr=r=hs($"$  rr=cLt|trtj|dddSt|tj r|j St|tr|Dcgc] }t|c}St|trtd|DS|Scc}w)z'Converts from a known data type to AST.N)rhr annotation type_commentc32K|]}t|ywr )_convert_to_ast)rZes rr\z"_convert_to_ast..s/#/s) rMstrrrfrQNrgr^rwr_)rqrxs rrwrws 3 99t4 HH:==! 557N4() *1OA  **5 /Q/ // ( +s(B!c "t|tstdt|z|D]}t ||||<t j tj|z}t j|t jd}g}|D]U}t|j|}t|ttfr|j|E|j!|W|Dcgc]}t#j$|}}|Scc}w)aBReplaces placeholders in a Python template. AST Name and Tuple nodes always receive the context that inferred from the template. However, when replacing more complex nodes (that can potentially contain Name children), then the caller is responsible for setting the appropriate context. Args: template: A string representing Python code. Any symbol name can be used that appears in the template code can be used as placeholder. **replacements: A mapping from placeholder names to (lists of) AST nodes that these placeholders will be replaced by. String values are also supported as a shorthand for AST Name nodes with the respective ID. Returns: An AST node or list of AST nodes with the replacements made. If the template was a function, a list will be returned. If the template was a node, the same node will be returned. If the template was a string, an AST node will be returned (a `Module` node in the case of a multi-line string, an `Expr` node otherwise). Raises: ValueError: if the arguments are incorrect. z Expected string template, got %sF) preamble_len single_node)rMryrarnrwrSTANDARD_PREAMBLEtextwrapdedentparseSTANDARD_PREAMBLE_LENr=rr^r_extendappendrresolve)templater@k template_strnodesresultsrr[s rreplacers2 Hc " 7$x.H II 7a%l1o6LO7))HOOH,EE, ,,// % 'd l + 1 1$ 7D$u & nnT nnT  -4 4qZ   " 4' 4 . 5s,D c t|fi|}t|dk7r td|\}t|tj r |j St|tjr|Std|z)zFVariant of replace that generates expressions, instead of code blocks.rmzBsingle expression expected; for more general templates use replacezSthe template is expected to generate an expression or a name node; instead found %s)rrorarMrExprr1rf)rr@ replacementrs rreplace_as_expressionrs|1L1+ L NN %$dii ::$ " K ! ""r)r:rgrr tensorflow.python.autograph.pyctrrrrNodeTransformerrr=rwrrrrrrrs[   1537F$d**F$Rk--k\  *Z"r