-Sh|TdZddlmZddlmZddlmZddlZddlZddlmZddl m Z ej dZ ejejejejejejejejejejejejejejejejejejej ej!ej"ej#ej$ej%ej&ej'ej(ej)fZ*Gd d ej+Z,e,Z-d Z.d Z/d Z0Gddej1Z2dZ3dZ4dZ5dZ6dS)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.ct|tr|Stt||SN) isinstance _AST_OP_NODES __class__superr visitselfnoders Z/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/pasta/base/ast_utils.pyrz_TreeNormalizer.visit*sC$ && ^^   $ ' ' - -d 3 33)__name__ __module__ __qualname____doc__r __classcell__rs@rr r 's>44444444444rr c|tjt|}t||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 rrr3s4 ?3'' ( ($ +rc|d}t|ddD]8\}}t|rt jdd|||<9d|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=s| nnT"")9RaR=))@@gaT""@VF$8$??il   rc`tfd}|||jS)Nc$t|Sr )r )n accept_typess rz$find_nodes_by_type..JsjL&A&Ar)FindNodeVisitorrresults)rr3visitors ` rfind_nodes_by_typer8Is4 AAAA B B' -- rc$eZdZdZfdZxZS)r5c"||_g|_dSr ) _conditionr6)r conditions r__init__zFindNodeVisitor.__init__QsDODLLLrc||r|j|tt||dSr )r;r6appendrr5rrs rrzFindNodeVisitor.visitUsQ t  l$ /4  &&t,,,,,r)rrrr=rrrs@rr5r5OsG---------rr5ct|tjr |jdS#t$rYdSwxYwt|tjrt |jdkr_t|jdtjr:tj |jddrt|jdS|jr |jdSnt|tj ryt |jdkr_t|jdtj r:tj |jddrt|jdSnttdrCt|tj r)|jr |jdS|jr |jdSnttdr/t|tjr|jr |jdSndttd rOt|tjr5|jr |jdS|jrt|jdS|jdS) 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?&cj!! Yr]  TTcf/ DKA*T[^SV"D"D  A **  DKN + ++ { [_$!!/ DI! 49Q< B B  ! n--  DIaL ) ))sE /z$88 / ~ ^B  { [_sL!!/js~&F&F/ ~ ^B  sK  /Zcm%D%D/ { [_ }/ DM"- . .. 2s ) 77ctj|D]6\}}t|tr||vr||dS7t jd|d|)Nz%Unable to find list containing child z on parent node )r iter_fieldsr listremoverInvalidAstError)parentchild_ field_values r remove_childr_s//  na+t$$ +)=)= ff3855&& B C CCrct|tjrRtj|dtj|dtj|dtj|d|jD]n}t ||d}||krt|||dSt|tr, ||| |<dS#t$rYjwxYwotj d|d|)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>""=GL(CGD($;$;<<<GL(CGD($;$;<<<~   et,,ID fe\*** ff It $ $  +7 )//$''(       vvNOOOs7C CCct|doU|joNt|jdtjo)t|jdjtjS)NrIr)rRrIr rExprvalueStrrUs r has_docstringrrsT $   2DI 2 TYq\38 , , 2 TYq\' 1 13r)7r __future__rrrrr* pasta.augmentr pasta.baserrNcompiler(AndOrEqNotEqIsIsNotInNotInLtLtEGtGtEAddSubMultDivModPowLShiftRShiftBitAndBitOrBitXorFloorDivInvertNotUAddUSubr NodeTransformerr r rrr8 NodeVisitorr5rPr_rmrrrrrs++'&&&&&%%%%%% (((((("*MNNGSVSVSY 3639FCGSVSWcgsw#'GSWcj#*cj#)SZL#*cgsx  44444c)444#?$$    - - - - -co - - -///dCCCPPP433333r