File: //usr/local/lib/python3.10/lib2to3/__pycache__/patcomp.cpython-310.opt-1.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
sD |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 d}|j}t|�d
kr�|d j t
jkr�|d j}|dd� }d}t|�dkr�|d j � jjkr�|d }|dd � }� � ||�}|du�r|j} | d }
|
j t
jkr�d} tj} n(|
j t
jkr�d} tj} n |
j t
jkr�� �| d � } } t| �d kr�� �| d
�} | 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 � 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 � ���� )r r" ZMatcher�childrenZ Alternatives�lenr �WildcardPattern�optimizeZ AlternativeZ NegatedUnit�
compile_basicZNegatedPatternr �EQUALr ZRepeater�STARZHUGE�PLUS�LBRACE�get_int�name)
r% �nodeZalts�pZunits�patternrF �nodes�repeatr< �childr7 r8 r r1 r r) C s^
z PatternCompiler.compile_nodec C s: |d }|j tjkrtt�|j��}t�t |�|�S |j tj
kry|j}|� � rB|t vr1t
d| ��|dd � r;t
d��t�t | �S |dkrId }n|�d�s_t| j|d �}|d u r_t
d| ��|dd � rq| �|d jd �g}nd }t�||�S |jdkr�| �|d �S |jd kr�| �|d �}tj|ggddd
�S d S ) Nr zInvalid token: %rr z Can't have details for token�any�_zInvalid symbol: %r�(�[r6 )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@ � s8
�z PatternCompiler.compile_basicc C s
t |j�S �N)�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 �NUMBERZTOKENc 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 r[ rU rR r$ r, r r r r �<module> s$
�