File: //lib64/python3.9/lib2to3/__pycache__/patcomp.cpython-39.pyc
a
�DOg� � @ 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/lib64/python3.9/lib2to3/patcomp.pyr s r c c sL t jt jt jh}t�t�| �j�}|D ] }|\}}}}}||vr&|V q&dS )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� Zdd d
�Zd d � ZdS )�PatternCompilerNc C sZ |du rt j| _t j| _nt�|�| _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 sr t |�}z| jj||d�}W n4 tjyP } ztt|��d�W Y d}~n
d}~0 0 |rd| �|�|fS | �|�S dS )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 � jjkrz� fdd�|jddd� D �}t|�dkrX|d S tjdd� |D �ddd�}|�� S |j � jj krʇ fd d�|jD �}t|�dkr�|d S tj|gddd�}|�� S |j � jj
kr�� � |jdd� �}t� |�}|�� S |j � jj
k�sJ �d}|j}t|�d
k�rR|d j tjk�rR|d j}|dd� }d}t|�dk�r�|d j � jjk�r�|d }|dd � }� � ||�}|du�r�|j � jjk�s�J �|j} | d }
|
j tjk�r�d} tj} n�|
j tjk�r�d} tj} np|
j tjk�r^| d j tjk�s J �t| �d v �s.J �� �| d � } } t| �d
k�rh� �| d
�} n
d�shJ �| dk�s|| dk�r�|�� }tj|gg| | d�}|du�r�||_|�� S )zXCompiles a node, recursively.
This is one big switch on the node type.
r c s 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 r2 ��min�maxc s g | ]}� � |��qS r r- r. r0 r r r1 V r2 � ���)r8 � r: F)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 sh
"
z PatternCompiler.compile_nodec C sn t |�dksJ �|d }|jtjkrDtt�|j��}t� t
|�|�S |jtj k�r|j}|� � r�|t
vrttd| ��|dd � r�td��t� t
| �S |dkr�d }n,|�d�s�t| j|d �}|d u r�td| ��|dd � r�| �|d jd �g}nd }t�||�S nV|jdk�r | �|d �S |jd k�r\|d u �s:J �| �|d �}tj|ggddd
�S d �sjJ |��d S ) Nr r zInvalid token: %rz Can't have details for token�any�_zInvalid symbol: %r�(�[r5 F)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_int)N)FF)N)r r
r r& r, r) r? rE r r r r r & s
G
#r )rS rQ rZ ZTOKENc C s. | d � � rtjS | tjv r&tj| S d S d S )Nr )�isalphar rS r Zopmap)r r r r rR � s
rR c C s> |\}}}}|s|| j v r*tj|||d�S tj|||d�S dS )z9Converts raw node information to a Node or Leaf instance.)�contextN)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"
�