a ,=ic|@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 e/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/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|||<qd|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_typesrrJz$find_nodes_by_type..)FindNodeVisitorr results)rr,visitorrr+rfind_nodes_by_typeIs r2cs$eZdZddZfddZZS)r/cCs||_g|_dSr) _conditionr0)r conditionrrr__init__QszFindNodeVisitor.__init__cs*||r|j|tt||dSr)r3r0appendr r/r rrrrr Us  zFindNodeVisitor.visit)rrrr5r rrrrrr/Osr/cCst|tjr.z |jdWSty,YdS0t|tjrt|jdkrzt|jdtjrzt |jddrzt |jdS|jr|jdSnt|tj rt|jdkrt|jdtj rt |jddrt |jdSnt tdrt|tj r|jr|jdS|jr|jdSnpt tdrHt|tjrH|jr|jdSnBt td rt|tjr|jrt|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 rModulebody IndexErrorIflenorelsefmtgetget_last_childWithhasattrr9 finalbodyr:r;handlersrrrrrD[s>            rDcCsLt|D]*\}}t|tr ||vr ||dSq td||fdS)Nz9Unable to find list containing child %r on parent node %r)r iter_fieldsr listremoverInvalidAstError)parentchild_Z field_valuerrr remove_childs rQc Cst|tjr8t|dt|dt|dt|d|jD]^}t||d}||krht|||dSt|t r>z||| |<WdSt yYq>0q>t d||fdS)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)rFrBZ PASTA_DICTsetrC_fieldsgetattrsetattrr rKindex ValueErrorrrM)rNr replace_withfieldZ field_valrrr replace_childs      r\cCs6t|do4|jo4t|jdtjo4t|jdjtjS)Nr=r)rFr=r rExprvalueStrrIrrr has_docstrings r`)7r __future__rrrrr%Z pasta.augmentrZ pasta.baserrBcompiler#AndOrEqNotEqIsIsNotInNotInLtLtEGtGtEAddSubMultDivModPowLShiftRShiftBitAndBitOrBitXorFloorDivInvertNotUAddUSubr NodeTransformerrrrrr2 NodeVisitorr/rDrQr\r`rrrrs.            2