File: //lib64/python3.9/lib2to3/__pycache__/btm_matcher.cpython-39.opt-2.pyc
a
�DOg� � @ sh d Z ddlZddlZddlmZ ddlmZ ddlmZ G dd� de �Z
G d d
� d
e �Z i a d d � Z
dS )
z+George Boutsioukis <gboutsioukis@gmail.com>� N)� defaultdict� )�pytree)� reduce_treec @ s e Zd Ze�� Zdd� ZdS )�BMNodec C s" i | _ g | _ttj�| _d| _d S )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__� itertoolsr r r r r r r s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
BottomMatcherc C s0 t � | _t� | _| jg| _g | _t�d�| _d S )NZRefactoringTool) �set�matchr �rootZnodesr �logging� getLogger�loggerr r r r r s
zBottomMatcher.__init__c C sH | j �|� t|j�}|�� }| j|| jd�}|D ]}|j �|� q2d S )N��start)r �appendr Z pattern_treeZget_linear_subpattern�addr )r �fixerZtreeZlinear� match_nodesZ
match_noder r r � add_fixer% s
zBottomMatcher.add_fixerc C s� |s
|gS t |d t�r`g }|d D ]6}| j||d�}|D ] }|�| �|dd � |�� q:q$|S |d |jvr�t� }||j|d < n|j|d }|dd � r�| j|dd � |d�}n|g}|S d S )Nr r r )�
isinstance�tupler! �extendr r )r �patternr r# � alternativeZ end_nodes�endZ next_noder r r r! 1 s" zBottomMatcher.addc C s� | j }tt�}|D ]�}|}|rd|_|jD ]$}t|tj�r*|jdkr*d|_ qPq*|j dkrb|j}n|j }||j
v r�|j
| }|j D ]}|| � |� q�nH| j }|j
d ur�|j
jr�q||j
v r�|j
| }|j D ]}|| � |� q�|j
}qq|S )NT�;Fr )r r �listZ was_checkedZchildrenr% r ZLeaf�value�typer r r �parent) r ZleavesZcurrent_ac_node�resultsZleafZcurrent_ast_node�childZ
node_tokenr" r r r �runS s8
�
zBottomMatcher.runc s* t d� � fdd�� � | j� t d� d S )Nz
digraph g{c sZ | j �� D ]J}| j | }td| j|jt|�t|j�f � |dkrLt|j� � |� q
d S )Nz%d -> %d [label=%s] //%sr )r �keys�printr � type_repr�strr r
)ZnodeZ subnode_keyZsubnode��
print_noder r r8 � s
�
z*BottomMatcher.print_ac.<locals>.print_node�})r4 r r r r7 r �print_ac� s
zBottomMatcher.print_acN)r r r r r$ r! r2 r: r r r r r s
"8r c C sD t s8ddlm} |j�� D ] \}}t|�tkr|t |< qt �| | �S )Nr )�python_symbols)� _type_reprsZpygramr; �__dict__�itemsr. �int�
setdefault)Ztype_numr; �name�valr r r r5 � s
r5 )�
__author__r r � collectionsr r r Z btm_utilsr �objectr r r< r5 r r r r �<module> s