B
    2*\Z                 @   s   d Z ddlZddlmZmZmZmZmZmZ ddl	m
Z
 ddl	mZ G dd deZd	d
 ZG dd deZejejejddZdd Zdd Zdd ZdS )z#Guido van Rossum <guido@python.org>    N   )driverliteralstokentokenizeparsegrammar)pytree)pygramc               @   s   e Zd ZdS )PatternSyntaxErrorN)__name__
__module____qualname__ r   r   :C:\ALexclude\prg\programme\Python37\Lib\lib2to3\patcomp.pyr      s   r   c       	      c   sP   t jt jt jh}tt| j}x(|D ] }|\}}}}}||kr(|V  q(W d S )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 Zd ZdddZdddZdd Zdd	d
Zdd ZdS )PatternCompilerNc             C   sZ   |d krt j| _t j| _nt|| _t | j| _t j| _	t j
| _tj| jtd| _d S )N)Zconvert)r
   Zpattern_grammarr   Zpattern_symbolssymsr   Zload_grammarZSymbolsZpython_grammarZ	pygrammarZpython_symbolspysymsZDriverpattern_convert)selfZgrammar_filer   r   r   __init__(   s    
zPatternCompiler.__init__Fc          
   C   sp   t |}y| jj||d}W n2 tjk
rN } ztt|d W d d }~X Y nX |rb| ||fS | |S d S )N)debug)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                sV  |j  jjkr|jd }|j  jjkrz fdd|jd d d D }t|dkrX|d S tjdd |D ddd}| S |j  jj	krʇ fdd|jD }t|dkr|d S tj|gddd}| S |j  jj
kr |jdd  }t|}| S d }|j}t|d	kr>|d j tjkr>|d j}|dd  }d }t|dkrx|d
 j  jjkrx|d
 }|d d
 } ||}|d k	r>|j}	|	d }
|
j tjkrd}tj}nX|
j tjkrd}tj}n>|
j tjkr |	d  }}t|	dkr |	d	 }n |dks"|dkr>| }tj|gg||d}|d k	rN||_| S )Nr   c                s   g | ]}  |qS r   )r(   ).0ch)r$   r   r   
<listcomp>O   s    z0PatternCompiler.compile_node.<locals>.<listcomp>   r   c             S   s   g | ]
}|gqS r   r   )r,   ar   r   r   r.   R   s    )minmaxc                s   g | ]}  |qS r   )r(   )r,   r-   )r$   r   r   r.   V   s          )r   r!   ZMatcherchildrenZAlternativeslenr	   WildcardPatternoptimizeZAlternativeZNegatedUnitcompile_basicZNegatedPatternr   EQUALr   ZRepeaterSTARZHUGEPLUSLBRACEget_intname)r$   nodeZaltspZunitspatternr@   nodesrepeatr6   Zchildr1   r2   r   )r$   r   r(   C   s^    

 
"

zPatternCompiler.compile_nodec             C   s@  |d }|j tjkr4tt|j}tt	||S |j tj
kr|j}| r|tkrbtd| |dd  rvtdtt| S |dkrd }n,|dst| j|d }|d krtd| |dd  r| |d jd g}nd }t||S nH|jdkr| |d S |jd	kr<| |d }tj|ggddd
S d S )Nr   zInvalid token: %rr   zCan't have details for tokenany_zInvalid symbol: %r([)r1   r2   )r   r   STRINGr'   r   Z
evalStringr   r	   ZLeafPattern_type_of_literalNAMEisupper	TOKEN_MAPr   
startswithgetattrr"   r(   r6   ZNodePatternr8   )r$   rD   rE   rA   r   r   Zcontent
subpatternr   r   r   r:      s8    
zPatternCompiler.compile_basicc             C   s
   t |jS )N)intr   )r$   rA   r   r   r   r?      s    zPatternCompiler.get_int)N)FF)N)r   r   r   r%   r+   r(   r:   r?   r   r   r   r   r    &   s
   

G
#r    )rL   rJ   NUMBERZTOKENc             C   s.   | d   rtjS | tjkr&tj|  S d S d S )Nr   )isalphar   rL   r   Zopmap)r   r   r   r   rK      s
    

rK   c             C   s>   |\}}}}|s|| j kr*tj|||dS tj|||dS d S )N)context)Znumber2symbolr	   ZNodeZLeaf)r   Zraw_node_infor   r   rU   r6   r   r   r   r#      s    r#   c             C   s   t  | S )N)r    r+   )rC   r   r   r   r+      s    r+   )
__author__r   Zpgen2r   r   r   r   r   r    r	   r
   	Exceptionr   r   objectr    rL   rJ   rS   rN   rK   r#   r+   r   r   r   r   <module>   s    
 		