o 2h|@s4dZddlmZddlmZddlmZddlZddlZddlmZddl m Z e dZ ejejejejejejejejejejejejejejejejejejej ej!ej"ej#ej$ej%ej&ej'ej(ej)fZ*Gd d d ej+Z,e,Z-d d Z.d dZ/ddZ0Gdddej1Z2ddZ3ddZ4ddZ5ddZ6dS)z%Helpers for working with python ASTs.)absolute_import)division)print_functionN)errors) formattingz+^[ ]*#.*?coding[:=][ ]*([-_.a-zA-Z0-9]+)cs eZdZdZfddZZS)_TreeNormalizerz,Replaces all op nodes with unique instances.cs"t|tr |Stt||SN) isinstance _AST_OP_NODES __class__superrvisitselfnoder O/var/www/html/chatgem/venv/lib/python3.10/site-packages/pasta/base/ast_utils.pyr *s z_TreeNormalizer.visit)__name__ __module__ __qualname____doc__r __classcell__rrrrr'srcCstt|}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 )srctreerrrr3s rcCsJ|d}t|ddD]\}}t|rtdd|||<q d|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_linesilinerrrr=s   rcs tfdd}|||jS)Ncs t|Sr)r )n accept_typesrrJs z$find_nodes_by_type..)FindNodeVisitorr results)rr-visitorrr,rfind_nodes_by_typeIs r2cs$eZdZddZfddZZS)r/cCs||_g|_dSr) _conditionr0)r conditionrrr__init__Qs zFindNodeVisitor.__init__cs*||r |j|tt||dSr)r3r0appendr r/r rrrrr Us  zFindNodeVisitor.visit)rrrr5r rrrrrr/Osr/cCs~t|tjrz|jdWStyYdSwt|tjrFt|jdkr=t|jdtjr=t |jddr=t |jdS|jrE|jdSntt|tj rmt|jdkrlt|jdtj rlt |jddrlt |jdSnMt tdrt|tj r|jr|jdS|jr|jdSn1t tdrt|tjr|jr|jdSnt td rt|tjr|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_childWithhasattrr; finalbodyr<r=handlersrrrrrF[sH             rFcCsHt|D]\}}t|tr||vr||dSqtd||f)Nz9Unable to find list containing child %r on parent node %r)r iter_fieldsr listremoverInvalidAstError)parentchild_ field_valuerrr remove_childs rTc Cst|tjrt|dt|dt|dt|d|jD]/}t||d}||kr4t|||dSt|t rNz ||| |<WdSt yMYqwqt d||f)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 %r is not a child of %r)rHrD PASTA_DICTsetrE_fieldsgetattrsetattrr rMindex ValueErrorrrO)rPr replace_withfield field_valrrr replace_childs"      racCs6t|do|jot|jdtjot|jdjtjS)Nr?r)rHr?r rExprvalueStrrKrrr has_docstrings re)7r __future__rrrrr% pasta.augmentr pasta.baserrDcompiler#AndOrEqNotEqIsIsNotInNotInLtLtEGtGtEAddSubMultDivModPowLShiftRShiftBitAndBitOrBitXorFloorDivInvertNotUAddUSubr NodeTransformerrrrrr2 NodeVisitorr/rFrTrarerrrrs0            2