1Vh|pdZddlmZddlmZddlmZddlZddlZddlmZddl m Z ejdZ ejejej ej"ej$ej&ej(ej*ej,ej.ej0ej2ej4ej6ej8ej:ej<ej>ej@ejBejDejFejHejJejLejNejPejRfZ*Gd d ejVZ,e,Z-d Z.d Z/d Z0GddejbZ2dZ3dZ4dZ5dZ6y)z%Helpers for working with python ASTs.)absolute_import)division)print_functionN)errors) formattingz+^[ ]*#.*?coding[:=][ ]*([-_.a-zA-Z0-9]+)c"eZdZdZfdZxZS)_TreeNormalizerz,Replaces all op nodes with unique instances.cjt|tr|jStt||SN) isinstance _AST_OP_NODES __class__superr visitselfnoders D/home/dcms/DCMS/lib/python3.12/site-packages/pasta/base/ast_utils.pyrz_TreeNormalizer.visit*s,$ & ^^  $ -d 33)__name__ __module__ __qualname____doc__r __classcell__rs@rr r 's444rr cltjt|}tj ||S)zReplaces ast.parse; ensures additional properties on the parsed tree. This enforces the assumption that each node in the ast is unique. )astparsesanitize_source_tree_normalizerr)srctrees rrr3s* ?3' ($ +rc|jd}t|ddD]5\}}tj|st j dd|||<7dj |S)zStrip the 'coding' directive from python source code, if present. This is a workaround for https://bugs.python.org/issue18960. Also see PEP-0263. TNz#.*$z# (removed coding)) splitlines enumerate_CODING_PATTERNmatchresubjoin)r! src_linesilines rrr=sh nnT")9Ra=)@gaT"VVF$8$?il@  rcZtfd}|j||jS)Nct|Sr )r )n accept_typess rz$find_nodes_by_type..JsjL&Ar)FindNodeVisitorrresults)rr3visitors ` rfind_nodes_by_typer8Is$ A B' -- rc$eZdZdZfdZxZS)r5c ||_g|_yr ) _conditionr6)r conditions r__init__zFindNodeVisitor.__init__QsDODLrc|j|r|jj|tt||yr )r;r6appendrr5rrs rrzFindNodeVisitor.visitUs0 t ll$ /4&t,r)rrrr=rrrs@rr5r5Os--rr5c<t|tjr |jdSt|tj rt |jdk(rbt|jdtj r;tj|jddrt|jdS|jr|jdSt|tjr}t |jdk(rrt|jdtjrJtj|jddr&t|jdSttdrPt|tjr6|jr|jdS|jr|jdSttdr5t|tjr|jrx|jdSttd rYt|tj r?|jr|jdS|j"rt|j"dS|jdS#t$rYywxYw) aRGet the last child node of a block statement. The input must be a block statement (e.g. ast.For, ast.With, etc). Examples: 1. with first(): second() last() 2. try: first() except: second() finally: last() In both cases, the last child is the node for `last`. Nris_elif is_continuedTry TryFinally TryExcept)r rModulebody IndexErrorIflenorelsefmtgetget_last_childWithhasattrrE finalbodyrFrGhandlersrs rrPrP[s&cjj! YYr]cff DKKA*T[[^SVV"D  A * DKKN ++ {{ [[_$! DII! 499Q< B  ! n- DIIaL ))sEz$8 ~~ ^^B  {{ [[_sL!js~~&F ~~ ^^B sK Zcmm%D {{ [[_ }} DMM"- .. 23  sJ JJctj|D]-\}}t|ts||vs|j |yt j d|d|)Nz%Unable to find list containing child z on parent node )r iter_fieldsr listremoverInvalidAstError)parentchild_ field_values r remove_childr_sb/ na+t$+)=   38& B CCrct|tjrVtj|dtj|dtj|dtj|d|j D]J}t ||d}||k(rt|||yt|ts5 |||j|<ytjd|d|#t$rYswxYw)zReplace a node's child with another node while preserving formatting. Arguments: parent: (ast.AST) Parent node to replace a child of. node: (ast.AST) Child node to replace. replace_with: (ast.AST) New child node. prefixsuffixNzNode z is not a child of )rRrN PASTA_DICTsetrO_fieldsgetattrsetattrr rXindex ValueErrorrrZ)r[r replace_withfield field_vals r replace_childrms T3>>"GGL(CGGD($;<GGL(CGGD($;<~~  et,ID fe\* It $ +7 )//$'(   vNOO   s3C%% C10C1ct|dxrh|jxrZt|jdtjxr1t|jdj tj S)NrIr)rRrIr rExprvalueStrrUs r has_docstringrrsX $  2DII 2 TYYq\388 , 2 TYYq\'' 13r)7r __future__rrrrr* pasta.augmentr pasta.baserrNcompiler(AndOrEqNotEqIsIsNotInNotInLtLtEGtGtEAddSubMultDivModPowLShiftRShiftBitAndBitOrBitXorFloorDivInvertNotUAddUSubr NodeTransformerr r rrr8 NodeVisitorr5rPr_rmrrrrrs:+'% ("**MNGGSVVSVVSYY 366399FFCGGSVVSWWcggsww#''GGSWWcjj#**cjj#))SZZLL#**cggsxx  4c))4#$  -coo -/dCP43r