o o^|@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 4lib/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 )srcZtreerrrr3s 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)rZ src_linesilinerrrr=s   rcs tfdd}|||jS)Ncs t|Sr)r )n accept_typesrrJs z$find_nodes_by_type..)FindNodeVisitorr results)rr+Zvisitorrr*rfind_nodes_by_typeIs r/cs$eZdZddZfddZZS)r-cCs||_g|_dSr) _conditionr.)rZ conditionrrr__init__Qs zFindNodeVisitor.__init__cs*||r |j|tt||dSr)r0r.appendr r-r rrrrr Us  zFindNodeVisitor.visit)rrrr1r 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`. NrZis_elifZ is_continuedTry TryFinally TryExcept)r rZModulebody IndexErrorZIflenorelsefmtgetget_last_childZWithhasattrr5 finalbodyr6r7handlersrrrrr>[sH             r>cCsHt|D]\}}t|tr||vr||dSqtd||f)Nz9Unable to find list containing child %r on parent node %r)rZ iter_fieldsr listremoverInvalidAstError)parentchild_Z field_valuerrr remove_childs rIc 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)r?r<Z PASTA_DICTsetr=_fieldsgetattrsetattrr rCindex ValueErrorrrE)rFrZ replace_withZfieldZ field_valrrr replace_childs"      rRcCs6t|do|jot|jdtjot|jdjtjS)Nr8r)r?r8r rZExprvalueZStrrBrrr has_docstrings rT)7rZ __future__rrrrr$Z pasta.augmentrZ pasta.baserr<compiler"ZAndZOrZEqZNotEqZIsZIsNotZInZNotInZLtZLtEZGtZGtEZAddZSubZMultZDivZModZPowZLShiftZRShiftZBitAndZBitOrZBitXorZFloorDivZInvertZNotZUAddZUSubr ZNodeTransformerrrrrr/Z NodeVisitorr-r>rIrRrTrrrrs0            2