o
    /h%                     @   s   d Z ddlmZ ddlmZ ddlmZmZ ddlmZm	Z	 zddl
mZ W n ey3   dd ZY nw G d	d
 d
ejZdd ZG dd deZG dd dZdd ZdS )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
    )ContentHandler)etree)ElementTree
SubElement)CommentProcessingInstruction)GenericAliasc                 C   s   | j  d|j  dS )N[])__name__clsitem r   </var/www/html/myenv/lib/python3.10/site-packages/lxml/sax.py_GenericAlias      r   c                   @   s   e Zd ZdZdS )SaxErrorzGeneral SAX error.
    N)r   
__module____qualname____doc__r   r   r   r   r      s    r   c                 C   s,   | d dkrt | dd  ddS d | fS )Nr   {   })tuplesplit)tagr   r   r   	_getNsTag"   s   r   c                   @   s   e Zd ZdZd"ddZdd ZeeejdZdd	 Zd
d Z	dd Z
dd Zdd Zdd Zd"ddZdd Zdd Zd"ddZdd Zdd ZeZd d! ZdS )#ElementTreeContentHandlerz/Build an lxml ElementTree from SAX events.
    Nc                 C   sL   t |  d | _g | _g | _d | _d d gi| _i | _|d u r!tj	}|| _
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_etree8   s   
z$ElementTreeContentHandler._get_etree)docc                 C      d S r   r   )r)   locatorr   r   r   setDocumentLocator>      z,ElementTreeContentHandler.setDocumentLocatorc                 C   r.   r   r   r+   r   r   r   startDocumentA   r1   z'ElementTreeContentHandler.startDocumentc                 C   r.   r   r   r+   r   r   r   endDocumentD   r1   z%ElementTreeContentHandler.endDocumentc                 C   sT   || j |< z
| j| | W n ty   |g| j|< Y nw |d u r(|| _d S d S r   )r&   r%   appendKeyErrorr$   )r)   prefixurir   r   r   startPrefixMappingG   s   

z,ElementTreeContentHandler.startPrefixMappingc                 C   s,   | j | }|  |d u r|d | _d S d S N)r%   popr$   )r)   r6   ns_uri_listr   r   r   endPrefixMappingP   s
   
z*ElementTreeContentHandler.endPrefixMappingc                 C   s8   |\}}|rd| }|S | j rd| j |f }|S |}|S )N{%s}%s)r$   )r)   ns_name_tuplens_uri
local_nameel_tagr   r   r   	_buildTagV   s   z#ElementTreeContentHandler._buildTagc                 C   s   |  |}|r6i }z| }W n ty   | }Y nw |D ]\}}|d r,d| }	n|d }	|||	< qnd }| j}
| jd u re| ||| j }| _| jr^t	|dr^| jD ]}|
| qV| jd d = n
t|
d ||| j}|
| | j  d S )Nr   r>   r   addpreviousr:   )rC   	iteritemsAttributeErroritemsr#   r!   r(   r&   r"   hasattrrD   r   r4   clear)r)   ns_nameqname
attributesel_nameattrsiter_attributes
name_tuplevalue	attr_nameelement_stackelementsiblingr   r   r   startElementNS`   s8   






z(ElementTreeContentHandler.startElementNSc                 C   s8   t ||}| jd u r| j| d S | jd | d S r9   )r   r!   r"   r4   r#   )r)   targetdatapir   r   r   processingInstruction   s   

z/ElementTreeContentHandler.processingInstructionc                 C   s.   | j  }| |}||jkrtd| d S )NzUnexpected element closed: )r#   r;   rC   r   r   )r)   rJ   rK   rT   rB   r   r   r   endElementNS   s
   


z&ElementTreeContentHandler.endElementNSc                 C   s,   |rdd |  D }| d |f|| d S )Nc                 S   s   i | ]	\}}d |f|qS r   r   ).0kvr   r   r   
<dictcomp>   s    z:ElementTreeContentHandler.startElement.<locals>.<dictcomp>)rG   rV   )r)   namerL   r   r   r   startElement   s   z&ElementTreeContentHandler.startElementc                 C   s   |  d |f| d S r   )r[   )r)   r`   r   r   r   
endElement   r   z$ElementTreeContentHandler.endElementc                 C   sN   | j d }z|d }|jpd| |_W d S  ty&   |jpd| |_Y d S w )Nr:    )r#   tail
IndexErrortext)r)   rX   last_elementr   r   r   
characters   s   
z$ElementTreeContentHandler.charactersc                 C   s
   t | |S r   )r   r   r   r   r   __class_getitem__   s   
z+ElementTreeContentHandler.__class_getitem__r   )r   r   r   r   r    r,   propertyr   r0   r2   r3   r8   r=   rC   rV   rZ   r[   ra   rb   rh   ignorableWhitespaceri   r   r   r   r   r   )   s&    
	

!

r   c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )ElementTreeProducerz5Produces SAX events for an element and children.
    c                 C   sR   z|  }W n ty   |}Y nw || _|| _ddlm} || _|i i | _d S )Nr   )AttributesNSImpl)getrootrF   _element_content_handlerxml.sax.xmlreaderrm   _attr_class_empty_attributes)r)   element_or_treecontent_handlerrT   
attr_classr   r   r   r       s   zElementTreeProducer.__init__c                 C   s   | j   | j}t|dr<g }| }t|dd tu r,|| | }t|dd tu s|d d d D ]}| |i  q3| |i  t|dre|	 }t|dd tu re| |i  |	 }t|dd tu sS| j 
  d S )Ngetpreviousr   r:   getnext)rp   r2   ro   rH   rw   getattrr   r4   _recursive_saxifyrx   r3   )r)   rT   siblingsrU   r   r   r   saxify   s&   



zElementTreeProducer.saxifyc                 C   s  | j }|j}|tu s|tu r&|tu r||j|j |j}|r$|| d S |j	}g }||krF|
 D ]\}}	|||	krE|||	f q3|
 }
|
rvi }i }|
D ]\}}t|}|||< | j|d |d |d dd||< qR| ||}n| j}t|\}	}| j|	|||jdd}|D ]
\}}||| q||	|f|| |j}|r|| |D ]}| || q||	|f| |D ]	\}}|| q|j}|r|| d S d S )Nr   r   T)preferred_prefixis_attributeF)r~   )rp   r   r   r   rZ   rW   rf   rd   rh   nsmaprG   getr4   r   _build_qnamerr   rs   r6   r8   rV   rz   r[   r=   )r)   rT   parent_nsmapru   r   rd   element_nsmapnew_prefixesr6   r@   attribsattr_valuesattr_qnamesattr_ns_namerQ   attr_ns_tuplesax_attributesrA   rK   r7   rf   childr   r   r   rz      sh   


z%ElementTreeProducer._recursive_saxifyc                    sv    d u r|S |s| | kr|}n fdd| D }t|dkr'|d n|r-t|nd }|d u r5|S |d | S )Nc                    s$   g | ]\}}|d ur| kr|qS r   r   )r\   pfxr7   r@   r   r   
<listcomp>
  s
    z4ElementTreeProducer._build_qname.<locals>.<listcomp>r   r   :)r   rG   lenmin)r)   r@   rA   r   r}   r~   r6   
candidatesr   r   r   r     s   
z ElementTreeProducer._build_qnameN)r   r   r   r   r    r|   rz   r   r   r   r   r   rl      s    6rl   c                 C   s   t | | S )zlOne-shot helper to generate SAX events from an XML tree and fire
    them against a SAX ContentHandler.
    )rl   r|   )rt   ru   r   r   r   r|     s   r|   N)r   xml.sax.handlerr   lxmlr   
lxml.etreer   r   r   r   typesr   r   ImportError	LxmlErrorr   r   r   rl   r|   r   r   r   r   <module>   s   ~s