3
�PfQ7 � @ s� d Z ddlmZmZmZ ddlZddlZddlZddlm Z ddl
m Z ddlm
Z
dd lm Z
dd
lmZ ddlj Z d Zejd �Ze jd
�jZG dd� de�ZG dd� de�Zdd� Zdd� ZG dd� de j�ZdS )a� Module for supporting the lxml.etree library. The idea here is to use as much
of the native library as possible, without using fragile hacks like custom element
names that break between releases. The downside of this is that we cannot represent
all possible trees; specifically the following are known to cause problems:
Text or comments as siblings of the root element
Docypes with no name
When any of these things occur, we emit a DataLossWarning
� )�absolute_import�division�unicode_literalsN� )�base� )�DataLossWarning)� constants)�etree)� _ihatexmlTz
{([^}]*)}(.*)Zasdc @ s e Zd Zdd� ZdS )� DocumentTypec C s || _ || _|| _d S )N)�name�publicId�systemId)�selfr
r r � r � /usr/lib/python3.6/etree_lxml.py�__init__# s zDocumentType.__init__N)�__name__�
__module__� __qualname__r r r r r r " s r c @ s, e Zd Zdd� Zdd� Zdd� Zee�ZdS )�Documentc C s d | _ g | _d S )N)� _elementTree� _childNodes)r r r r r * s zDocument.__init__c C s | j j� j|j� d S )N)r �getrootZaddnext�_element)r �elementr r r � appendChild. s zDocument.appendChildc C s | j S )N)r )r r r r �_getChildNodes1 s zDocument._getChildNodesN)r r r r r r �propertyZ
childNodesr r r r r ) s r c s6 g �t jdd�� d� ��fdd� ��| d� dj��S )NT)�preventDoubleDashCommentsr c
sD t | d��s8t | d�rˆjd� | jjrz| jjp6| jjsFd| jj }nd| jj| jj| jjf }�jdd|d |f � | j� }x|j� d k r�|j� }q�W x�|d k r��||d � |j � }q�W nrt
| t �s�t
| t ��r t
| t �s�t
jd dks�t��jd
d| | f � n(�jd � x | D ]}�||d � �q W �n| jtk�r��jd d| | jf � t | d
��r@| j�r@�jd
d| | jf � �n�t
| tj��s�t�tjj| j�}|d k �r�|jd�}|jd�}tj| }�jdd| |� j |�f � n �jdd| � j | j�f � t | d��r�g }xr| j j � D ]d\} }
tj| �}|d k �rx|j � \}} � j | �} tj| }d|| f } n
� j | �} |j| |
f� �q.W x2t |�D ]&\} }
�jdd|d | |
f � �q�W | j�r�jd
d|d | jf � |d7 }x| D ]} �| |� �q�W t | d
��r@| j�r@�jd
d|d | jf � d S )N�tagr z #documentz
zz|%s%s� r r z|%s"%s"z#document-fragmentz|%s�tailr z
|%s<%s %s>z|%s<%s>�attribz%s %sz
|%s%s="%s")!�hasattr�append�docinfo� internalDTDZ public_idZ
system_url� root_namer Z getpreviousZgetnext�
isinstance�str�bytes�sys� version_info�AssertionErrorr! � comment_type�textr# r
Z_Element�etree_builders�
tag_regexp�match�groupr �prefixes� fromXmlNamer$ �items�groups�sorted)
r �indent�dtd_strZ next_elementZnsmatch�nsr! �prefix�
attributesr
�valueZ attr_string�child)�
infosetFilter�rv�serializeElementr r rD ; st
"
z(testSerializer..serializeElement�
)r )r �
InfosetFilter�join)r r )rB rC rD r �testSerializer7 s
F
rH c s$ g � � �fdd���| � dj � �S )z4Serialize an element and its child nodes to a stringc s
t | d�sH| jjr:| jjr$| jj}n d| jj }� j|� �| j� � n�| jtkrf� jd| j f � n�| j
s�� jd| jf � n.dj dd� | j
j � D ��}� jd| j|f � | j r�� j| j � x| D ] }�|� q�W � jd | jf � t | d
�o�| j
�r� j| j
� d S ) Nr! z
z z<%s>r" c S s g | ]\}}d ||f �qS )z%s="%s"r )�.0r
r@ r r r �
� s z6tostring..serializeElement..z<%s %s>zr# )r% r' r( �doctyper) r& r r! r0 r1 r$ rG r8 r# )r r< �attrrA )rC rD r r rD � s*
z"tostring..serializeElement� )rG )r r )rC rD r �tostring� s rN c sz e Zd ZeZeZdZdZeZ e
Z ddd�Z dd� Z
dd� Zd d
� Zd d � Zd
d� Zddd�Zd� fdd� Zdd� Z� ZS )� TreeBuilderNFc s� t jt|d��tjdd� �| _|| _G �fdd�dt�� G � ��fdd�d�j�}G ��fdd �d �j �}|| _
|| _ t j
j| |� d S )
N)�fullTreeT)r c s& e Zd Zd� fdd� Z� fdd�ZdS )z(TreeBuilder.__init__..AttributesNc sv |d kr i }|| _ tj| |� xR| j� D ]F\}}t|t�rVd|d � j|d �f }n
� j|�}|| j j j|< q(W d S )Nz{%s}%sr r )r �dictr r8 r* �tuple�coerceAttributer$ )r r r@ �keyr
)rB r r r � s
z1TreeBuilder.__init__..Attributes.__init__c sP t j| ||� t|t�r4d|d � j|d �f }n
� j|�}|| jjj|< d S )Nz{%s}%sr r )rQ � __setitem__r* rR rS r r$ )r rT r@ r
)rB r r rU � s
z4TreeBuilder.__init__..Attributes.__setitem__)N)r r r r rU r )rB r r �
Attributes� s rV c sx e Zd Z� ��fdd�Z�fdd�Z�fdd�Zeee�Zdd� Z� fd d
�Z eee �Z
d��fd d
� Z �fdd�Z d S )z%TreeBuilder.__init__..Elementc s* �j |�}�jj| ||d� � | �| _d S )N)� namespace)�
coerceElement�Elementr � _attributes)r r
rW )rV �builderrB r r r � s
z.TreeBuilder.__init__..Element.__init__c s$ � j |�| _| j| j| j�| j_d S )N)rX �_nameZ _getETreeTagZ
_namespacer r! )r r
)rB r r �_setName� s z.TreeBuilder.__init__..Element._setNamec s � j | j�S )N)r7 r\ )r )rB r r �_getName� s z.TreeBuilder.__init__..Element._getNamec S s | j S )N)rZ )r r r r �_getAttributes� s z4TreeBuilder.__init__..Element._getAttributesc s � | |�| _ d S )N)rZ )r r? )rV r r �_setAttributes� s z4TreeBuilder.__init__..Element._setAttributesNc s �j |�}� jj| ||� d S )N)ZcoerceCharactersrY �
insertText)r �dataZ insertBefore)r[ rB r r ra � s
z0TreeBuilder.__init__..Element.insertTextc s � j j| |� d S )N)rY r )r rA )r[ r r r � s z1TreeBuilder.__init__..Element.appendChild)N)
r r r r r] r^ r r
r_ r` r? ra r r )rV r[ rB r r rY � s
rY c s8 e Zd Z� �fdd�Z�fdd�Zdd� Zeee�ZdS )z%TreeBuilder.__init__..Commentc s �j |�}� jj| |� d S )N)�
coerceComment�Commentr )r rb )r[ rB r r r � s
z.TreeBuilder.__init__..Comment.__init__c s � j |�}|| j_d S )N)rc r r1 )r rb )rB r r �_setData� s
z.TreeBuilder.__init__..Comment._setDatac S s | j jS )N)r r1 )r r r r �_getData� s z.TreeBuilder.__init__..Comment._getDataN)r r r r re rf r rb r )r[ rB r r rd � s rd )r2 ZgetETreeModuler
r rF rB �namespaceHTMLElementsrQ rY rd � elementClass� commentClassr rO r )r rg rP rY rd r )rV r[ rB r r � s zTreeBuilder.__init__c C s$ t jj| � | j| _g | _d | _d S )N)r rO �reset�insertCommentInitial�
insertComment�initial_commentsrK )r r r r rj s zTreeBuilder.resetc C s t |�S )N)rH )r r r r r rH s zTreeBuilder.testSerializerc C s t r | jjS | jjj� S d S )N)rP �documentr r )r r r r � getDocument s zTreeBuilder.getDocumentc C sF g }| j d j}|jr"|j|j� |jt|�� |jrB|j|j� |S )Nr )� openElementsr r1 r& �extend�listr# )r Zfragmentr r r r � getFragment s zTreeBuilder.getFragmentc C sh |d }|d }|d }|s0t jdt� d | _n4| jj|�}||krPt jdt� | j|||�}|| _d S )Nr
r r z#lxml cannot represent empty doctypez%lxml cannot represent non-xml doctype)�warnings�warnr rK rB rX � doctypeClass)r �tokenr
r r Z coercedNamerK r r r �
insertDoctype s zTreeBuilder.insertDoctypec C s6 |d ks|| j kst�| j jd ks&t�| jj|� d S )N)rn r/ r rm r& )r rb �parentr r r rk , s z TreeBuilder.insertCommentInitialc sB || j kr,| j jj� d jtkr,tjdt� tt | �j
||� d S )Nr z@lxml cannot represent adjacent comments beyond the root elements�����) rn r r r! r0 rt ru r �superrO rl )r rb ry )� __class__r r �insertCommentMain1 s
z TreeBuilder.insertCommentMainc C s� d}| j r�| j jst�|d| j j 7 }| j jdk s>| j jdk r�|d| jj| j jpRd� 7 }| j jr�| j j}|jd�dkr�|jd�dkr�tj dt
� |j dd �}|jd�dkr�|d
| 7 }q�|d | 7 }n|d 7 }|d
7 }| j j|d kr�tj dt
� |d7 }t j
|�}x*| jD ] }| j|d �}|j|j� �qW | j� | _|j� | j_|d }|jd| j�}|dk�rn|} n d||f } | |_| j||�}
||
_| jjj|
� | j j|
� | j | _ dS )zCreate the document rootrM z r
zGlxml cannot represent doctype with a different name to the root elementz$rb rW z{%s}%s) rK r
r/ r r rB Z coercePubid�findrt ru r �replacer
Z
fromstringrm ri Z addpreviousr �
documentClassrn Z getroottreer �getZdefaultNamespacer! rh r r& rp r} rl ) r rw ZdocStrZsysid�root�
comment_tokenZcommentr
rW Z etree_tagZ root_elementr r r �
insertRoot7 sL
zTreeBuilder.insertRoot)F)N)N)r r r r r� r rv rh ri Z
fragmentClassr
�implementationr rj rH ro rs rx rk r} r� �
__classcell__r r )r| r rO � s
L
rO ) �__doc__Z
__future__r r r rt �rer- rM r r r r
r2 r Z
lxml.etreerP �compiler3 rd r! r0 �objectr r rH rN rO r r r r �
s$
O)