a
    z|a2                     @   s   d dl mZmZmZ d dlmZ d dlZd dlmZ ddlm	Z	 ddlm
Z
 dd	lmZ dd
lmZ ddlmZ edZdddZeeZdS )    )absolute_importdivisionunicode_literals)	text_typeN)copy   )base   )	_ihatexml)	constants)
namespaces)moduleFactoryFactoryz{([^}]*)}(.*)Fc              	      s    djG fdddtjG fddd G fdddG fdd	d	G fd
ddfdd	fdd}G  	fdddtj}t S )NZasdc                       s   e Zd Zd$ fdd	Zdd Zdd Zdd	 ZeeeZd
d Z	dd Z
ee
e	Zdd Zdd ZeeeZdd Zdd ZeeeZdd Zdd Zdd Zdd Zd%ddZd d! Zd"d# ZdS )&z getETreeBuilder.<locals>.ElementNc                    s^   || _ || _ | ||| _|d u r:td | 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 Ilib/python3.9/site-packages/bleach/_vendor/html5lib/treebuilders/etree.py__init__   s    
z)getETreeBuilder.<locals>.Element.__init__c                 S   s   |d u r|}nd||f }|S )N{%s}%sr   )r   r   r   Z	etree_tagr   r   r   r   %   s    z-getETreeBuilder.<locals>.Element._getETreeTagc                 S   s   || _ | | j | j| j_d S N)r   r   r   r   tag)r   r   r   r   r   _setName,   s    z)getETreeBuilder.<locals>.Element._setNamec                 S   s   | j S r    )r   r   r   r   r   _getName0   s    z)getETreeBuilder.<locals>.Element._getNamec                 S   s   || _ | | j| j | j_d S r    )r   r   r   r   r!   )r   r   r   r   r   _setNamespace5   s    z.getETreeBuilder.<locals>.Element._setNamespacec                 S   s   | j S r    )r   r#   r   r   r   _getNamespace9   s    z.getETreeBuilder.<locals>.Element._getNamespacec                 S   s   | j jS r    )r   attribr#   r   r   r   _getAttributes>   s    z/getETreeBuilder.<locals>.Element._getAttributesc                 S   sV   | j j}|  |rR| D ]4\}}t|trDd|d |d f }n|}|||< qd S )Nr   r	   r   )r   r'   clearitems
isinstancetuple)r   
attributesZ	el_attribkeyvaluer   r   r   r   _setAttributesA   s    
z/getETreeBuilder.<locals>.Element._setAttributesc                 S   s   | j S r    )r   r#   r   r   r   _getChildNodesP   s    z/getETreeBuilder.<locals>.Element._getChildNodesc                 S   s*   | j d d = g | _|D ]}| | qd S r    )r   r   ZinsertChild)r   r/   elementr   r   r   _setChildNodesS   s    z/getETreeBuilder.<locals>.Element._setChildNodesc                 S   s   t | jjpt| jS )z,Return true if the node has children or text)boolr   textlenr#   r   r   r   
hasContent[   s    z+getETreeBuilder.<locals>.Element.hasContentc                 S   s$   | j | | j|j | |_d S r    )r   appendr   r   r   noder   r   r   appendChild_   s    z,getETreeBuilder.<locals>.Element.appendChildc                 S   s,   t | j|j}| j||j | |_d S r    )listr   indexinsertr   )r   r:   ZrefNoder=   r   r   r   insertBefored   s    z-getETreeBuilder.<locals>.Element.insertBeforec                 S   s$   | j | | j|j d |_d S r    )r   remover   r   r9   r   r   r   removeChildi   s    z,getETreeBuilder.<locals>.Element.removeChildc                 S   s   t | js,| jjsd| j_| j j|7  _n|d u rb| jd jsLd| jd _| jd  j|7  _nxt| j}||j}|dkr| j|d  jsd| j|d  _| j|d   j|7  _n | jjsd| j_| j j|7  _d S )N r   r   )r6   r   r5   tailr<   r=   )r   datar?   Zchildrenr=   r   r   r   
insertTextn   s"    

z+getETreeBuilder.<locals>.Element.insertTextc                 S   s.   t | | j| j}| jjr*t| jj|j_|S r    )typer   r   r   r'   r   r   r2   r   r   r   	cloneNode   s    z*getETreeBuilder.<locals>.Element.cloneNodec                 S   sl   |j r"|j d j j| jj7  _n0|jjs2d|j_| jjd urR|j j| jj7  _d| j_tj| | d S )NrC   rB   )
childNodesr   rD   r5   r   NodereparentChildren)r   Z	newParentr   r   r   rL      s    z1getETreeBuilder.<locals>.Element.reparentChildren)N)N)__name__
__module____qualname__r   r   r"   r$   propertyr   r%   r&   r   r(   r0   r-   r1   r3   rJ   r7   r;   r?   rA   rF   rI   rL   r   r   r   r   r      s*   




r   c                       s2   e Zd Z fddZdd Zdd ZeeeZdS )z getETreeBuilder.<locals>.Commentc                    s"     || _d | _g | _g | _d S r    )Commentr   r   r   r   )r   rE   r   r   r   r      s    z)getETreeBuilder.<locals>.Comment.__init__c                 S   s   | j jS r    r   r5   r#   r   r   r   _getData   s    z)getETreeBuilder.<locals>.Comment._getDatac                 S   s   || j _d S r    rR   r   r/   r   r   r   _setData   s    z)getETreeBuilder.<locals>.Comment._setDataN)rM   rN   rO   r   rS   rU   rP   rE   r   r   r   r   rQ      s   rQ   c                       sL   e Zd Z fddZdd Zdd ZeeeZdd Zd	d
 Z	eee	Z
dS )z%getETreeBuilder.<locals>.DocumentTypec                    s$     | d || j_|| _|| _d S )N
<!DOCTYPE>)r   r   r5   publicIdsystemId)r   r   rW   rX   r   r   r   r      s    z.getETreeBuilder.<locals>.DocumentType.__init__c                 S   s   | j ddS )NrW   rB   r   getr#   r   r   r   _getPublicId   s    z2getETreeBuilder.<locals>.DocumentType._getPublicIdc                 S   s   |d ur| j d| d S )NrW   r   setrT   r   r   r   _setPublicId   s    z2getETreeBuilder.<locals>.DocumentType._setPublicIdc                 S   s   | j ddS )NrX   rB   rZ   r#   r   r   r   _getSystemId   s    z2getETreeBuilder.<locals>.DocumentType._getSystemIdc                 S   s   |d ur| j d| d S )NrX   r]   rT   r   r   r   _setSystemId   s    z2getETreeBuilder.<locals>.DocumentType._setSystemIdN)rM   rN   rO   r   r\   r_   rP   rW   r`   ra   rX   r   rY   r   r   DocumentType   s   
rb   c                       s   e Zd Z fddZdS )z!getETreeBuilder.<locals>.Documentc                    s     | d d S )NDOCUMENT_ROOTr   r#   rY   r   r   r      s    z*getETreeBuilder.<locals>.Document.__init__NrM   rN   rO   r   r   rY   r   r   Document   s   rf   c                       s   e Zd Z fddZdS )z)getETreeBuilder.<locals>.DocumentFragmentc                    s     | d d S )NZDOCUMENT_FRAGMENTrd   r#   rY   r   r   r      s    z2getETreeBuilder.<locals>.DocumentFragment.__init__Nre   r   rY   r   r   DocumentFragment   s   rg   c                    s*   g  d fdd	| d d  S )Nr   c                    s  t | ds|  } | jdkrz| ds0| drd| dp<d}| dpJd}d| j||f  nd| jf  n| jdkrd	 | jd urd
d|d  | jf  | jd urtdt | drt| j	rtdnh| j krdd| | jf  n@t
| jts4J dt| j| jf t| j}|d u rR| j}n"| \}}tj| }d||f }dd| |f  t | dr*g }| j	 D ]T\}}	t|}|d ur| \}}tj| }d||f }
n|}
||
|	f qt|D ]&\}}	dd|d  ||	f  q| jrNd
d|d  | jf  |d7 }| D ]}|| qZ| jrd
d|d  | jf  d S )Nr!   rV   rW   rX   rB   z<!DOCTYPE %s "%s" "%s"><!DOCTYPE %s>rc   z	#documentz|%s"%s" r	   Document node cannot have tailr'   $Document node cannot have attributesz|%s<!-- %s -->zExpected unicode, got %s, %sz%s %sz|%s<%s>z
|%s%s="%s")hasattrgetrootr!   r[   r8   r5   rD   	TypeErrorr6   r'   r+   r   rG   
tag_regexpmatchgroupsr   prefixesr*   sorted)r2   indentrW   rX   Znsmatchr   nsprefixr-   r/   Zattr_stringchild)ElementTreeCommentTypervserializeElementr   r   rz      sb    











 zAgetETreeBuilder.<locals>.testSerializer.<locals>.serializeElement
)r   )joinr2   )rx   )ry   rz   r   testSerializer   s    7
z'getETreeBuilder.<locals>.testSerializerc                    s2   g t    fdd|  dS )z4Serialize an element and its child nodes to a stringc                    s  t |  jr|  } | jdkr|| ds2| drf| dp>d}| dpLd}d| j||f  nd| jf  n| jdkr| jd ur| j | jd urtdt	| d	rt
| jrtd
| D ]}| qn| jkrd| jf  n| jsd| jf  n2dfdd| j D }d| j|f  | jrd| j | D ]}| qhd| jf  | jr| j d S )NrV   rW   rX   rB   z<!DOCTYPE %s PUBLIC "%s" "%s">rh   rc   rj   r'   rk   z	<!--%s-->z<%s>ri   c                    s"   g | ]\}}d   ||f qS )z%s="%s")fromXmlName).0r   r/   )filterr   r   
<listcomp>)  s   
zOgetETreeBuilder.<locals>.tostring.<locals>.serializeElement.<locals>.<listcomp>z<%s %s>z</%s>)r+   r   rm   r!   r[   r8   r5   rD   rn   rl   r6   r'   r   r|   r*   )r2   rW   rX   rw   attr)r   rx   r   ry   rz   r   r   rz     sD    





z;getETreeBuilder.<locals>.tostring.<locals>.serializeElementrB   )r
   ZInfosetFilterr|   r}   )r   rx   )r   ry   rz   r   tostring  s
    -z!getETreeBuilder.<locals>.tostringc                       sD   e Zd ZZZZ ZZZfddZ	fddZ
dd ZdS )z$getETreeBuilder.<locals>.TreeBuilderc                    s    |S r    r   rH   )r~   r   r   r~   D  s    z3getETreeBuilder.<locals>.TreeBuilder.testSerializerc                    s<    r| j jS | jd ur*| j jd| j S | j jdS d S )Nz{%s}htmlr   )Zdocumentr   ZdefaultNamespacefindr#   )fullTreer   r   getDocumentG  s    
z0getETreeBuilder.<locals>.TreeBuilder.getDocumentc                 S   s   t j| jS r    )r   TreeBuildergetFragmentr   r#   r   r   r   r   Q  s    z0getETreeBuilder.<locals>.TreeBuilder.getFragmentN)rM   rN   rO   ZdocumentClassZdoctypeClassZelementClassZcommentClassZfragmentClassimplementationr~   r   r   r   )rQ   rf   rg   rb   r   ElementTreeImplementationr   r~   r   r   r   <  s   
r   )rQ   r!   r   rK   r   locals)r   r   r   r   r   )
rQ   rf   rg   rb   r   r   rx   r   r   r~   r   getETreeBuilder   s    >6$r   )F)Z
__future__r   r   r   Zsixr   rer   rB   r   r
   r   r   Z_utilsr   compilero   r   ZgetETreeModuler   r   r   r   <module>   s   

  F