B
    n&=[1                 @   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 edZdddZeeZdS )    )absolute_importdivisionunicode_literals)	text_typeN   )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 k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 :lib/python3.7/site-packages/html5lib/treebuilders/etree.py__init__   s    

z)getETreeBuilder.<locals>.Element.__init__c             S   s   |d kr|}nd||f }|S )Nz{%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 )N)r   )r   r   r   r   _getName.   s    z)getETreeBuilder.<locals>.Element._getNamec             S   s   || _ | | j| j | j_d S )N)r   r   r   r   r   )r   r   r   r   r   _setNamespace3   s    z.getETreeBuilder.<locals>.Element._setNamespacec             S   s   | j S )N)r   )r   r   r   r   _getNamespace7   s    z.getETreeBuilder.<locals>.Element._getNamespacec             S   s   | j jS )N)r   attrib)r   r   r   r   _getAttributes<   s    z/getETreeBuilder.<locals>.Element._getAttributesc             S   sp   x"t | jj D ]}| jj|= qW xF| D ]:\}}t|trVd|d |d f }n|}| j|| q.W d S )Nz{%s}%sr   r   )listr   r"   keysitems
isinstancetupleset)r   
attributeskeyvaluer   r   r   r   _setAttributes?   s    
z/getETreeBuilder.<locals>.Element._setAttributesc             S   s   | j S )N)r   )r   r   r   r   _getChildNodesM   s    z/getETreeBuilder.<locals>.Element._getChildNodesc             S   s.   | j d d = g | _x|D ]}| | qW d S )N)r   r   ZinsertChild)r   r,   elementr   r   r   _setChildNodesP   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len)r   r   r   r   
hasContentX   s    z+getETreeBuilder.<locals>.Element.hasContentc             S   s$   | j | | j|j | |_d S )N)r   appendr   r   )r   noder   r   r   appendChild\   s    z,getETreeBuilder.<locals>.Element.appendChildc             S   s,   t | j|j}| j||j | |_d S )N)r$   r   indexinsertr   )r   r6   ZrefNoder8   r   r   r   insertBeforea   s    z-getETreeBuilder.<locals>.Element.insertBeforec             S   s$   | j | | j|j d |_d S )N)r   remover   r   )r   r6   r   r   r   removeChildf   s    z,getETreeBuilder.<locals>.Element.removeChildc             S   s   t | js,| jjsd| j_| j j|7  _n|d k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   )r3   r   r2   tailr$   r8   )r   datar:   Zchildrenr8   r   r   r   
insertTextk   s"    

z+getETreeBuilder.<locals>.Element.insertTextc             S   s8   t | | j| j}x | j D ]\}}||j|< qW |S )N)typer   r   r*   r&   )r   r/   r   r,   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 k	rR|j j| jj7  _d| j_tj| | d S )Nr>   r=   )
childNodesr   r?   r2   r   NodereparentChildren)r   Z	newParentr   r   r   rF      s    z1getETreeBuilder.<locals>.Element.reparentChildren)N)N)__name__
__module____qualname__r   r   r   r   propertyr   r    r!   r   r#   r-   r*   r.   r0   rD   r4   r7   r:   r<   rA   rC   rF   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 )N)Commentr   r   r   r   )r   r@   )r   r   r   r      s    z)getETreeBuilder.<locals>.Comment.__init__c             S   s   | j jS )N)r   r2   )r   r   r   r   _getData   s    z)getETreeBuilder.<locals>.Comment._getDatac             S   s   || j _d S )N)r   r2   )r   r,   r   r   r   _setData   s    z)getETreeBuilder.<locals>.Comment._setDataN)rG   rH   rI   r   rL   rM   rJ   r@   r   )r   r   r   rK      s   rK   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 )Nz
<!DOCTYPE>)r   r   r2   publicIdsystemId)r   r   rN   rO   )r   r   r   r      s    z.getETreeBuilder.<locals>.DocumentType.__init__c             S   s   | j ddS )NrN   r=   )r   get)r   r   r   r   _getPublicId   s    z2getETreeBuilder.<locals>.DocumentType._getPublicIdc             S   s   |d k	r| j d| d S )NrN   )r   r)   )r   r,   r   r   r   _setPublicId   s    z2getETreeBuilder.<locals>.DocumentType._setPublicIdc             S   s   | j ddS )NrO   r=   )r   rP   )r   r   r   r   _getSystemId   s    z2getETreeBuilder.<locals>.DocumentType._getSystemIdc             S   s   |d k	r| j d| d S )NrO   )r   r)   )r   r,   r   r   r   _setSystemId   s    z2getETreeBuilder.<locals>.DocumentType._setSystemIdN)rG   rH   rI   r   rQ   rR   rJ   rN   rS   rT   rO   r   )r   r   r   DocumentType   s   
rU   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   )r   r   r   r      s    z*getETreeBuilder.<locals>.Document.__init__N)rG   rH   rI   r   r   )r   r   r   Document   s   rW   c                   s   e Zd Z fddZdS )z)getETreeBuilder.<locals>.DocumentFragmentc                s     | d d S )NZDOCUMENT_FRAGMENT)r   )r   )r   r   r   r      s    z2getETreeBuilder.<locals>.DocumentFragment.__init__N)rG   rH   rI   r   r   )r   r   r   DocumentFragment   s   rX   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 k	rd
d|d  | jf  | jd k	rtdt | drt| j	rtdnp| j krdd| | jf  nHt
| jts4tdt| j| jf t| j}|d krR| j}n"| \}}tj| }d||f }dd| |f  t | dr2g }xb| j	 D ]T\}}	t|}|d k	r| \}}tj| }d||f }
n|}
||
|	f qW x2t|D ]&\}}	dd|d  ||	f  qW | jrVd
d|d  | jf  |d7 }x| D ]}|| qdW | jrd
d|d  | jf  d S )Nr   z
<!DOCTYPE>rN   rO   r=   z<!DOCTYPE %s "%s" "%s">z<!DOCTYPE %s>rV   z	#documentz|%s"%s" r   zDocument node cannot have tailr"   z$Document node cannot have attributesz|%s<!-- %s -->zExpected unicode, got %s, %sz%s %sz|%s<%s>z
|%s%s="%s")hasattrgetrootr   rP   r5   r2   r?   	TypeErrorr3   r"   r'   r   AssertionErrorrB   
tag_regexpmatchgroupsr
   prefixesr&   sorted)r/   indentrN   rO   Znsmatchr   nsprefixr*   r,   Zattr_stringchild)ElementTreeCommentTypervserializeElementr   r   ri      s^    










"
zAgetETreeBuilder.<locals>.testSerializer.<locals>.serializeElement
)r   )join)r/   )rg   )rh   ri   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 k	r| j | jd k	rtdt	| d	rt
| jrtd
x| D ]}| qW n| jkrd| jf  n| js$d| jf  n2dfdd| j D }d| j|f  | jrj| j x| D ]}| qpW d| jf  | jr| j d S )Nz
<!DOCTYPE>rN   rO   r=   z<!DOCTYPE %s PUBLIC "%s" "%s">z<!DOCTYPE %s>rV   zDocument node cannot have tailr"   z$Document node cannot have attributesz	<!--%s-->z<%s>rY   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   r[   r   rP   r5   r2   r?   r\   rZ   r3   r"   rm   rk   r&   )r/   rN   rO   rf   attr)r   rg   ro   rh   ri   r   r   ri     s@    





z;getETreeBuilder.<locals>.tostring.<locals>.serializeElementr=   )r	   ZInfosetFilterrk   )r/   )r   rg   )ro   rh   ri   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 )Nr   )r   r/   )rl   r   r   rl   A  s    z3getETreeBuilder.<locals>.TreeBuilder.testSerializerc                s<    r| j jS | jd k	r*| j jd| j S | j jdS d S )Nz{%s}htmlr   )Zdocumentr   ZdefaultNamespacefind)r   )fullTreer   r   getDocumentD  s    
z0getETreeBuilder.<locals>.TreeBuilder.getDocumentc             S   s   t j| jS )N)r   TreeBuildergetFragmentr   )r   r   r   r   rw   N  s    z0getETreeBuilder.<locals>.TreeBuilder.getFragmentN)rG   rH   rI   ZdocumentClassZdoctypeClassZelementClassZcommentClassZfragmentClassimplementationrl   ru   rw   r   )rK   rW   rX   rU   r   ElementTreeImplementationrt   rl   r   r   rv   9  s   
rv   )rK   r   r   rE   rv   locals)ry   rt   rr   rv   r   )
rK   rW   rX   rU   r   r   rg   ry   rt   rl   r   getETreeBuilder   s    ~>6$r{   )F)Z
__future__r   r   r   Zsixr   rer=   r   r	   r
   r   Z_utilsr   compiler^   r{   ZgetETreeModuler   r   r   r   <module>   s   

  E