Your IP : 3.16.212.203
o
6��f� � @ 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 �6/opt/alt/python310/lib64/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)Zconvert)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 )r/ �ar r r r1 R s ��min�maxc r, r r- r. r0 r r r1 V r2 � ���)r8 � 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�patternrF �nodes�repeatr; �childr6 r7 r r0 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�(�[r5 )r< r r �STRINGr' r Z
evalStringr r Z LeafPattern�_type_of_literal�NAME�isupper� TOKEN_MAPr �
startswith�getattrr" r( r; Z NodePatternr= )r$ rJ rK rG 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$ rG r r r rE � s
zPatternCompiler.get_intrY )FF)r r
r r% r+ r( r? rE r r r r r & s
G #r )rS rQ rZ ZTOKENc C s* | d � � r tjS | tjv rtj| S d S )Nr )�isalphar rS r Zopmap)r r r r rR � s
rR 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 rY )r r+ )rI r r r r+ � s r+ )�__doc__�
__author__r Zpgen2r r r r r r � r r
� Exceptionr r �objectr rS rQ rZ rU rR r# r+ r r r r �<module> s$
�