✘✘ GRAYBYTE WORDPRESS FILE MANAGER ✘✘

a

�i��@shdZd
dl
Z
d
dlZd
dlmZ
ddlmZ
ddlmZ
Gdd�de	�Z
Gd	d
�d
e	�Ziadd�Z
dS)
z+George Boutsioukis <gboutsioukis@gmail.com>�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�+/usr/lib64/python3.9/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	�logging�	getLogger�loggerrrrrrs






zBottomMatcher.__init__cCsH|j�
|
�

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

q2dS)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
�r`g}|
d
D]6}|j||d�}|D]}|�|�|
dd�|��

q:q$|S|
d
|jv
r�t�}||j|
d
<n|j|
d
}|
dd�r�|j|
dd�|d�}n|g
}|SdS)Nr
rr)�
isinstance�tupler!�extendrr)r�patternrr#�alternativeZ	end_nodes�endZ	next_noderrrr!1s"







zBottomMatcher.addc	Cs�|j}t
t�
}|
D]�}|}|rd
|_|jD]$}t|tj�r*|jdkr*d|_
qPq*|j	dkrb|j}n|j	}||j
vr�|j
|}|jD]}||�|�

q�nH|j}|j
du
r�|j
jr�q||j
vr�|j
|}|jD]}||�|�

q�|j
}qq|S)NT�
;Fr)rr�listZwas_checkedZchildrenr%rZLeaf�value�typerr	r �parent)	rZleavesZcurrent_ac_node�resultsZleafZcurrent_ast_node�childZ
node_tokenr"rrr�runSs8


















�






zBottomMatcher.runc

s*td
�

�f
dd���|j
�

td�

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

|
dkrLt|j�

�|�

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




�


z*BottomMatcher.print_ac.<locals>.print_node�
})r4rrrr7r�print_ac�s


zBottomMatcher.print_acN)rrrrr$r!r2r:rrrrrs
"8rc
CsDts8d
dl
m}

|
j��D]\}}t|�
tkr|t|<qt�||�S)Nr)
�python_symbols)�_type_reprsZpygramr;�__dict__�itemsr.�int�
setdefault)Ztype_numr;�name�valrrrr5�s



r5)�
__author__rr�collectionsrrrZ	btm_utilsr�objectrrr<r5rrrr�<module>s