o e|@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 DD:\Projects\ConvertPro\env\Lib\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)rZ 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 r1cs$eZdZddZfddZZS)r.cCs||_g|_dSr) _conditionr/)r conditionrrr__init__Qs zFindNodeVisitor.__init__cs*||r |j|tt||dSr)r2r/appendr r.r rrrrr Us  zFindNodeVisitor.visit)rrrr4r 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 rModulebody IndexErrorIflenorelsefmtgetget_last_childWithhasattrr8 finalbodyr9r:handlersrrrrrC[sH             rCcCsHt|D]\}}t|tr||vr||dSqtd||f)Nz9Unable to find list containing child %r on parent node %r)r iter_fieldsr listremoverInvalidAstError)parentchild_Z field_valuerrr remove_childs rPc 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)rErAZ PASTA_DICTsetrB_fieldsgetattrsetattrr rJindex ValueErrorrrL)rMr replace_withfieldZ field_valrrr replace_childs"      r[cCs6t|do|jot|jdtjot|jdjtjS)Nr<r)rEr<r rExprvalueStrrHrrr has_docstrings r_)7r __future__rrrrr%Z pasta.augmentrZ pasta.baserrAcompiler#AndOrEqNotEqIsIsNotInNotInLtLtEGtGtEAddSubMultDivModPowLShiftRShiftBitAndBitOrBitXorFloorDivInvertNotUAddUSubr NodeTransformerrrrrr1 NodeVisitorr.rCrPr[r_rrrrs0            2