3
\�m � @ s� d Z ddlZddlZddlmZ dZi add� ZG dd� de�Z G d d
� d
e �Z
G d d � d e �Z d
d� Z G dd� de�Z
G dd� de
�ZG dd� de
�ZG dd� de
�ZG dd� de
�Zdd� ZdS )z#Guido van Rossum � N)�StringIOi���c C sH t st| j j�D ].\}}|| kr| j j� | j j|= d | _ |S qW d S )N)r" � enumerater# r& )r �ir. r
r
r �remove� s
z Base.removec C sZ | j d krd S xFt| j j�D ]6\}}|| kr y| j j|d S tk
rP d S X q W d S )Nr )r" r1 r# �
IndexError)r r2 �childr
r
r � next_sibling� s
zBase.next_siblingc C sP | j d krd S x s z Node.clone..)rJ )rI r r# rJ )r r
r
r r s z
Node.clonec c s( x | j D ]}|j� E d H qW | V d S )N)r# r )r r5 r
r
r r s zNode.post_orderc c s( | V x | j D ]}|j� E d H qW d S )N)r# r )r r5 r
r
r r
s zNode.pre_orderc C s | j s
dS | j d jS )Nr: r )r# r; )r r
r
r �_prefix_getter s zNode._prefix_getterc C s | j r|| j d _d S )Nr )r# r; )r r; r
r
r �_prefix_setter s zNode._prefix_setterc C s( | |_ d | j| _ || j|< | j� d S )N)r" r# r& )r r2 r5 r
r
r � set_child! s
zNode.set_childc C s | |_ | jj||� | j� d S )N)r" r# �insertr& )r r2 r5 r
r
r � insert_child+ s zNode.insert_childc C s | |_ | jj|� | j� d S )N)r" r# r% r& )r r5 r
r
r � append_child4 s zNode.append_child)NNN)r= r )rB rC rD rL rM rP rG rH rA r r r r rS rT rF r; rU rW rX r
r
r
r rI � s"
rI c @ s� e Zd ZdZdZdZddg fdd�Zdd� Zdd � Ze j
dkrBeZ d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� Zeee�ZdS )r, r: r Nc C sF |d k r|\| _ \| _| _|| _|| _|d k r4|| _ |d d � | _d S )N)�_prefixr- �columnr �valuerJ )r r r[ rK r; rJ r
r
r rL G s z
Leaf.__init__c C s d| j j| j| jf S )Nz
%s(%r, %r))r rB r r[ )r r
r
r rM Z s z
Leaf.__repr__c C s | j t| j� S )N)r; r? r[ )r r
r
r rP ` s zLeaf.__unicode__r= c C s | j | jf|j |jfkS )N)r r[ )r r r
r
r r k s zLeaf._eqc C s$ t | j| j| j| j| jff| jd�S )N)rJ )r, r r[ r; r- rZ rJ )r r
r
r r o s
z
Leaf.clonec c s
| V d S )Nr
)r r
r
r r8 u s z Leaf.leavesc c s
| V d S )Nr
)r r
r
r r x s zLeaf.post_orderc c s
| V d S )Nr
)r r
r
r r | s zLeaf.pre_orderc C s | j S )N)rY )r r
r
r rS � s zLeaf._prefix_getterc C s | j � || _d S )N)r& rY )r r; r
r
r rT � s zLeaf._prefix_setter)r= r )rB rC rD rY r- rZ rL rM rP rG rH rA r r r8 r r rS rT rF r; r
r
r
r r, > s$
r, c C sN |\}}}}|s|| j krt |�}x$t|�D ]\}}t|t�r"d| _q"W || _|| _|| _d S )NT)r! r1 r �WildcardPattern� wildcardsr r` r )r r r` r r2 �itemr
r
r rL % s
zNodePattern.__init__c C s� | j rJx>t| j|j�D ],\}}|t|j�kr|d k r>|j|� dS qW dS t| j�t|j�krbdS x*t| j|j�D ]\}}|j||�srdS qrW dS )NTF)rn rj r` r# r\ rd �ziprg )r r. re �crf �
subpatternr5 r
r
r rc B s
zNodePattern._submatch)NNN)N)rB rC rD rn rL rc r
r
r
r rl ! s
rl c @ sZ e Zd Zddedfdd�Zdd� Zddd�Zdd d
�Zd d � Zd
d� Z dd� Z
dd� Z dS )rm Nr c C s@ |d k r$t tt |��}x |D ]}q W || _|| _|| _|| _d S )N)�tuplerO r` �min�maxr )r r` rt ru r �altr
r
r rL l s
zWildcardPattern.__init__c C s� d }| j d k r\}}|t|�kr |d k rF|j|� | jrFt|�|| j< dS q W dS )NTF)rj r\ rd r r! )r rh re rq rf r
r
r ri � s
zWildcardPattern.match_seqc c s: | j d krXxJt| jdtt|�| j� �D ]*}i }| jrH|d |� || j< ||fV q(W n�| jdkrp| j|�V n�ttd�r�tj }t
� t_ z�y@x:| j |d�D ]*\}}| jr�|d |� || j< ||fV q�W W nR t k
�r x:| j
|�D ],\}}| j�r
|d |� || j< ||fV q�W Y nX W d ttd��r4|t_ X d S )Nr Z bare_name� getrefcountr )r` �rangert r\ ru r �_bare_name_matches�hasattrrG �stderrr �_recursive_matches� RuntimeError�_iterative_matches)r rh �countrf Z save_stderrr
r
r rj � s.
"
z WildcardPattern.generate_matchesc c s t |�}d| jkr di fV g }x>| jD ]4}x.t||�D ] \}}||fV |j||f� q8W q(W x�|�rg }x�|D ]�\}} ||k rr|| jkrrxn| jD ]d}x^t|||d � �D ]H\}
} |
dkr�i }|j| � |j| � ||
|fV |j||
|f� q�W q�W qrW |}qbW d S )Nr )r\ rt r` rj r% ru rd ) r rh Znodelenre rv rq rf Z new_results�c0�r0�c1�r1r
r
r r~ � s*
z"WildcardPattern._iterative_matchesc C sx d}i }d}t |�}xH| r\||k r\d}x0| jD ]&}|d j|| |�r0|d7 }d}P q0W qW |d |� || j< ||fS )Nr FTr )r\ r` rg r )r rh r rf �doneru Zleafr
r
r ry � s
z"WildcardPattern._bare_name_matchesc c s� || j krdi fV || jk r�xr| jD ]h}xbt||�D ]T\}}xJ| j||d � |d �D ].\}}i }|j|� |j|� || |fV qXW q6W q&W d S )Nr r )rt ru r` rj r| rd ) r rh r rv r� r� r� r� rf r
r
r r| s
"
z"WildcardPattern._recursive_matches)N)N) rB rC rD �HUGErL rb rg ri rj r~ ry r| r
r
r
r rm ^ s #
- rm c @ s. e Zd Zd
dd�Zdd� Zdd� Zdd � ZdS ) �NegatedPatternNc C s |d k r|| _ d S )N)r` )r r` r
r
r rL s zNegatedPattern.__init__c C s dS )NFr
)r r. r
r
r rg ) s zNegatedPattern.matchc C s t |�dkS )Nr )r\ )r rh r
r
r ri - s zNegatedPattern.match_seqc c sL | j d kr"t|�dkrHdi fV n&x| j j|�D ]
\}}d S W di fV d S )Nr )r` r\ rj )r rh rq rf r
r
r rj 1 s
z NegatedPattern.generate_matches)N)rB rC rD rL rg ri rj r
r
r
r r� s
r� c c s� | sdi fV n�| d | dd � }}xl|j |�D ]^\}}|sJ||fV q2xDt |||d � �D ].\}}i }|j|� |j|� || |fV q^W q2W d S )Nr r )rj rd ) Zpatternsrh �p�restr� r� r� r� rf r
r
r rj = s
rj )�
__author__rG �warnings�ior r� r r r r rI r, r] r^ rk rl rm r� rj r
r
r
r �
s$
1nNV,= =#