File: //lib64/python3.9/lib2to3/__pycache__/patcomp.cpython-39.opt-2.pyc
a
    �DOg�  �                   @   s�   d Z ddlZddlmZmZmZmZmZmZ ddl	m
Z
 ddl	mZ G dd� de�Z
d	d
� ZG dd� de�Zejejejdd
�Zdd� Zdd� Zdd� ZdS )z#Guido van Rossum <guido@python.org>�    N�   )�driver�literals�token�tokenize�parse�grammar)�pytree)�pygramc                   @   s   e Zd ZdS )�PatternSyntaxErrorN)�__name__�
__module__�__qualname__� r   r   �'/usr/lib64/python3.9/lib2to3/patcomp.pyr      s   r   c           	      c   sL   t jt jt jh}t�t�| �j�}|D ] }|\}}}}}||vr&|V  q&d S �N)	r   �NEWLINE�INDENT�DEDENTr   �generate_tokens�io�StringIO�readline)	�input�skip�tokensZ	quintuple�type�value�start�endZ	line_textr   r   r   �tokenize_wrapper   s    r    c                   @   s:   e Zd Zd
dd�Zddd�Zdd� Zdd	d
�Zdd� ZdS )�PatternCompilerNc                 C   sZ   |d u rt j| _t j| _nt�|�| _t �| j�| _t j| _	t j
| _tj| jt
d�| _d S )N)�convert)r
   Zpattern_grammarr   Zpattern_symbols�symsr   Zload_grammarZSymbolsZpython_grammarZ	pygrammarZpython_symbols�pysymsZDriver�pattern_convert)�selfZgrammar_filer   r   r   �__init__(   s    
zPatternCompiler.__init__Fc              
   C   sr   t |�}z| jj||d�}W n4 tjyP } ztt|��d �W Y d }~n
d }~0 0 |rd| �|�|fS | �|�S d S )N)�debug)r    r   Zparse_tokensr   Z
ParseErrorr   �str�compile_node)r&