a
    ¢Ž$iž  ã                   @   sV   d Z ddlmZmZ ddlmZmZ G dd„ dƒZG dd„ dƒZG dd	„ d	ej	ƒZ	d
S )zù
Legacy module - don't use in new code!

html5lib now has its own proper implementation.

This module implements a tree builder for html5lib that generates lxml
html element trees.  This module uses camelCase as it follows the
html5lib style guide.
é    )Ú_baseÚetree)Úhtmlr   c                   @   s   e Zd Zdd„ ZdS )ÚDocumentTypec                 C   s   || _ || _|| _d S ©N)ÚnameÚpublicIdÚsystemId)Úselfr   r   r	   © r   úe/var/www/staging/api/virtual_environments/venv/lib/python3.9/site-packages/lxml/html/_html5builder.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S )ÚDocumentc                 C   s   d | _ g | _d S r   )Ú_elementTreeÚ
childNodes©r
   r   r   r   r      s    zDocument.__init__c                 C   s   | j  ¡  |j¡ d S r   )r   ZgetrootZaddnextÚ_element)r
   Úelementr   r   r   ÚappendChild   s    zDocument.appendChildN)r   r   r   r   r   r   r   r   r   r      s   r   c                   @   sZ   e Zd ZeZeZdZdZeZ	dd„ Z
dd„ Zdd„ Zdd	„ Zd
d„ Zddd„Zdd„ ZdS )ÚTreeBuilderNc                 O   sJ   t jtdd}t jtdd}|j| _|j| _tj	j
| g|¢R i |¤Ž d S )NF)ZfullTree)Úetree_buildersZgetETreeModuler   r   ZElementÚelementClassÚCommentÚcommentClassr   r   r   )r
   ÚargsÚkwargsZhtml_builderZetree_builderr   r   r   r   '   s
    zTreeBuilder.__init__c                 C   s"   t j | ¡ d| _g | _d | _d S )NF)r   r   ÚresetÚrootInsertedÚinitialCommentsÚdoctyper   r   r   r   r   .   s    zTreeBuilder.resetc                 C   s   | j jS r   )Údocumentr   r   r   r   r   ÚgetDocument4   s    zTreeBuilder.getDocumentc                 C   sF   g }| j d j}|jr"| |j¡ | | ¡ ¡ |jrB| |j¡ |S )Nr   )ÚopenElementsr   ÚtextÚappendÚextendZgetchildrenÚtail)r
   Úfragmentr   r   r   r   ÚgetFragment7   s    zTreeBuilder.getFragmentc                 C   s   |   |||¡}|| _d S r   )ÚdoctypeClassr"   )r
   r   r   r	   r"   r   r   r   ÚinsertDoctypeA   s    zTreeBuilder.insertDoctypec                 C   s(   | j s| j |¡ ntj | ||¡ d S r   )r    r!   r'   r   r   ÚinsertComment)r
   ÚdataÚparentr   r   r   r.   E   s    zTreeBuilder.insertCommentc                 C   sà   g }| j r`| j jr`| d| j j ¡ | j jd us<| j jd urV| d| j j| j jf ¡ | d¡ | d¡ t d |¡¡}| jD ]}| 	t
 |¡¡ q€|  ¡ | _| ¡ | j_|  |¡}||_| jj |¡ | j |¡ d| _d S )Nz<!DOCTYPE %sz PUBLIC "%s" "%s"ú>z<html></html>Ú T)r"   r   r'   r   r	   r   Z
fromstringÚjoinr!   Zaddpreviousr   r   ÚdocumentClassr#   Zgetroottreer   r   r   r   r%   r    )r
   r   ÚbufÚrootÚcommentZroot_elementr   r   r   Ú
insertRootK   s&    ÿ




zTreeBuilder.insertRoot)N)r   r   r   r   r4   r   r,   r   r   ZfragmentClassr   r   r$   r+   r-   r.   r8   r   r   r   r   r       s   

r   N)
Ú__doc__Zhtml5lib.treebuildersr   r   r   Zlxmlr   r   r   r   r   r   r   r   Ú<module>   s
   

