File: //home/arjun/projects/env/lib64/python3.10/site-packages/lxml/__pycache__/sax.cpython-310.pyc
o
we�$ � @ 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 �A/home/arjun/projects/env/lib/python3.10/site-packages/lxml/sax.pyr s r c C s, | d dkrt | dd � �dd��S d | fS )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 �| � d | _g | _g | _d | _d d gi| _i | _|d u r!tj }|| _
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 � d S r r )r" �locatorr r r �setDocumentLocator8 � z,ElementTreeContentHandler.setDocumentLocatorc C r' r r r$ r r r �
startDocument; r* z'ElementTreeContentHandler.startDocumentc C r' r r r$ r r r � endDocument> r* z%ElementTreeContentHandler.endDocumentc C sT || j |< z
| j| �|� W n ty |g| j|< Y nw |d u r(|| _d S d S r )r r �append�KeyErrorr )r" �prefix�urir r r �startPrefixMappingA s
�
�z,ElementTreeContentHandler.startPrefixMappingc C s, | j | }|�� |d u r|d | _d S d S �N���)r �popr )r" r/ � ns_uri_listr r r �endPrefixMappingJ s
�z*ElementTreeContentHandler.endPrefixMappingc C s8 |\}}|r d| }|S | j rd| j |f }|S |}|S )N�{%s}%s)r )r" �
ns_name_tuple�ns_uri�
local_name�el_tagr r r � _buildTagP s ��z#ElementTreeContentHandler._buildTagc
C s� | � |�}|r6i }z|�� }W n
ty |�� }Y nw |D ]\}}|d r,d| } n|d } ||| < q nd }| j}
| jd u re| �||| j� } | _| jr^t | d�r^| jD ]} | �
| � qV| jd d �= n
t |
d ||| j�} |
� | � | j�
� d S )Nr r7 r � addpreviousr3 )r< � iteritems�AttributeError�itemsr r r! r r �hasattrr= r r- �clear)
r" �ns_name�qname�
attributes�el_name�attrs�iter_attributes�
name_tuple�value� attr_name�
element_stack�element�siblingr r r �startElementNSZ s8
�
�
�
�
z(ElementTreeContentHandler.startElementNSc C s8 t ||�}| jd u r| j�|� d S | jd �|� d S r2 )r r r r- r )r" �target�data�pir r r �processingInstruction{ s
z/ElementTreeContentHandler.processingInstructionc C s. | j �� }| �|�}||jkrtd| ��d S )NzUnexpected element closed: )r r4 r< r r )r" rC rD rM r; r r r � endElementNS� s
�z&ElementTreeContentHandler.endElementNSc C s0 |r
t dd� |�� D ��}| �d |f||� d S )Nc S s g | ]
\}}d |f|f�qS r r )�.0�k�vr r r �
<listcomp>� s z:ElementTreeContentHandler.startElement.<locals>.<listcomp>)�dictr@ rO )r" �namerE r r r � startElement� s
�z&ElementTreeContentHandler.startElementc C s | � d |f|� d S r )rT )r" rZ r r r �
endElement� s z$ElementTreeContentHandler.endElementc C sN | j d }z|d }|jpd| |_W d S ty& |jp d| |_Y d S w )Nr3 � )r �tail�
IndexError�text)r" rQ � last_elementr r r �
characters� s
�z$ElementTreeContentHandler.charactersr )r
r r r
r r% �propertyr r) r+ r, r1 r6 r< rO rS rT r[ r\ rb �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 sR z|� � }W n ty |}Y nw || _|| _ddlm} || _|i i �| _d S )Nr )�AttributesNSImpl)�getrootr? �_element�_content_handler�xml.sax.xmlreaderrf � _attr_class�_empty_attributes)r" �element_or_tree�content_handlerrM �
attr_classr r r r � s � z ElementTreeProducer.__init__c C s� | j �� | j}t|d�r<g }|�� }t|dd �tu r,|�|� |�� }t|dd �tu s|d d d� D ]}| �|i � q3| �|i � t|d�re|� � }t|dd �tu re| �|i � |� � }t|dd �tu sS| j �
� d S )N� getpreviousr r3 �getnext) ri r+ rh rA rp �getattrr r- �_recursive_saxifyrq r, )r" rM �siblingsrN r r r �saxify� s&
�
�zElementTreeProducer.saxifyc C s� | j }|j}|tu s|tu r&|tu r|�|j|j� |j}|r$|�|� d S |j }g }||krF|�
� D ]\}} |� |�| krE|� || f� q3|�
� }
|
rvi } i } |
D ] \}
}t
|
�}|| |< | j|d |d |d dd�| |< qR| �| | �}n| j}t
|�\} }| j| |||jdd�}|D ]
\}}|�||� q�|�| |f||� |j}|r�|�|� |D ]}| �||� q�|�| |f|� |D ] \}}|�|� q�|j}|r�|�|� d S d S )Nr r T)�preferred_prefix� is_attributeF)rw )ri r r r rS rP r` r^ rb �nsmapr@ �getr- r � _build_qnamerk rl r/ r1 rO rs rT r6 )r" rM � parent_nsmaprn r r^ �
element_nsmap� new_prefixesr/ r9 �attribs� attr_values� attr_qnames� attr_ns_namerJ �
attr_ns_tuple�sax_attributesr: rD r0 r` �childr r r rs � sh �
� � �
�
�z%ElementTreeProducer._recursive_saxifyc sv � d u r|S |s|� |�� kr|}n � fdd�|�� D �}t|�dkr'|d n|r-t|�nd }|d u r5|S |d | S )Nc s$ g | ]\}}|d ur|� kr|�qS r r )rU �pfxr0 �r9 r r rX s
�z4ElementTreeProducer._build_qname.<locals>.<listcomp>r r �:)ry r@ �len�min)r" r9 r: rx rv rw r/ �
candidatesr r� r rz � s
� � z ElementTreeProducer._build_qnameN)r
r r r
r ru rs rz r r r r re � s 6re c C s t | |��� S )zlOne-shot helper to generate SAX events from an XML tree and fire
them against a SAX ContentHandler.
)re ru )rm rn r r r ru s ru N)r
�
__future__r �xml.sax.handlerr �lxmlr �
lxml.etreer r r r � LxmlErrorr r r �objectre ru r r r r �<module> s | s