`i dZddlZddlZ ejZn#e$r ejZYnwxYwGddeZddee e e efdZ ddddd e ej ee eef d Ze ej ee eefd ZGd d ejZdZdZdS)a Part of the astor library for Python AST manipulation. License: 3-clause BSD Copyright 2012-2015 (c) Patrick Maupin Copyright 2013-2015 (c) Berker Peksag Utilities for node (and, by extension, tree) manipulation. For a whole-tree approach, see the treewalk submodule. NceZdZdZdS) NonExistentz/This is not the class you are looking for. N)__name__ __module__ __qualname____doc__c/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/astor/node_util.pyrrsDr rc#K||dd}|a|D]}||||} | |ur| |fV|A|tt|t|z dSdS|||r |D] } | |fV dSdS)aIterates over an object: - If the object has a _fields attribute, it gets attributes in the order of this and returns name, value pairs. - Otherwise, if the object is a list instance, it returns name, value pairs for each item in the list, where the name is passed into this function (defaults to blank). - Can update an unknown set with information about attributes that do not exist in fields. _fieldsN)updatesetvars) nodenameunknownlistgetattr isinstance enumeratemissingfieldsvalues r iter_noders$WT9d + +F  " "DGD$00EG##Tk!!!   NN3tDzz??S[[8 9 9 9 9 9   D$    E+      r  xPc N d    f d  |||S)z}Dumps an AST or similar structure: - Pretty-prints with indentation - Doesn't print line/column/ctx info Nr c | z|r|dzpd}  |} | rd|zd}}nO|r||jdd}}n5 |r||jzd}}n|t|Sfd|D}|d||} | |zkrd |zS|r; | |d zkr||d d }d z|}|d ||S)N=r z%s[]()c<g|]\}}|dk ||Sctxr ).0abdumplevels r z+dump_tree..dump..Ts0CCCdaU Q5!! r z, z%sr,z, %s )rreprjoinpoplstrip)rrindentvaluesprefixsuffixoneliner-r, indentationrrlenrmaxline maxmergedspecialtypes @r r,zdump_tree..dumpHs$"s (biioo&& :dD ! ! /"T\3FFF  /(,dd4jj.A.A.ABCFFF Zg & & /!DDJJ$77FFF!T4:::. .CCCCCfCCC$fdiiooovv> 3w<<##f++ % / /'> !  5CCKK##d1g,,.:: & 4F%%%d++2244%vvuuddFF;;r )Nr r ) rrinitial_indentr:r<r=rr>rrr?r;r,s `````````@r dump_treerA=sm<<<<<<<<<<<<<<<( 4dN + ++r cZtfd|dS)zStrips an AST by removing all attributes not in _fields. Returns a set of the names of all attributes stripped. This canonicalizes two trees for comparison purposes. ct}d} ||D]\}}d} ||dz|r(| rtt|} ||D]}t||t |drGt|dd|jvr0 |j}|d||_dSdSdS)NT)rFrr()rrrdelattrhasattrrremove)rr5rleafsubnode_rmylistrrrr>stripstrippeds r rKzstrip_tree..stripks%%#)D'::: , ,JGQD E'6F? + + + +  *z$(( *d4jj//     D D$     4   & D%  $$dl++ e$$$%  & &$$r r )r) rrr>rrr?r;rKrLs ```` @@r strip_treerM_s[uuH&&&&&&&&&&$ E$OOO Or c"eZdZdZdZefdZdS)ExplicitNodeVisitorz[This expands on the ast module's NodeVisitor class to remove any implicit visits. c>d}t||jjz)Nz&No defined handler for node of type %s)AttributeError __class__r)rmsgs r abort_visitzExplicitNodeVisitor.abort_visits6S4>#::;;;r cXd|jjz}t|||}||S)z Visit a node.visit_)rRrr)selfrabortmethodvisitors r visitzExplicitNodeVisitor.visits0DN33$..wt}}r N)rrrrrTr[r r r rOrOsF <<<!,r rOc\Gddt}ttD]q}t |tkrt |tjr< tt|j |gz|_ a#t$rYmwxYwrdS)aPThis ugly little monkey-patcher adds in a helper class to all the AST node types. This helper class allows eq/ne comparisons to work, so that entire trees can be easily compared by Python's comparison machinery. Used by the anti8 functions to compare old and new ASTs. Could also be used by the test library. ceZdZdZdZdS)+allow_ast_comparison..CompareHelperct|t|kot|t|kSNr?rrWothers r __eq__z2allow_ast_comparison..CompareHelper.__eq__s/::e,JdtE{{1J Jr ct|t|kpt|t|kSr`rarbs r __ne__z2allow_ast_comparison..CompareHelper.__ne__s/::e,IT d5kk0I Ir N)rrrrdrfr r r CompareHelperr^s: K K K J J J J Jr rgN) objectrastr6r? issubclassASTtupler __bases__ TypeError)rgitems r allow_ast_comparisonrpsJJJJJJJJS   "" ::    dCG $ $  !&tDN';';}o'M!N!N     s0*B B('B(c tjj ||fg}|j}|j}t t f}t}t}t}|r|\  d}  d} d| D} | d| DkrdS| fd| Dn8#|$r0| |ur|| Y krYYdSwxYw|dS)z This is optimized to compare two AST trees for equality. It makes several assumptions that are currently true for AST trees used by rtrip, and it doesn't examine the _attributes. rcg|] }|dk| Sr'r r)xs r r.z fast_compare..s...1::!:::r cg|] }|dk| Sr'r rss r r.z fast_compare..s222AqEzzazzzr Fc3HK|]}||fVdSr`r )r)fnamegetan1n2s r zfast_compare..s<FF%DDUOOTT"e__5FFFFFFr T) rirk__getattribute__r3extendrnrQ zip_longestr?r)tree1tree2workr3r} exceptionzipltype_list_f1f2rxryrzs @@@r fast_comparers\ 7 #D EN D (C [F>)I D E E GB Gb)$$Bb)$$B/.R...B2222222u FFFFFFF2FFF F F F F   uRyyE!!ttB||$$$Rxx55   G$ 4sB++'C C C )rri itertoolsr~rQ izip_longestrhrrrrrrrkr?r;rArM NodeVisitorrOrprr r r rs[   )'KK)))(KKK)     &   TZ!;>bfR!37JTs ,,,,H#CGZdD#/"8"""""s  ##