File: //usr/local/lib/python3.10/lib2to3/__pycache__/patcomp.cpython-310.pyc
o
�i� � @ s� d Z dZddlZddlmZmZmZmZmZm Z ddl
m Z ddl
m Z G dd � d e
�Zd
d � ZG d d
� d
e�Zejejejdd�Zdd� Zdd� Zdd� ZdS )z�Pattern compiler.
The grammar is taken from PatternGrammar.txt.
The compiler compiles a pattern to a pytree.*Pattern instance.
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/local/lib/python3.10/lib2to3/patcomp.pyr s r c c sN � t jt jt jh}t�t�| �j�}|D ]}|\}}}}}||vr$|V qdS )z6Tokenizes a string suppressing significant whitespace.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� Zd
d d
�Zd d � ZdS )�PatternCompilerNc C sZ |du r
t j| _t j| _n
t�|�| _t �| j�| _t j| _ t j
| _ tj | jt
d�| _dS )z^Initializer.
Takes an optional alternative filename for the pattern grammar.
N)�convert)r
Zpattern_grammarr Zpattern_symbols�symsr Z load_grammarZSymbolsZpython_grammarZ pygrammarZpython_symbols�pysymsZDriver�pattern_convert)�selfZ grammar_filer r r �__init__( s
zPatternCompiler.__init__Fc
C sb t |�}z
| jj||d�}W n tjy" } ztt|��d�d}~ww |r,| �|�|fS | �|�S )z=Compiles a pattern string to a nested pytree.*Pattern object.)�debugN)r r Z parse_tokensr Z
ParseErrorr �str� compile_node)r% r r' Z with_treer �root�er r r �compile_pattern7 s ��
z PatternCompiler.compile_patternc
s� |j � jjkr |jd }|j � jjkr=� fdd�|jddd� D �}t|�dkr,|d S tjdd� |D �ddd�}|�� S |j � jj kre� fd d�|jD �}t|�dkrX|d S tj|gddd�}|�� S |j � jj
kr� � |jdd� �}t� |�}|�� S |j � jj
ks�J �d}|j}t|�d
kr�|d j tjkr�|d j}|dd� }d}t|�dkr�|d j � jjkr�|d }|dd � }� � ||�}|du�rA|j � jjks�J �|j} | d }
|
j tjkr�d} tj} nA|
j tjkr�d} tj} n5|
j tjk�r'| d j tjk�sJ �t| �d v �sJ �� �| d � } } t| �d
k�r&� �| d
�} nJ �| dk�s3| dk�rA|�� }tj|gg| | d�}|du�rI||_|�� S )zXCompiles a node, recursively.
This is one big switch on the node type.
r c � g | ]}� � |��qS r �r) ��.0Zch�r% r r �
<listcomp>O � z0PatternCompiler.compile_node.<locals>.<listcomp>N� r c S s g | ]}|g�qS r r )r0 �ar r r r2 R s ��min�maxc r- r r. r/ r1 r r r2 V r3 � ���)r9 � r; )r r" ZMatcher�childrenZ Alternatives�lenr �WildcardPattern�optimizeZ AlternativeZ NegatedUnit�
compile_basicZNegatedPatternZUnitr �EQUALr ZRepeater�STARZHUGE�PLUS�LBRACE�RBRACE�get_int�name)
r% �nodeZalts�pZunits�patternrG �nodes�repeatr< �childr7 r8 r r1 r r) C sj
�
z PatternCompiler.compile_nodec C sZ t |�dksJ �|d }|jtjkr"tt�|j��}t� t
|�|�S |jtj kr�|j}|� � rJ|t
vr9td| ��|dd � rCtd��t� t
| �S |dkrQd }n|�d�sgt| j|d �}|d u rgtd| ��|dd � ry| �|d jd �g}nd }t�||�S |jdkr�| �|d �S |jd kr�|d u s�J �| �|d �}tj|ggddd
�S J |��) Nr r zInvalid token: %rz Can't have details for token�any�_zInvalid symbol: %r�(�[r6 )r= r r �STRINGr( r Z
evalStringr r Z LeafPattern�_type_of_literal�NAME�isupper� TOKEN_MAPr �
startswith�getattrr# r) r< Z NodePatternr> )r% rK rL rH r r Zcontent�
subpatternr r r r@ � s<
z PatternCompiler.compile_basicc C s |j tjksJ �t|j�S �N)r r �NUMBER�intr )r% rH r r r rF � s
zPatternCompiler.get_intrZ )FF)r r
r r& r, r) r@ rF r r r r r & s
G #r )rT rR r[ ZTOKENc C s* | d � � r tjS | tjv rtj| S d S )Nr )�isalphar rT r Zopmap)r r r r rS � s
rS c C s: |\}}}}|s
|| j v rtj|||d�S tj|||d�S )z9Converts raw node information to a Node or Leaf instance.)�context)Z
number2symbolr ZNodeZLeaf)r Z
raw_node_infor r r^ r< r r r r$ � s r$ c C s t � �| �S rZ )r r, )rJ r r r r, � s r, )�__doc__�
__author__r Zpgen2r r r r r r � r r
� Exceptionr r �objectr rT rR r[ rV rS r$ r, r r r r �<module> s$
�