o
    %e8                     @   s   d dl mZmZmZ d dlmZ ddlmZmZm	Z	 dZ
eedfeee	d dfhB dfeee	d d	fe	d d
fhB dfee	d dfe	d dfgdfee	d dfe	d dfgdfdZG dd deZG dd deZG dd deZdS )    )absolute_importdivisionunicode_literals)	text_type   )scopingElementstableInsertModeElements
namespacesNFhtmlbuttonolultableoptgroupoptionT)Nr   listr   selectc                   @   sb   e Zd ZdZdd Zdd Zdd Zdd	 ZdddZdd Z	dd Z
dd Zdd Zdd Zd
S )NodezRepresents an item in the treec                 C   s(   || _ d| _d| _i | _g | _g | _dS )zRCreates a Node

        :arg name: The tag name associated with the node

        N)nameparentvalue
attributes
childNodes_flags)selfr    r   M/var/www/html/venv/lib/python3.10/site-packages/html5lib/treebuilders/base.py__init__   s   
zNode.__init__c                 C   s6   d dd | j D }|rd| j|f S d| j S )N c                 S   s   g | ]
\}}d ||f qS )z%s="%s"r   ).0r   r   r   r   r   
<listcomp>.   s    z Node.__str__.<locals>.<listcomp>z<%s %s><%s>)joinr   itemsr   )r   attributesStrr   r   r   __str__-   s   

zNode.__str__c                 C   s
   d| j  S )Nr!   )r   r   r   r   r   __repr__6   s   
zNode.__repr__c                 C      t )z[Insert node as a child of the current node

        :arg node: the node to insert

        NotImplementedErrorr   noder   r   r   appendChild9      zNode.appendChildNc                 C   r(   )aB  Insert data as text in the current node, positioned before the
        start of node insertBefore or to the end of the node's text.

        :arg data: the data to insert

        :arg insertBefore: True if you want to insert the text before the node
            and False if you want to insert it after the node

        r)   )r   datainsertBeforer   r   r   
insertTextA      
zNode.insertTextc                 C   r(   )a  Insert node as a child of the current node, before refNode in the
        list of child nodes. Raises ValueError if refNode is not a child of
        the current node

        :arg node: the node to insert

        :arg refNode: the child node to insert the node before

        r)   )r   r,   refNoder   r   r   r0   M   r2   zNode.insertBeforec                 C   r(   )zhRemove node from the children of the current node

        :arg node: the child node to remove

        r)   r+   r   r   r   removeChildY   r.   zNode.removeChildc                 C   s    | j D ]}|| qg | _ dS )zMove all the children of the current node to newParent.
        This is needed so that trees that don't store text as nodes move the
        text in the correct way

        :arg newParent: the node to move all this node's children to

        N)r   r-   )r   	newParentchildr   r   r   reparentChildrena   s   
	
zNode.reparentChildrenc                 C   r(   )zReturn a shallow copy of the current node i.e. a node with the same
        name and attributes but with no parent or child nodes
        r)   r&   r   r   r   	cloneNoden   s   zNode.cloneNodec                 C   r(   )zFReturn true if the node has children or text, false otherwise
        r)   r&   r   r   r   
hasContentt   s   zNode.hasContentN)__name__
__module____qualname____doc__r   r%   r'   r-   r1   r0   r4   r7   r8   r9   r   r   r   r   r      s    	
r   c                   @   s   e Zd Zdd Zdd ZdS )ActiveFormattingElementsc                 C   sf   d}|t kr+| d d d D ]}|t kr n| ||r|d7 }|dkr*| |  nqt| | d S )Nr         )Marker
nodesEqualremover   append)r   r,   
equalCountelementr   r   r   rF   {   s   
zActiveFormattingElements.appendc                 C   s$   |j |j ksdS |j|jksdS dS )NFT)	nameTupler   )r   node1node2r   r   r   rD      s
   z#ActiveFormattingElements.nodesEqualN)r;   r<   r=   rF   rD   r   r   r   r   r?   z   s    r?   c                   @   s   e Zd ZdZdZdZdZdZdZdd Z	dd 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d Zdd Zdd ZeeeZdd Zdd Zd+dd Zd!d" Zd+d#d$Zd%d& Zd'd( Zd)d* ZdS ),TreeBuildera  Base treebuilder implementation

    * documentClass - the class to use for the bottommost node of a document
    * elementClass - the class to use for HTML Elements
    * commentClass - the class to use for comments
    * doctypeClass - the class to use for doctypes

    Nc                 C   s   |rd| _ nd| _ |   dS )zmCreate a TreeBuilder

        :arg namespaceHTMLElements: whether or not to namespace HTML elements

        zhttp://www.w3.org/1999/xhtmlN)defaultNamespacereset)r   namespaceHTMLElementsr   r   r   r      s   zTreeBuilder.__init__c                 C   s.   g | _ t | _d | _d | _d| _|  | _d S )NF)openElementsr?   activeFormattingElementsheadPointerformPointerinsertFromTabledocumentClassdocumentr&   r   r   r   rN      s   zTreeBuilder.resetc                 C   s   t |d}|st|trtd |f}t|tsJ t| \}}t| jD ]}|r/||kr/ dS |s9|j|kr9 dS ||j|v A rC dS q$J )NrI   r
   TF)	hasattr
isinstancer   r	   tuplelistElementsMapreversedrP   rI   )r   targetvariant	exactNodelistElementsinvertr,   r   r   r   elementInScope   s   

zTreeBuilder.elementInScopec                 C   s   | j sd S t| j d }| j | }|tks|| jv rd S |tkr>|| jvr>|dkr,d}n|d8 }| j | }|tkr>|| jvs%	 |d7 }| j | }| }| d|j|j|jd}|| j |< || j d krgd S q?)NrA   r   r@   TStartTag)typer   	namespacer/   )	rQ   lenrC   rP   r8   insertElementr   rd   r   )r   ientryclonerH   r   r   r   #reconstructActiveFormattingElements   s4   

	

z/TreeBuilder.reconstructActiveFormattingElementsc                 C   s@   | j  }| j r|tkr| j  }| j r|tksd S d S d S d S r:   )rQ   poprC   )r   rh   r   r   r   clearActiveFormattingElements  s   

z)TreeBuilder.clearActiveFormattingElementsc                 C   s:   | j ddd D ]}|tkr dS |j|kr|  S qdS )zCheck if an element exists between the end of the active
        formatting elements and the last marker. If it does, return it, else
        return falseNr@   F)rQ   rC   r   )r   r   itemr   r   r   !elementInActiveFormattingElements  s   
z-TreeBuilder.elementInActiveFormattingElementsc                 C   s&   |  |}| j| | j| d S r:   )createElementrP   rF   rV   r-   )r   tokenrH   r   r   r   
insertRoot  s   
zTreeBuilder.insertRootc                 C   s6   |d }|d }|d }|  |||}| j| d S )Nr   publicIdsystemId)doctypeClassrV   r-   )r   rp   r   rr   rs   doctyper   r   r   insertDoctype   s
   zTreeBuilder.insertDoctypec                 C   s*   |d u r	| j d }|| |d  d S )Nr@   r/   )rP   r-   commentClass)r   rp   r   r   r   r   insertComment(  s   
zTreeBuilder.insertCommentc                 C   s0   |d }| d| j}| ||}|d |_|S )z.Create an element but don't insert it anywherer   rd   r/   )getrM   elementClassr   r   rp   r   rd   rH   r   r   r   ro   -  s
   
zTreeBuilder.createElementc                 C      | j S r:   )_insertFromTabler&   r   r   r   _getInsertFromTable5  s   zTreeBuilder._getInsertFromTablec                 C   s"   || _ |r| j| _dS | j| _dS )zsSwitch the function used to insert an element from the
        normal one to the misnested table one and back againN)r}   insertElementTablerf   insertElementNormal)r   r   r   r   r   _setInsertFromTable8  s   zTreeBuilder._setInsertFromTablec                 C   sb   |d }t |tsJ d| |d| j}| ||}|d |_| jd | | j| |S )Nr   zElement %s not unicoderd   r/   r@   )	rX   r   ry   rM   rz   r   rP   r-   rF   r{   r   r   r   r   C  s   
zTreeBuilder.insertElementNormalc                 C   s`   |  |}| jd jtvr| |S |  \}}|du r"|| n||| | j| |S )z-Create an element and insert it into the treer@   N)	ro   rP   r   r   r   getTableMisnestedNodePositionr-   r0   rF   )r   rp   rH   r   r0   r   r   r   r   M  s   

zTreeBuilder.insertElementTablec                 C   sX   |du r	| j d }| jr| jr| j d jtvr|| dS |  \}}||| dS )zInsert text data.Nr@   )rP   rT   r   r   r1   r   )r   r/   r   r0   r   r   r   r1   ]  s   

zTreeBuilder.insertTextc                 C   s   d}d}d}| j ddd D ]}|jdkr|} nq|r7|jr(|j}|}||fS | j | j |d  }||fS | j d }||fS )zsGet the foster parent element, and sibling to insert before
        (or None) when inserting a misnested table nodeNr@   r   rA   r   )rP   r   r   index)r   	lastTablefosterParentr0   elmr   r   r   r   l  s&   

z)TreeBuilder.getTableMisnestedNodePositionc                 C   s@   | j d j}|tdv r||kr| j   | | d S d S d S )Nr@   )dddtlir   r   prprt)rP   r   	frozensetrk   generateImpliedEndTags)r   excluder   r   r   r   r     s   
z"TreeBuilder.generateImpliedEndTagsc                 C   r|   )zReturn the final tree)rV   r&   r   r   r   getDocument  s   zTreeBuilder.getDocumentc                 C   s   |   }| jd | |S )zReturn the final fragmentr   )fragmentClassrP   r7   )r   fragmentr   r   r   getFragment  s   zTreeBuilder.getFragmentc                 C   r(   )zSerialize the subtree of node in the format required by unit tests

        :arg node: the node from which to start serializing

        r)   r+   r   r   r   testSerializer  r.   zTreeBuilder.testSerializerr:   )r;   r<   r=   r>   rU   rz   rw   rt   r   r   rN   ra   rj   rl   rn   rq   rv   rx   ro   r~   r   propertyrT   r   r   r1   r   r   r   r   r   r   r   r   r   rL      s8    
.

	



rL   )
__future__r   r   r   sixr   	constantsr   r   r	   rC   r   rZ   objectr   r   r?   rL   r   r   r   r   <module>   s2    



c