a XC?h��@s�dZdZddlZddlmZmZmZmZmZm Z ddl mZddl mZGdd �d e �Zd d�ZGdd �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 �N�)�driver�literals�token�tokenize�parse�grammar)�pytree)�pygramc@seZdZdS)�PatternSyntaxErrorN)�__name__� __module__�__qualname__�rr�4/opt/alt/python39/lib64/python3.9/lib2to3/patcomp.pyrsrc csLtjtjtjh}t�t�|�j�}|D] }|\}}}}}||vr&|Vq&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_textrrr�tokenize_wrappersrc@s:eZdZd dd�Zddd�Zdd�Zdd d �Zdd�ZdS)�PatternCompilerNcCsZ|durtj|_tj|_nt�|�|_t�|j�|_tj|_ tj |_tj|jt d�|_dS)z^Initializer. Takes an optional alternative filename for the pattern grammar. N)Zconvert)r Zpattern_grammarrZpattern_symbols�symsrZload_grammarZSymbolsZpython_grammarZ pygrammarZpython_symbols�pysymsZDriver�pattern_convert)�selfZgrammar_filerrr�__init__(s zPatternCompiler.__init__Fc Csrt|�}z|jj||d�}Wn4tjyP}ztt|��d�WYd}~n d}~00|rd|�|�|fS|�|�SdS)z=Compiles a pattern string to a nested pytree.*Pattern object.)�debugN)rrZparse_tokensrZ ParseErrorr�str�compile_node)r$rr&Z with_treer�root�errr�compile_pattern7s$zPatternCompiler.compile_patternc sV|j�jjkr|jd}|j�jjkrz�fdd�|jddd�D�}t|�dkrX|dStjdd�|D�ddd�}|��S|j�jj krʇfd d�|jD�}t|�dkr�|dStj|gddd�}|��S|j�jj kr��|jdd��}t�|�}|��Sd}|j}t|�d k�r>|djt jk�r>|dj}|dd�}d}t|�dk�rx|dj�jjk�rx|d}|dd�}��||�}|du�r>|j} | d} | jt jk�r�d}tj}nX| jt jk�r�d}tj}n>| jt jk�r��| d�}}t| �dk�r��| d �}n|dk�s"|dk�r>|��}tj|gg||d�}|du�rN||_|��S) zXCompiles a node, recursively. This is one big switch on the node type. rcsg|]}��|��qSr�r(��.0Zch�r$rr� O�z0PatternCompiler.compile_node..N�rcSsg|] }|g�qSrr)r.�arrrr0Rr1��min�maxcsg|]}��|��qSrr,r-r/rrr0Vr1��)rr!ZMatcher�childrenZAlternatives�lenr �WildcardPattern�optimizeZAlternativeZNegatedUnit� compile_basicZNegatedPatternr�EQUALrZRepeater�STARZHUGE�PLUS�LBRACE�get_int�name) r$�nodeZalts�pZunits�patternrD�nodes�repeatr:�childr5r6rr/rr(Cs^ " zPatternCompiler.compile_nodecCs@|d}|jtjkr4tt�|j��}t�t |�|�S|jtj kr�|j}|��r�|tvrbt d|��|dd�rvt d��t�t|�S|dkr�d}n,|�d�s�t|j|d�}|dur�t d|��|dd�r�|�|djd�g}nd}t�||�SnH|jdk�r|�|d�S|jd k�r<|�|d�}tj|ggddd �SdS)NrzInvalid token: %rrzCan't have details for token�any�_zInvalid symbol: %r�(�[r4)rr�STRINGr'rZ evalStringrr ZLeafPattern�_type_of_literal�NAME�isupper� TOKEN_MAPr� startswith�getattrr"r(r:ZNodePatternr<)r$rHrIrErrZcontent� subpatternrrrr>�s8 zPatternCompiler.compile_basiccCs t|j�S�N)�intr)r$rErrrrC�szPatternCompiler.get_int)N)FF)N)rr rr%r+r(r>rCrrrrr &s G #r )rQrO�NUMBERZTOKENcCs.|d��rtjS|tjvr&tj|SdSdS)Nr)�isalpharrQrZopmap)rrrrrP�s rPcCs>|\}}}}|s||jvr*tj|||d�Stj|||d�SdS)z9Converts raw node information to a Node or Leaf instance.)�contextN)Z number2symbolr ZNodeZLeaf)rZ raw_node_inforrr[r:rrrr#�sr#cCst��|�SrW)r r+)rGrrrr+�sr+)�__doc__� __author__rZpgen2rrrrrr�r r � Exceptionrr�objectr rQrOrYrSrPr#r+rrrr�s" �