3
�Pf�1 � @ s� d dl mZmZmZ d dlmZ d dlZddlmZ ddlm Z ddlm
Z
dd l
m Z dd
l m
Z
ejd �Zdd
d�Ze
e�ZdS )� )�absolute_import�division�unicode_literals)� text_typeN� )�base� )� _ihatexml)� constants)�
namespaces)�moduleFactoryFactoryz
{([^}]*)}(.*)Fc s� ���j d�j�G �fdd�dtj��G �fdd�d��� G �fdd�d���G �fdd �d ���G �fd
d �d ����fd d
�� ��fdd�}G � ������� fdd�dtj�}t� S )NZasdc s� e Zd Zd$� fdd� Zdd� Zdd� Zdd � Zeee�Zd
d � Z d d
� Z
ee
e �Z dd� Z dd� Z
ee e
�Zdd� Zdd� Zeee�Zdd� Zdd� Zdd� Zd d � Zd%d d �Zd d!� Zd"d#� ZdS )&z getETreeBuilder..ElementNc s^ || _ || _� j| j||��| _|d kr:td | j f| _n| j| j f| _d | _g | _g | _ d S )N�html)
�_name�
_namespace�Element� _getETreeTag�_elementr Z nameTuple�parent� _childNodes�_flags)�self�name� namespace)� ElementTree� �/usr/lib/python3.6/etree.py�__init__ s
z)getETreeBuilder..Element.__init__c S s |d kr|}n d||f }|S )Nz{%s}%sr )r r r Z etree_tagr r r r # s z-getETreeBuilder..Element._getETreeTagc S s || _ | j| j | j�| j_d S )N)r r r r �tag)r r r r r �_setName* s z)getETreeBuilder..Element._setNamec S s | j S )N)r )r r r r �_getName. s z)getETreeBuilder..Element._getNamec S s || _ | j| j| j �| j_d S )N)r r r r r )r r r r r �
_setNamespace3 s z.getETreeBuilder..Element._setNamespacec S s | j S )N)r )r r r r �
_getNamespace7 s z.getETreeBuilder..Element._getNamespacec S s | j jS )N)r �attrib)r r r r �_getAttributes< s z/getETreeBuilder..Element._getAttributesc S sp x"t | jjj� �D ]}| jj|= qW xF|j� D ]:\}}t|t�rVd|d |d f }n|}| jj||� q.W d S )Nz{%s}%sr r )�listr r" �keys�items�
isinstance�tuple�set)r �
attributes�key�valuer r r r �_setAttributes? s
z/getETreeBuilder..Element._setAttributesc S s | j S )N)r )r r r r �_getChildNodesM s z/getETreeBuilder..Element._getChildNodesc S s. | j d d �= g | _x|D ]}| j|� qW d S )N)r r Z insertChild)r r, �elementr r r �_setChildNodesP s
z/getETreeBuilder..Element._setChildNodesc S s t | jjpt| j��S )z,Return true if the node has children or text)�boolr �text�len)r r r r �
hasContentX s z+getETreeBuilder..Element.hasContentc S s$ | j j|� | jj|j� | |_d S )N)r �appendr r )r �noder r r � appendChild\ s z,getETreeBuilder..Element.appendChildc S s, t | j�j|j�}| jj||j� | |_d S )N)r$ r �index�insertr )r r6 ZrefNoder8 r r r � insertBeforea s z-getETreeBuilder..Element.insertBeforec S s$ | j j|� | jj|j� d |_d S )N)r �remover r )r r6 r r r � removeChildf s z,getETreeBuilder..Element.removeChildc S s� t | j�s,| jjsd| j_| j j|7 _n�|d krb| jd jsLd| jd _| jd j|7 _nxt| j�}|j|j�}|dkr�| j|d js�d| j|d _| j|d j|7 _n | jjs�d| j_| j j|7 _d S )N� r r �����r> r> )r3 r r2 �tailr$ r8 )r �datar: Zchildrenr8 r r r �
insertTextk s"
z+getETreeBuilder..Element.insertTextc S s8 t | �| j| j�}x | jj� D ]\}}||j|< q W |S )N)�typer r r* r& )r r/ r r, r r r � cloneNode� s z*getETreeBuilder..Element.cloneNodec S sl |j r"|j d j j| jj7 _n0|jjs2d|j_| jjd k rR|j j| jj7 _d| j_tjj| |� d S )Nr r= r> )�
childNodesr r? r2 r �Node�reparentChildren)r Z newParentr r r rF � s z1getETreeBuilder..Element.reparentChildren)N)N)�__name__�
__module__� __qualname__r r r r �propertyr r r! r r# r- r* r. r0 rD r4 r7 r: r< rA rC rF r )r r r r s*
r c s2 e Zd Z� fdd�Zdd� Zdd� Zeee�ZdS )z getETreeBuilder..Commentc s" � j |�| _d | _g | _g | _d S )N)�Commentr r r r )r r@ )r r r r � s z)getETreeBuilder..Comment.__init__c S s | j jS )N)r r2 )r r r r �_getData� s z)getETreeBuilder..Comment._getDatac S s || j _d S )N)r r2 )r r, r r r �_setData� s z)getETreeBuilder..Comment._setDataN)rG rH rI r rL rM rJ r@ r )r r r rK � s rK c sL e Zd Z� fdd�Zdd� Zdd� Zeee�Zdd� Zd d
� Z eee �Z
d S ) z%getETreeBuilder..DocumentTypec s$ � j | d� || j_|| _|| _d S )Nz
)r r r2 �publicId�systemId)r r rN rO )r r r r � s z.getETreeBuilder..DocumentType.__init__c S s | j jdd�S )NrN r= )r �get)r r r r � _getPublicId� s z2getETreeBuilder..DocumentType._getPublicIdc S s |d k r| j jd|� d S )NrN )r r) )r r, r r r � _setPublicId� s z2getETreeBuilder..DocumentType._setPublicIdc S s | j jdd�S )NrO r= )r rP )r r r r � _getSystemId� s z2getETreeBuilder..DocumentType._getSystemIdc S s |d k r| j jd|� d S )NrO )r r) )r r, r r r � _setSystemId� s z2getETreeBuilder..DocumentType._setSystemIdN) rG rH rI r rQ rR rJ rN rS rT rO r )r r r � DocumentType� s
rU c s e Zd Z� fdd�ZdS )z!getETreeBuilder..Documentc s � j | d� d S )N�
DOCUMENT_ROOT)r )r )r r r r � s z*getETreeBuilder..Document.__init__N)rG rH rI r r )r r r �Document� s rW c s e Zd Z� fdd�ZdS )z)getETreeBuilder..DocumentFragmentc s � j | d� d S )NZDOCUMENT_FRAGMENT)r )r )r r r r � s z2getETreeBuilder..DocumentFragment.__init__N)rG rH rI r r )r r r �DocumentFragment� s rX c s* g � d�� �fdd� ��| d� dj � �S )Nr c sz t | d�s| j� } | jdkrz| jd�s0| jd�rd| jd�pW | j�rv�jd
d |d | jf � d S )Nr z
rN rO r= zz
rV z #documentz|%s"%s"� r z Document node cannot have tailr" z$Document node cannot have attributesz|%sz%s %sz|%s<%s>z
|%s%s="%s")�hasattr�getrootr rP r5 r2 r? � TypeErrorr3 r" �
tag_regexp�match�groupsr
�prefixesr& �sorted) r/ �indentrN rO Znsmatchr �ns�prefixr* r, Z attr_string�child)�ElementTreeCommentType�rv�serializeElementr r rh � sZ
"
zAgetETreeBuilder..testSerializer..serializeElement�
)r )�join)r/ )rf )rg rh r �testSerializer� s 7
z'getETreeBuilder..testSerializerc s2 g �t j� � ��� ��fdd���| � dj��S )z4Serialize an element and its child nodes to a stringc s� t | � j�r| j� } | jdkr|| jd�s2| jd�rf| jd�p>d}| jd�pLd}�jd| j||f � n�jd| jf � �n| jdkr�| jd k r��j| j� | jd k r�td��t | d �r�t
| j �r�td
��x�| D ] }�|� q�W n�| j�k�r�jd | jf � n�| j �s$�jd �j | j�f � n2d
j
�fdd�| j j� D ��}�jd| j|f � | j�rj�j| j� x| D ]}�|� �qpW �jd| jf � | j�r��j| j� d S )Nz
rN rO r= z z
rV z Document node cannot have tailr" z$Document node cannot have attributesz z<%s>rY c s" g | ]\}}d � j |�|f �qS )z%s="%s")� fromXmlName)�.0r r, )�filterr r �
& s zOgetETreeBuilder..tostring..serializeElement..z<%s %s>z)r' r r[ r rP r5 r2 r? r\ rZ r3 r" rl rj r& )r/ rN rO re �attr)r rf rn rg rh r r rh s@
z;getETreeBuilder..tostring..serializeElementr= )r Z
InfosetFilterrj )r/ )r rf )rn rg rh r �tostring s
-z!getETreeBuilder..tostringc sD e Zd Z�Z�Z�Z� Z�Z�Z�fdd�Z �fdd�Z
dd� Z dS )z$getETreeBuilder..TreeBuilderc s � |�S )Nr )r r/ )rk r r rk A s z3getETreeBuilder..TreeBuilder.testSerializerc s< � r | j jS | jd k r*| j jjd| j �S | j jjd�S d S )Nz{%s}htmlr
)Zdocumentr ZdefaultNamespace�find)r )�fullTreer r � getDocumentD s
z0getETreeBuilder..TreeBuilder.getDocumentc S s t jj| �jS )N)r � TreeBuilder� getFragmentr )r r r r rv N s z0getETreeBuilder..TreeBuilder.getFragmentN) rG rH rI Z
documentClassZ doctypeClassZ elementClassZ commentClassZ
fragmentClass�implementationrk rt rv r )rK rW rX rU r �ElementTreeImplementationrs rk r r ru 9 s
ru )rK r r rE ru �locals)rx rs rq ru r )
rK rW rX rU r r rf rx rs rk r �getETreeBuilder s ~ >6$rz )F)Z
__future__r r r Zpip._vendor.sixr �rer= r r r
r Z_utilsr �compiler] rz ZgetETreeModuler r r r � s
E