XylotrechusZ
o
S�g� � @ sj d Z ddlZddlZddlmZ ddlmZ ddlmZ G dd� de �Z
G d d
� d
e �Z i a d d � Z
dS )
z+George Boutsioukis <[email protected]>� N)� defaultdict� )�pytree)� reduce_treec @ s e Zd Z e�� Zdd� ZdS )�BMNodec C s" i | _ g | _ttj�| _d| _d S )N� )�transition_table�fixers�nextr �count�id�content��self� r �:/opt/alt/python310/lib64/python3.10/lib2to3/btm_matcher.py�__init__ s
zBMNode.__init__N)�__name__�
__module__� __qualname__� itertoolsr r r r r r r s r c @ s6 e Zd Z dd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
BottomMatcherc C s0 t � | _t� | _| jg| _g | _t�d�| _d S )NZRefactoringTool) �set�matchr �rootZnodesr �loggingZ getLoggerZloggerr r r r r s
zBottomMatcher.__init__c C sJ | j �|� t|j�}|�� }| j|| jd�}|D ]}|j �|� qd 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�r1g }|d D ]}| j||d�}|D ]}|�| �|dd � |�� q q|S |d |jvrCt� }||j|d < n|j|d }|dd � r]| j|dd � |d�}|S |g}|S )Nr r r )�
isinstance�tupler �extendr r )r �patternr r! Z alternativeZ end_nodes�endZ next_noder r r r 1 s( � �zBottomMatcher.addc C s� | j }tt�}|D ]l}|}|rvd|_|jD ]}t|tj�r(|jdkr(d|_ nq|j dkr2|j}n|j }||j
v rM|j
| }|j D ] }|| � |� qBn$| j }|j
d urZ|j
jrZn ||j
v rq|j
| }|j D ] }|| � |� qg|j
}|sq
|S )NT�;Fr )r r �listZ was_checked�childrenr# r ZLeaf�value�typer r r �parent) r ZleavesZcurrent_ac_nodeZresultsZleafZcurrent_ast_node�childZ
node_tokenr r r r �runS sB
�
�
�
��"zBottomMatcher.runc s, t d� � fdd�� � | j� t d� d S )Nz
digraph g{c sZ | j �� D ]%}| j | }td| j|jt|�t|j�f � |dkr&t|j� � |� qd S )Nz%d -> %d [label=%s] //%sr )r �keys�printr � type_repr�strr r
)ZnodeZ subnode_keyZsubnode��
print_noder r r5 � s
�
�z*BottomMatcher.print_ac.<locals>.print_node�})r1 r r r r4 r �print_ac� s
zBottomMatcher.print_acN)r r r r r" r r/ r7 r r r r r s " 8r c C sD t s ddlm} |j�� D ]\}}t|�tkr|t |< q
t �| | �S )Nr )�python_symbols)� _type_reprsZpygramr8 �__dict__�itemsr, �int�
setdefault)Ztype_numr8 �name�valr r r r2 � s � r2 )�
__author__r r � collectionsr r r Z btm_utilsr �objectr r r9 r2 r r r r �<module> s