3
�Pfv6 � @ s� d dl mZmZmZ d dlmZ ddlmZmZm Z dZ
e e�dfe ee e d dfg�B �dfe ee e d d fe d d
fg�B �dfe e d dfe d d fg�dfe e d d fe d d
fg�dfd�Z
G dd� de�ZG dd� de�ZG dd� de�ZdS )� )�absolute_import�division�unicode_literals)� text_type� )�scopingElements�tableInsertModeElements�
namespacesNF�html�buttonZolZul�table�optgroup�optionT)Nr �listr Zselectc @ s^ e Zd Zdd� Zdd� Zdd� Zdd� Zdd
d �Zd d
� Zdd� Z dd� Z
dd� Z dd� Z d S )�Nodec C s( || _ d| _d| _i | _g | _g | _dS )a6 Node representing an item in the tree.
name - The tag name associated with the node
parent - The parent of the current node (or None for the document node)
value - The value of the current node (applies to text nodes and
comments
attributes - a dict holding name, value pairs for attributes of the node
childNodes - a list of child nodes of the current node. This must
include all elements but not necessarily other node types
_flags - A list of miscellaneous flags that can be set on the node
N)�name�parent�value�
attributes�
childNodesZ_flags)�selfr � r �/usr/lib/python3.6/base.py�__init__ s z
Node.__init__c C s: dj dd� | jj� D ��}|r,d| j|f S d| j S d S )N� c S s g | ]\}}d ||f �qS )z%s="%s"r )�.0r r r r r �
+ s z Node.__str__..z<%s %s>z<%s>)�joinr �itemsr )r Z
attributesStrr r r �__str__* s
z Node.__str__c C s
d| j S )Nz<%s>)r )r r r r �__repr__3 s z
Node.__repr__c C s t �dS )z3Insert node as a child of the current node
N)�NotImplementedError)r �noder r r � appendChild6 s zNode.appendChildNc C s t �dS )z�Insert data as text in the current node, positioned before the
start of node insertBefore or to the end of the node's text.
N)r! )r �data� insertBeforer r r �
insertText; s zNode.insertTextc C s t �dS )z�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 nodeN)r! )r r" ZrefNoder r r r% A s zNode.insertBeforec C s t �dS )z:Remove node from the children of the current node
N)r! )r r" r r r � removeChildG s zNode.removeChildc C s$ x| j D ]}|j|� qW g | _ dS )z�Move 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
N)r r# )r Z newParentZchildr r r �reparentChildrenL s zNode.reparentChildrenc C s t �dS )z�Return a shallow copy of the current node i.e. a node with the same
name and attributes but with no parent or child nodes
N)r! )r r r r � cloneNodeV s zNode.cloneNodec C s t �dS )zFReturn true if the node has children or text, false otherwise
N)r! )r r r r �
hasContent\ s zNode.hasContent)N)
�__name__�
__module__� __qualname__r r r r# r& r% r' r( r) r* r r r r r s
r c @ s e Zd Zdd� Zdd� ZdS )�ActiveFormattingElementsc C sf d}|t krVxH| d d d� D ]6}|t kr*P | j||�r>|d7 }|dkr | j|� P q W tj| |� d S )Nr � � �����)�Marker�
nodesEqual�remover �append)r r" Z
equalCount�elementr r r r5 c s
z ActiveFormattingElements.appendc C s$ |j |j ksdS |j|jks dS dS )NFT)� nameTupler )r Znode1Znode2r r r r3 p s
z#ActiveFormattingElements.nodesEqualN)r+ r, r- r5 r3 r r r r r. b s
r. c @ s� e Zd ZdZdZdZdZdZdZdd� Z dd� Z
d+dd�Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zd,dd�Zdd� Zdd� Zdd� Zeee�Zdd � Zd d � Zd-d d �Zd!d"� Zd.d#d$�Zd%d&� Z d'd(� Z d)d*� Z dS )/� 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| _ nd | _ | j� d S )Nz http://www.w3.org/1999/xhtml)�defaultNamespace�reset)r ZnamespaceHTMLElementsr r r r � s zTreeBuilder.__init__c C s. g | _ t� | _d | _d | _d| _| j� | _d S )NF)� openElementsr. �activeFormattingElementsZ headPointerZ formPointer�insertFromTable�
documentClass�document)r r r r r: � s zTreeBuilder.resetc C s~ t |d�}|s$t|t�r$td |f}t| \}}xHt| j�D ]:}|rP||krPdS | rd|j|krddS ||j|kA r s
zTreeBuilder.insertTextc C sv d}d}d}x(| j ddd� D ]}|jdkr |}P q W |rd|jrL|j}|}qn| j | j j|�d }n
| j d }||fS )zsGet the foster parent element, and sibling to insert before
(or None) when inserting a misnested table nodeNr/ r r r1 )r; r r �index)r Z lastTableZ fosterParentr% Zelmr r r rb M s
z)TreeBuilder.getTableMisnestedNodePositionc
C s8 | j d
j}|td �kr4||kr4| j j� | j|� d S ) Nr/ �dd�dt�lir r
�p�rp�rtr1 )rd re rf r r
rg rh ri )r; r � frozensetrN �generateImpliedEndTags)r �excluder r r r rk g s
z"TreeBuilder.generateImpliedEndTagsc C s | j S )zReturn the final tree)r? )r r r r � getDocumentq s zTreeBuilder.getDocumentc C s | j � }| jd j|� |S )zReturn the final fragmentr )�
fragmentClassr; r( )r Zfragmentr r r � getFragmentu s zTreeBuilder.getFragmentc C s t �dS )zzSerialize the subtree of node in the format required by unit tests
node - the node from which to start serializingN)r! )r r" r r r �testSerializer| s zTreeBuilder.testSerializer)N)N)N)N) r+ r, r- �__doc__r> r\ rY rW rn r r: rF rM rO rQ rT rX rZ rR r^ ra �propertyr= r` r_ r& rb rk rm ro rp r r r r r8 z s6
.
r8 )Z
__future__r r r Zpip._vendor.sixr Z constantsr r r r2 rj �setrB �objectr r r. r8 r r r r � s
K