File manager

3


 \��@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
dZej�Zd
d�ZdS)�BMNodec

Cs"i|_g|_
ttj�
|_d
|_dS)N�)�transition_table�fixers�nextr�count�id�content)
�self�r�+/usr/lib64/python3.6/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|_tjd
�
|_dS)NZRefactoringTool)	�set�matchr�rootZnodesr	�loggingZ	getLoggerZlogger)
rrrrrs






zBottomMatcher.__init__cCsL|jj
|
�

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

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






zBottomMatcher.add_fixercCs�|
s
|g
St|
d
t
�rhg}xF|
d
D]:}|j||d�}x&|D]}|j|j|
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�patternrr�alternativeZ	end_nodes�endZ	next_noderrrr1s"




"



zBottomMatcher.addc	Cs0
|j}t
t�
}�
x|
D�
]}|}�
x|�
r&d
|_x,|jD]"}t|tj�r8|jdkr8d|_Pq8W|j	dkrp|j}n|j	}||j
kr�|j
|}x�|jD]"}||kr�g||<||j|�

q�Wnd|j}|j
dk	r�|j
jr�P||j
k�
r|j
|}x2|jD](}||j�k�
rg||<||j|�

q�W|j
}q$WqW|S)NT�
;Fr)rr�listZwas_checkedZchildrenr!rZLeaf�value�typerr	r�parent�keys)	rZleavesZcurrent_ac_nodeZresultsZleafZcurrent_ast_nodeZchildZ
node_tokenrrrr�runSs>

























zBottomMatcher.runc

s*td
�

�f
dd���|j
�

td�

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

|
dkrNt|j�

�|�

qWdS)Nz%d -> %d [label=%s] //%sr)rr,�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
"=rc
CsHts<d
dl
m}

x*|
jj�D]\}}t|�
tkr|t|<qWtj||�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