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 <guido@python.org>� N)�StringIOi���c C sH t s<ddlm} x*|jj� D ] \}}t|�tkr |t |< q W t j| | �S )N� )�python_symbols)� _type_reprsZpygramr �__dict__�items�type�int�
setdefault)Ztype_numr �name�val� r
�&/usr/lib64/python3.6/lib2to3/pytree.py� type_repr s r c @ s� e Zd ZdZdZf ZdZdZdd� Zdd� Z dZ
dd� Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zedd� �Zedd� �Zdd � Zd d � Zd d � Zejd%k r�d#d$� ZdS )&�BaseNFc O s
t j| �S )N)�object�__new__)�cls�args�kwdsr
r
r r 2 s z Base.__new__c C s | j |j k rtS | j|�S )N)� __class__�NotImplemented�_eq)�self�otherr
r
r �__eq__7 s z Base.__eq__c C s t �d S )N)�NotImplementedError)r r r
r
r r C s zBase._eqc C s t �d S )N)r )r r
r
r �cloneN s z
Base.clonec C s t �d S )N)r )r r
r
r �
post_orderV s zBase.post_orderc C s t �d S )N)r )r r
r
r � pre_order^ s zBase.pre_orderc C s� t |t�s|g}g }d}x:| jjD ].}|| krF|d k r@|j|� d}q"|j|� q"W | jj� || j_x|D ] }| j|_qlW d | _d S )NFT)�
isinstance�list�parent�children�extend�append�changed)r �newZ
l_children�found�ch�xr
r
r �replacef s
z Base.replacec C s. | }x"t |t�s&|jsd S |jd }qW |jS )Nr )r �Leafr# �lineno)r �noder
r
r �
get_lineno} s zBase.get_linenoc C s | j r| j j� d| _d S )NT)r" r& � was_changed)r r
r
r r& � s
z Base.changedc C sJ | j rFx>t| 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<t| j j�D ],\}}|| kr |dkr8d S | j j|d S q W d S )Nr r )r" r1 r# )r r2 r5 r
r
r � prev_sibling� s
zBase.prev_siblingc c s" x | j D ]}|j� E d H qW d S )N)r# �leaves)r r5 r
r
r r8 � s z Base.leavesc C s | j d krdS d| j j� S )Nr r )r" �depth)r r
r
r r9 � s
z
Base.depthc C s | j }|d krdS |jS )N� )r6 �prefix)r Znext_sibr
r
r �
get_suffix� s zBase.get_suffix� r c C s t | �jd�S )N�ascii)�str�encode)r r
r
r �__str__� s z Base.__str__)r= r ) �__name__�
__module__� __qualname__r r" r# r0 Z was_checkedr r �__hash__r r r r r+ r/ r& r3 �propertyr6 r7 r8 r9 r< �sys� version_inforA r
r
r
r r s.
r c @ s� e Zd Zd dd�Zdd� Zdd� Zejd kr0eZd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zee
e�Zdd� Zdd� Zdd� ZdS ) �NodeNc C sR || _ t|�| _x| jD ]
}| |_qW |d k r4|| _|rH|d d � | _nd | _d S )N)r r! r# r" r; �fixers_applied)r r r# �contextr; rJ r) r
r
r �__init__� s
z
Node.__init__c C s d| j jt| j�| jf S )Nz
%s(%s, %r))r rB r r r# )r r
r
r �__repr__� s z
Node.__repr__c C s dj tt| j��S )Nr: )�join�mapr? r# )r r
r
r � __unicode__� s zNode.__unicode__r= r c C s | j | jf|j |jfkS )N)r r# )r r r
r
r r � s zNode._eqc C s t | jdd� | jD �| jd�S )Nc S s g | ] }|j � �qS r
)r )�.0r) r
r
r �
<listcomp> s z Node.clone.<locals>.<listcomp>)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 kr<t|�dkr.|d S t|||d�S t|||d�S d S )Nr r )rK )Z
number2symbol�lenrI r, )ZgrZraw_noder r[ rK r# r
r
r �convert� s r] c @ sL e Zd ZdZdZdZdd� Zdd� Zdd� Zddd �Z dd
d �Z
d d
� Z dS )� BasePatternNc O s
t j| �S )N)r r )r r r r
r
r r � s zBasePattern.__new__c C sL t | j�| j| jg}x|r.|d d kr.|d= qW d| jjdjtt|��f S )Nr z%s(%s)z, ���r_ ) r r �contentr r rB rN rO �repr)r r r
r
r rM � s
zBasePattern.__repr__c C s | S )Nr
)r r
r
r �optimize� s zBasePattern.optimizec C sn | j d k r|j | j krdS | jd k rRd }|d k r4i }| j||�sDdS |rR|j|� |d k rj| jrj||| j< dS )NFT)r r` � _submatch�updater )r r. �results�rr
r
r �match� s
zBasePattern.matchc C s t |�dkrdS | j|d |�S )Nr Fr )r\ rg )r �nodesre r
r
r � match_seq� s zBasePattern.match_seqc c s& i }|r"| j |d |�r"d|fV d S )Nr r )rg )r rh rf r
r
r �generate_matches� s z BasePattern.generate_matches)N)N) rB rC rD r r` r r rM rb rg ri rj r
r
r
r r^ � s
r^ c @ s* e Zd Zddd�Zd dd�Zd
dd�ZdS ) � LeafPatternNc C s& |d k r|d k r|| _ || _|| _d S )N)r r` r )r r r` r r
r
r rL � s
zLeafPattern.__init__c C s t |t�sdS tj| ||�S )NF)r r, r^ rg )r r. re r
r
r rg s
zLeafPattern.matchc C s | j |jkS )N)r` r[ )r r. re r
r
r rc s
zLeafPattern._submatch)NNN)N)N)rB rC rD rL rg rc r
r
r
r rk � s
rk c @ s$ e Zd ZdZddd�Zddd�ZdS ) � NodePatternFNc C sT |d k r|d k r>t |�}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| j �dkr<t| j d �dkr<| j d d }| jdkr�| jdkr�| j d krft| jd�S |d k r�| j|jkr�|j� S | jdkr�t|t�r�|jdkr�| j|jkr�t|j | j|j | j|j |j�S | S )Nr r )r ) r` r\ rt ru rl r rb r rm )r rr r
r
r rb � s
zWildcardPattern.optimizec C s | j |g|�S )N)ri )r r. re r
r
r rg � s zWildcardPattern.matchc C sR xL| j |�D ]>\}}|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 �<module>
s$
1nNV,= =#