B
    Kkd1                 @   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   	nameTupleparent_childNodes_flags)selfname	namespace)ElementTree H/tmp/pip-install-gxxfd9b7/pip/pip/_vendor/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   r7   refNoder9   r   r   r   insertBeforea   s    z-getETreeBuilder.<locals>.Element.insertBeforec             S   s$   | j | | j|j d |_d S )N)r   remover   r   )r   r7   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   )r4   r   r3   tailr%   r9   )r   datar<   childrenr9   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   r0   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   rA   r3   r   NodereparentChildren)r   	newParentr   r   r   rI      s    z1getETreeBuilder.<locals>.Element.reparentChildren)N)N)__name__
__module____qualname__r   r   r   r    propertyr   r!   r"   r   r$   r.   r+   r/   r1   rG   r5   r8   r<   r>   rD   rF   rI   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   rB   )r   r   r   r      s    z)getETreeBuilder.<locals>.Comment.__init__c             S   s   | j jS )N)r   r3   )r   r   r   r   _getData   s    z)getETreeBuilder.<locals>.Comment._getDatac             S   s   || j _d S )N)r   r3   )r   r-   r   r   r   _setData   s    z)getETreeBuilder.<locals>.Comment._setDataN)rK   rL   rM   r   rP   rQ   rN   rB   r   )r   r   r   rO      s   rO   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   r3   publicIdsystemId)r   r   rR   rS   )r   r   r   r      s    z.getETreeBuilder.<locals>.DocumentType.__init__c             S   s   | j ddS )NrR   r?   )r   get)r   r   r   r   _getPublicId   s    z2getETreeBuilder.<locals>.DocumentType._getPublicIdc             S   s   |d k	r| j d| d S )NrR   )r   r*   )r   r-   r   r   r   _setPublicId   s    z2getETreeBuilder.<locals>.DocumentType._setPublicIdc             S   s   | j ddS )NrS   r?   )r   rT   )r   r   r   r   _getSystemId   s    z2getETreeBuilder.<locals>.DocumentType._getSystemIdc             S   s   |d k	r| j d| d S )NrS   )r   r*   )r   r-   r   r   r   _setSystemId   s    z2getETreeBuilder.<locals>.DocumentType._setSystemIdN)rK   rL   rM   r   rU   rV   rN   rR   rW   rX   rS   r   )r   r   r   DocumentType   s   
rY   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)rK   rL   rM   r   r   )r   r   r   Document   s   r[   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)rK   rL   rM   r   r   )r   r   r   DocumentFragment   s   r\   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>rR   rS   r?   z<!DOCTYPE %s "%s" "%s">z<!DOCTYPE %s>rZ   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   rT   r6   r3   rA   	TypeErrorr4   r#   r(   r   AssertionErrorrE   
tag_regexpmatchgroupsr
   prefixesr'   sorted)r0   indentrR   rS   Znsmatchr   nsprefixr+   r-   Zattr_stringchild)ElementTreeCommentTypervserializeElementr   r   rm      s^    










"
zAgetETreeBuilder.<locals>.testSerializer.<locals>.serializeElement
)r   )join)r0   )rk   )rl   rm   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>rR   rS   r?   z<!DOCTYPE %s PUBLIC "%s" "%s">z<!DOCTYPE %s>rZ   zDocument node cannot have tailr#   z$Document node cannot have attributesz	<!--%s-->z<%s>r]   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   rT   r6   r3   rA   r`   r^   r4   r#   rq   ro   r'   )r0   rR   rS   rj   attr)r   rk   rs   rl   rm   r   r   rm     s@    





z;getETreeBuilder.<locals>.tostring.<locals>.serializeElementr?   )r	   ZInfosetFilterro   )r0   )r   rk   )rs   rl   rm   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   r0   )rp   r   r   rp   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   )documentr   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   r}   N  s    z0getETreeBuilder.<locals>.TreeBuilder.getFragmentN)rK   rL   rM   documentClassdoctypeClasselementClasscommentClassfragmentClassimplementationrp   r{   r}   r   )rO   r[   r\   rY   r   ElementTreeImplementationrz   rp   r   r   r|   9  s   
r|   )rO   r   r   rH   r|   locals)r   rz   rv   r|   r   )
rO   r[   r\   rY   r   r   rk   r   rz   rp   r   getETreeBuilder   s    ~>6$r   )F)
__future__r   r   r   pip._vendor.sixr   rer?   r   r	   r
   r   _utilsr   compilerb   r   getETreeModuler   r   r   r   <module>   s   

  E