B
    @)H[˜8  ã               @   sØ   d Z ddlmZmZmZ ddlZddl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 dZe d¡Ze d¡jZG dd„ deƒZG dd„ deƒZdd„ Zdd„ ZG dd„ de	jƒZdS )aÃ  Module for supporting the lxml.etree library. The idea here is to use as much
of the native library as possible, without using fragile hacks like custom element
names that break between releases. The downside of this is that we cannot represent
all possible trees; specifically the following are known to cause problems:

Text or comments as siblings of the root element
Docypes with no name

When any of these things occur, we emit a DataLossWarning
é    )Úabsolute_importÚdivisionÚunicode_literalsNé   )Úbaseé   )ÚDataLossWarning)Ú	constants)Úetree)Ú	_ihatexmlTz{([^}]*)}(.*)Úasdc               @   s   e Zd Zdd„ ZdS )ÚDocumentTypec             C   s   || _ || _|| _d S )N)ÚnameÚpublicIdÚsystemId)Úselfr   r   r   © r   úmC:\Users\HIRONO~1\AppData\Local\Temp\pip-install-qz7kpjmn\pip\pip\_vendor\html5lib\treebuilders\etree_lxml.pyÚ__init__#   s    zDocumentType.__init__N)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   "   s   r   c               @   s,   e Zd Zdd„ Zdd„ Zdd„ ZeeƒZdS )ÚDocumentc             C   s   d | _ g | _d S )N)Ú_elementTreeÚ_childNodes)r   r   r   r   r   *   s    zDocument.__init__c             C   s   | j  ¡  |j¡ d S )N)r   ÚgetrootZaddnextÚ_element)r   Úelementr   r   r   ÚappendChild.   s    zDocument.appendChildc             C   s   | j S )N)r   )r   r   r   r   Ú_getChildNodes1   s    zDocument._getChildNodesN)r   r   r   r   r   r   ÚpropertyÚ
childNodesr   r   r   r   r   )   s   r   c                s6   g ‰t jdd�‰ d‡ ‡‡fdd„	‰ˆ| dƒ d ˆ¡S )NT)ÚpreventDoubleDashCommentsr   c                sD  t | dƒ�s8t | dƒrÂˆ d¡ | jjrz| jjsF| jjsFd| jj }nd| jj| jj| jjf }ˆ dd|d  |f ¡ |  ¡ }x| ¡ d k	rš| ¡ }q„W x–|d k	r¾ˆ||d ƒ | 	¡ }qžW nrt
| tƒsØt
| tƒ�rt
| tƒsôtjd	 dksôt‚ˆ d
d| | f ¡ n(ˆ d¡ x| D ]}ˆ||d ƒ �qW �n| jtk�rŒˆ dd| | jf ¡ t | dƒ�r@| j�r@ˆ d
d| | jf ¡ �n´t
| tjƒ�sžt‚tj | j¡}|d k	�rô| d¡}| d¡}tj| }ˆ dd| |ˆ  |¡f ¡ nˆ dd| ˆ  | j¡f ¡ t | dƒ�rÊg }xr| j ¡ D ]d\}	}
t |	¡}|d k	�rx| ¡ \}}	ˆ  |	¡}	tj| }d||	f }n
ˆ  |	¡}| ||
f¡ �q.W x2t |ƒD ]&\}	}
ˆ dd|d  |	|
f ¡ �q W | j�rîˆ d
d|d  | jf ¡ |d7 }x| D ]}ˆ||ƒ �qüW t | dƒ�r@| j�r@ˆ d
d|d  | jf ¡ d S )NÚtagr   z	#documentz<!DOCTYPE %s>z<!DOCTYPE %s "%s" "%s">z|%s%sú r   r   z|%s"%s"z#document-fragmentz|%s<!-- %s -->Útailr   z
|%s<%s %s>z|%s<%s>Úattribz%s %sz
|%s%s="%s")!ÚhasattrÚappendÚdocinfoÚinternalDTDÚ	public_idZ
system_urlÚ	root_namer   ZgetpreviousZgetnextÚ
isinstanceÚstrÚbytesÚsysÚversion_infoÚAssertionErrorr#   Úcomment_typeÚtextr%   r
   Z_ElementÚetree_buildersÚ
tag_regexpÚmatchÚgroupr	   ÚprefixesÚfromXmlNamer&   ÚitemsÚgroupsÚsorted)r   ÚindentÚdtd_strZnext_elementÚnsmatchÚnsr#   ÚprefixÚ
attributesr   ÚvalueÚattr_stringÚchild)ÚinfosetFilterÚrvÚserializeElementr   r   rI   ;   st    













"
z(testSerializer.<locals>.serializeElementÚ
)r   )r   ÚInfosetFilterÚjoin)r   r   )rG   rH   rI   r   ÚtestSerializer7   s
    F
rM   c                s$   g ‰ ‡ ‡fdd„‰ˆ| ƒ d  ˆ ¡S )z4Serialize an element and its child nodes to a stringc                s  t | dƒsH| jjr:| jjr$| jj}nd| jj }ˆ  |¡ ˆ|  ¡ ƒ n | jtkrfˆ  d| j	f ¡ n‚| j
s€ˆ  d| jf ¡ n.d dd„ | j
 ¡ D ƒ¡}ˆ  d| j|f ¡ | j	rÀˆ  | j	¡ x| D ]}ˆ|ƒ qÆW ˆ  d	| jf ¡ t | d
ƒ�r| j�rˆ  | j¡ d S )Nr#   z<!DOCTYPE %s>z	<!--%s-->z<%s>r$   c             S   s   g | ]\}}d ||f ‘qS )z%s="%s"r   )Ú.0r   rD   r   r   r   ú
<listcomp>œ   s   z6tostring.<locals>.serializeElement.<locals>.<listcomp>z<%s %s>z</%s>r%   )r'   r)   r*   Údoctyper,   r(   r   r#   r3   r4   r&   rL   r;   r%   )r   r?   ÚattrrF   )rH   rI   r   r   rI   Š   s*    





z"tostring.<locals>.serializeElementÚ )rL   )r   r   )rH   rI   r   Útostring†   s     rS   c                   sz   e Zd ZeZeZdZdZeZ	e
Zddd„Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Zddd„Zd‡ fdd„	Zdd„ Z‡  ZS )ÚTreeBuilderNFc                sŠ   t jt|d�‰tjdd� ‰| _|| _G ‡fdd„dtƒ‰ G ‡ ‡‡fdd„dˆjƒ}G ‡‡fdd	„d	ˆj	ƒ}|| _
|| _tj | |¡ d S )
N)ÚfullTreeT)r"   c                   s&   e Zd Zd‡ fdd„	Z‡ fdd„ZdS )z(TreeBuilder.__init__.<locals>.AttributesNc                sv   |d kri }|| _ t | |¡ xR|  ¡ D ]F\}}t|tƒrVd|d ˆ  |d ¡f }n
ˆ  |¡}|| j j j|< q(W d S )Nz{%s}%sr   r   )r   Údictr   r;   r-   ÚtupleÚcoerceAttributer&   )r   r   rD   Úkeyr   )rG   r   r   r   ½   s    

z1TreeBuilder.__init__.<locals>.Attributes.__init__c                sP   t  | ||¡ t|tƒr4d|d ˆ  |d ¡f }n
ˆ  |¡}|| jjj|< d S )Nz{%s}%sr   r   )rV   Ú__setitem__r-   rW   rX   r   r&   )r   rY   rD   r   )rG   r   r   rZ   É   s
    

z4TreeBuilder.__init__.<locals>.Attributes.__setitem__)N)r   r   r   r   rZ   r   )rG   r   r   Ú
Attributes¼   s   r[   c                   sx   e Zd Z‡ ‡‡fdd„Z‡fdd„Z‡fdd„ZeeeƒZdd„ Z‡ fd	d
„Z	eee	ƒZ
d‡‡fdd„	Z‡fdd„ZdS )z%TreeBuilder.__init__.<locals>.Elementc                s*   ˆ  |¡}ˆjj| ||d� ˆ | ƒ| _d S )N)Ú	namespace)ÚcoerceElementÚElementr   Ú_attributes)r   r   r\   )r[   ÚbuilderrG   r   r   r   Ò   s    
z.TreeBuilder.__init__.<locals>.Element.__init__c                s$   ˆ   |¡| _|  | j| j¡| j_d S )N)r]   Ú_nameÚ_getETreeTagÚ
_namespacer   r#   )r   r   )rG   r   r   Ú_setName×   s    z.TreeBuilder.__init__.<locals>.Element._setNamec                s   ˆ   | j¡S )N)r:   ra   )r   )rG   r   r   Ú_getNameÜ   s    z.TreeBuilder.__init__.<locals>.Element._getNamec             S   s   | j S )N)r_   )r   r   r   r   Ú_getAttributesá   s    z4TreeBuilder.__init__.<locals>.Element._getAttributesc                s   ˆ | |ƒ| _ d S )N)r_   )r   rC   )r[   r   r   Ú_setAttributesä   s    z4TreeBuilder.__init__.<locals>.Element._setAttributesNc                s   ˆ  |¡}ˆ j | ||¡ d S )N)ÚcoerceCharactersr^   Ú
insertText)r   ÚdataÚinsertBefore)r`   rG   r   r   ri   é   s    
z0TreeBuilder.__init__.<locals>.Element.insertTextc                s   ˆ j  | |¡ d S )N)r^   r   )r   rF   )r`   r   r   r   í   s    z1TreeBuilder.__init__.<locals>.Element.appendChild)N)r   r   r   r   rd   re   r    r   rf   rg   rC   ri   r   r   )r[   r`   rG   r   r   r^   Ñ   s   

r^   c                   s8   e Zd Z‡ ‡fdd„Z‡fdd„Zdd„ ZeeeƒZdS )z%TreeBuilder.__init__.<locals>.Commentc                s   ˆ  |¡}ˆ j | |¡ d S )N)ÚcoerceCommentÚCommentr   )r   rj   )r`   rG   r   r   r   ñ   s    
z.TreeBuilder.__init__.<locals>.Comment.__init__c                s   ˆ   |¡}|| j_d S )N)rl   r   r4   )r   rj   )rG   r   r   Ú_setDataõ   s    
z.TreeBuilder.__init__.<locals>.Comment._setDatac             S   s   | j jS )N)r   r4   )r   r   r   r   Ú_getDataù   s    z.TreeBuilder.__init__.<locals>.Comment._getDataN)r   r   r   r   rn   ro   r    rj   r   )r`   rG   r   r   rm   ð   s   rm   )r5   ÚgetETreeModuler
   r   rK   rG   ÚnamespaceHTMLElementsrV   r^   rm   ÚelementClassÚcommentClassr   rT   r   )r   rq   rU   r^   rm   r   )r[   r`   rG   r   r   ·   s    zTreeBuilder.__init__c             C   s$   t j | ¡ | j| _g | _d | _d S )N)r   rT   ÚresetÚinsertCommentInitialÚinsertCommentÚinitial_commentsrP   )r   r   r   r   rt     s    zTreeBuilder.resetc             C   s   t |ƒS )N)rM   )r   r   r   r   r   rM   	  s    zTreeBuilder.testSerializerc             C   s   t r| jjS | jj ¡ S d S )N)rU   Údocumentr   r   )r   r   r   r   ÚgetDocument  s    zTreeBuilder.getDocumentc             C   sF   g }| j d j}|jr"| |j¡ | t|ƒ¡ |jrB| |j¡ |S )Nr   )ÚopenElementsr   r4   r(   ÚextendÚlistr%   )r   Úfragmentr   r   r   r   ÚgetFragment  s    zTreeBuilder.getFragmentc             C   sh   |d }|d }|d }|s0t  dt¡ d | _n4| j |¡}||krPt  dt¡ |  |||¡}|| _d S )Nr   r   r   z#lxml cannot represent empty doctypez%lxml cannot represent non-xml doctype)ÚwarningsÚwarnr   rP   rG   r]   ÚdoctypeClass)r   Útokenr   r   r   ZcoercedNamerP   r   r   r   ÚinsertDoctype  s    zTreeBuilder.insertDoctypec             C   s6   |d ks|| j kst‚| j jd ks&t‚| j |¡ d S )N)rx   r2   r   rw   r(   )r   rj   Úparentr   r   r   ru   ,  s    z TreeBuilder.insertCommentInitialc                sB   || j kr,| j j ¡ d jtkr,t dt¡ tt	| ƒ 
||¡ d S )Néÿÿÿÿz@lxml cannot represent adjacent comments beyond the root elements)rx   r   r   r#   r3   r   r€   r   ÚsuperrT   rv   )r   rj   r„   )Ú	__class__r   r   ÚinsertCommentMain1  s    
zTreeBuilder.insertCommentMainc             C   s¸  d}| j rô| j jst‚|d| j j 7 }| j jd k	s>| j jd k	rÐ|d| j | j jpRd¡ 7 }| j jrÈ| j j}| d¡dkrž| d¡dkržt 	dt
¡ | dd¡}| d¡dkrº|d	| 7 }qÐ|d
| 7 }n|d7 }|d7 }| j j|d krôt 	dt
¡ |d7 }t |¡}x*| jD ] }|  |d ¡}| |j¡ �qW |  ¡ | _| ¡ | j_|d }| d| j¡}|d k�rn|}	nd||f }	|	|_|  ||¡}
||
_| jj |
¡ | j |
¡ | j| _d S )NrR   z<!DOCTYPE %sz PUBLIC "%s" ú'r   ú"z6DOCTYPE system cannot contain single and double quotesZU00027z"%s"z'%s'z''ú>r   zGlxml cannot represent doctype with a different name to the root elementz$<THIS_SHOULD_NEVER_APPEAR_PUBLICLY/>rj   r\   z{%s}%s)rP   r   r2   r   r   rG   ÚcoercePubidÚfindr   r€   r   Úreplacer
   Ú
fromstringrw   rs   Zaddpreviousr   ÚdocumentClassrx   Zgetroottreer   ÚgetÚdefaultNamespacer#   rr   r   r(   rz   rˆ   rv   )r   r‚   ZdocStrÚsysidÚrootÚcomment_tokenÚcommentr   r\   Ú	etree_tagZroot_elementr   r   r   Ú
insertRoot7  sL    


zTreeBuilder.insertRoot)F)N)N)r   r   r   r   r�   r   r�   rr   rs   ÚfragmentClassr
   Úimplementationr   rt   rM   ry   r~   rƒ   ru   rˆ   r˜   Ú__classcell__r   r   )r‡   r   rT   ¯   s   
L

rT   )Ú__doc__Ú
__future__r   r   r   r   Úrer0   rR   r   r	   r   r
   r5   r   Z
lxml.etreerU   Úcompiler6   rm   r#   r3   Úobjectr   r   rM   rS   rT   r   r   r   r   Ú<module>
   s$   
O)