$61 GRAYBYTE WORDPRESS FILE MANAGER $94

B

YI~d��@shdZd
dl
Z
d
dlZd
dlmZ
ddlmZ
ddlmZ
Gdd�de	�Z
Gd	d
�d
e	�Ziadd�Z
dS)
z+George Boutsioukis <[email protected]>�N)
�defaultdict�
)
�pytree)
�reduce_treec@seZ
dZe��Zd
d�ZdS)�BMNodec

Cs"i|_g|_
ttj�
|_d
|_dS)N�)�transition_table�fixers�nextr�count�id�content)
�self�r�8/opt/alt/python37/lib64/python3.7/lib2to3/btm_matcher.py�__init__s



zBMNode.__init__N)�__name__�
__module__�__qualname__�	itertoolsrrrrrrrs
rc@s4eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�ZdS)�
BottomMatcherc

Cs0t�|_
t�|_|jg
|_g|_t�d
�
|_dS)NZRefactoringTool)	�set�matchr�rootZnodesr	�loggingZ	getLoggerZlogger)
rrrrrs






zBottomMatcher.__init__cCsL|j�
|
�

t|
j�
}|��}|j||jd
�}x|D]}|j�
|
�

q4WdS)N)
�start)r	�appendrZpattern_treeZget_linear_subpattern�addr)r�fixerZtreeZlinear�match_nodesZ
match_noderrr�	add_fixer%s






zBottomMatcher.add_fixerc	Cs�|
s
|g
St|
d
t
�rhg}xF|
d
D]:}|j||d�}x&|D]}|�|�|
dd�|��

q>Wq&W|S|
d
|jkr�t�}||j|
d
<n|j|
d
}|
dd�r�|j|
dd�|d�}n|g
}|SdS)Nr
)
rr)�
isinstance�tupler�extendrr)r�patternrrZalternativeZ	end_nodes�endZ	next_noderrrr1s"




"



zBottomMatcher.addc	Cs
|j}t
t�
}x�|
D]�}|}x�|r�d
|_x,|jD]"}t|tj�r0|jdkr0d|_Pq0W|j	dkrh|j}n|j	}||j
kr�|j
|}xj|jD]}||�|�

q�WnL|j}|j
dk	r�|j
jr�P||j
kr�|j
|}x|jD]}||�|�

q�W|j
}qWqW|S)NT�
;Fr)rr�listZwas_checkedZchildrenr!rZLeaf�value�typerr	r�parent)	rZleavesZcurrent_ac_node�resultsZleafZcurrent_ast_nodeZchildZ
node_tokenrrrr�runSs6






















zBottomMatcher.runc

s*td
�

�f
dd���|j
�

td�

dS)Nz
digraph g{c
s^xX|j�
�D]J}
|j|
}td
|j|jt|
�
t|j�
f�

|
dkrNt|j�

�|�

qWdS)Nz%d -> %d [label=%s] //%sr)r�keys�printr�	type_repr�strr	r
)ZnodeZsubnode_keyZsubnode)
�
print_noderrr1�s








z*BottomMatcher.print_ac.<locals>.print_node�
})r.r)
rr)
r1r�print_ac�s


zBottomMatcher.print_acN)rrrrr rr,r3rrrrrs
"8rc
CsHts<d
dl
m}

x*|
j��D]\}}t|�
tkr|t|<qWt�||�S)Nr)
�python_symbols)�_type_reprsZpygramr4�__dict__�itemsr)�int�
setdefault)Ztype_numr4�name�valrrrr/�s


r/)�
__author__rr�collectionsrrrZ	btm_utilsr�objectrrr5r/rrrr�<module>s