File: //usr/local/lib64/python3.6/site-packages/lxml/__pycache__/sax.cpython-36.pyc
3
��?e�$ � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZm Z ddlm
Z
m Z G dd� dej �Z
d d
� ZG d d � d e�ZG d
d� de�Zdd� ZdS )a?
SAX-based adapter to copy trees from/to the Python standard library.
Use the `ElementTreeContentHandler` class to build an ElementTree from
SAX events.
Use the `ElementTreeProducer` class or the `saxify()` function to fire
the SAX events of an ElementTree against a SAX ContentHandler.
See https://lxml.de/sax.html
� )�absolute_import)�ContentHandler)�etree)� ElementTree�
SubElement)�Comment�ProcessingInstructionc @ s e Zd ZdZdS )�SaxErrorzGeneral SAX error.
N)�__name__�
__module__� __qualname__�__doc__� r r �(/tmp/pip-build-_veujq9v/lxml/lxml/sax.pyr s r c C s0 | d dkr$t | dd � jdd��S d | fS d S )Nr �{� �})�tuple�split)�tagr r r � _getNsTag s r c @ s� e Zd ZdZd dd�Zdd� Zeeejd�Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
d!dd�Zdd� Zdd� Zd"dd�Zd d � Zd d � ZeZdS )#�ElementTreeContentHandlerz/Build an lxml ElementTree from SAX events.
Nc C sL t j| � d | _g | _g | _d | _d d gi| _i | _|d krBtj }|| _
d S )N) r �__init__�_root�_root_siblings�_element_stack� _default_ns� _ns_mapping�
_new_mappingsr �Element� _makeelement)�self� makeelementr r r r & s
z"ElementTreeContentHandler.__init__c C s
t | j�S )z=Contains the generated ElementTree after parsing is finished.)r r )r! r r r �
_get_etree2 s z$ElementTreeContentHandler._get_etree)�docc C s d S )Nr )r! �locatorr r r �setDocumentLocator8 s z,ElementTreeContentHandler.setDocumentLocatorc C s d S )Nr )r! r r r �
startDocument; s z'ElementTreeContentHandler.startDocumentc C s d S )Nr )r! r r r � endDocument> s z%ElementTreeContentHandler.endDocumentc C sR || j |< y| j| j|� W n tk
r> |g| j|< Y nX |d krN|| _d S )N)r r �append�KeyErrorr )r! �prefix�urir r r �startPrefixMappingA s
z,ElementTreeContentHandler.startPrefixMappingc C s( | j | }|j� |d kr$|d | _d S )Nr ���)r �popr )r! r+ Z ns_uri_listr r r �endPrefixMappingJ s
z*ElementTreeContentHandler.endPrefixMappingc C s4 |\}}|rd| }n| j r,d| j |f }n|}|S )Nz{%s}%s)r )r! Z
ns_name_tuple�ns_uri�
local_name�el_tagr r r � _buildTagP s
z#ElementTreeContentHandler._buildTagc
C s | j |�}|rri }y |j� }W n tk
r: |j� }Y nX x8|D ]*\}}|d r\d| } n|d } ||| < qBW nd }| j}
| jd kr�| j||| j� } | _| jr�t | d�r�x| jD ]} | j
| � q�W | jd d �= nt |
d ||| j�} |
j | � | jj
� d S )Nr z{%s}%sr � addpreviousr. )r4 � iteritems�AttributeError�itemsr r r r r �hasattrr5 r r) �clear)
r! �ns_name�qname�
attributesZel_name�attrsZiter_attributesZ
name_tuple�value� attr_nameZ
element_stack�element�siblingr r r �startElementNSZ s0
z(ElementTreeContentHandler.startElementNSc C s6 t ||�}| jd kr"| jj|� n| jd j|� d S )Nr r. )r r r r) r )r! �target�data�pir r r �processingInstruction{ s
z/ElementTreeContentHandler.processingInstructionc C s. | j j� }| j|�}||jkr*td| ��d S )NzUnexpected element closed: )r r/ r4 r r )r! r; r< rA r3 r r r � endElementNS� s
z&ElementTreeContentHandler.endElementNSc C s0 |rt dd� |j� D ��}| jd |f||� d S )Nc S s g | ]\}}d |f|f�qS )Nr )�.0�k�vr r r �
<listcomp>� s z:ElementTreeContentHandler.startElement.<locals>.<listcomp>)�dictr8 rC )r! �namer= r r r � startElement� s z&ElementTreeContentHandler.startElementc C s | j d |f|� d S )N)rH )r! rN r r r �
endElement� s z$ElementTreeContentHandler.endElementc C sP | j d }y |d }|jp d| |_W n$ tk
rJ |jp>d| |_Y nX d S )Nr � r. r. )r �tail�
IndexError�text)r! rE Z last_elementr r r �
characters� s
z$ElementTreeContentHandler.characters)N)N)N)r
r r r
r r# �propertyr r& r' r( r- r0 r4 rC rG rH rO rP rU �ignorableWhitespacer r r r r # s"
!
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �ElementTreeProducerz5Produces SAX events for an element and children.
c C sT y |j � }W n tk
r$ |}Y nX || _|| _ddlm} || _|i i �| _d S )Nr )�AttributesNSImpl)�getrootr7 �_element�_content_handlerZxml.sax.xmlreaderrY � _attr_class�_empty_attributes)r! �element_or_tree�content_handlerrA Z
attr_classr r r r � s
z ElementTreeProducer.__init__c C s� | j j� | j}t|d�rrg }|j� }x&t|dd �tkrL|j|� |j� }q(W x"|d d d� D ]}| j|i � q^W | j|i � t|d�r�|j � }x(t|dd �tkr�| j|i � |j � }q�W | j j
� d S )N� getpreviousr r �getnextr. ) r\ r' r[ r9 ra �getattrr r) �_recursive_saxifyrb r( )r! rA ZsiblingsrB r r r �saxify� s"
zElementTreeProducer.saxifyc C s� | j }|j}|tks |tkrL|tkr4|j|j|j� |j}|rH|j|� d S |j }g }||kr�x0|j
� D ]$\}} |j |�| krh|j || f� qhW |j
� }
|
r�i } i } x@|
D ]8\}
}t
|
�}|| |< | j|d |d |d dd�| |< q�W | j| | �}n| j}t
|�\} }| j| |||jdd�}x |D ]\}}|j||� �q"W |j| |f||� |j}|�rd|j|� x|D ]}| j||� �qjW |j| |f|� x |D ]\}}|j|� �q�W |j}|�r�|j|� d S )Nr r T)�preferred_prefix� is_attributeF)rg )r\ r r r rG rD rT rR rU �nsmapr8 �getr) r � _build_qnamer] r^ r+ r- rC rd rH r0 )r! rA Z parent_nsmapr` r rR Z
element_nsmapZ new_prefixesr+ r1 ZattribsZ attr_valuesZ attr_qnamesZ attr_ns_namer? Z
attr_ns_tupleZsax_attributesr2 r< r, rT �childr r r rd � s\
z%ElementTreeProducer._recursive_saxifyc sx � d kr |S | r&|j |�� kr&|}n:� fdd�|j� D �}t|�dkrP|d n|r\t|�nd }|d krl|S |d | S )Nc s$ g | ] \}}|d k r|� kr|�qS )Nr )rI �pfxr, )r1 r r rL s z4ElementTreeProducer._build_qname.<locals>.<listcomp>r r �:)ri r8 �len�min)r! r1 r2 rh rf rg r+ �
candidatesr )r1 r rj � s z ElementTreeProducer._build_qnameN)r
r r r
r re rd rj r r r r rX � s
6rX c C s t | |�j� S )zlOne-shot helper to generate SAX events from an XML tree and fire
them against a SAX ContentHandler.
)rX re )r_ r` r r r re s re N)r
�
__future__r Zxml.sax.handlerr �lxmlr Z
lxml.etreer r r r Z LxmlErrorr r r �objectrX re r r r r �<module>
s |s