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�}|s2t|t�r$td |f}t|t�s2t�t| \}}xHt| j�D ]:}|r^||kr^dS | rr|j |krrdS ||j |kA rJdS qJW ds�t�d S )Nr7 r
TF)
�hasattr�
isinstancer r �tuple�AssertionError�listElementsMap�reversedr; r7 )r �targetZvariantZ exactNodeZ listElements�invertr" r r r �elementInScope� s
zTreeBuilder.elementInScopec C s� | j s
d S t| j �d }| j | }|tks4|| jkr8d S x6|tkrn|| jkrn|dkrZd}P |d8 }| j | }q:W xR|d7 }| j | }|j� }| jd|j|j|jd��}|| j |< || j d krrP qrW d S )Nr/ r ZStartTag)�typer � namespacer$ r1 r1 ) r< �lenr2 r; r) �
insertElementr rJ r )r �i�entryZcloner6 r r r �#reconstructActiveFormattingElements� s.
z/TreeBuilder.reconstructActiveFormattingElementsc C s, | j j� }x | j r&|tkr&| j j� }q W d S )N)r< �popr2 )r rN r r r � clearActiveFormattingElements� s
z)TreeBuilder.clearActiveFormattingElementsc C s8 x2| j ddd� D ] }|tkr"P q|j|kr|S qW dS )z�Check if an element exists between the end of the active
formatting elements and the last marker. If it does, return it, else
return falseNr/ Fr1 )r< r2 r )r r �itemr r r �!elementInActiveFormattingElements� s
z-TreeBuilder.elementInActiveFormattingElementsc C s& | j |�}| jj|� | jj|� d S )N)�
createElementr; r5 r? r# )r �tokenr6 r r r �
insertRoot� s
zTreeBuilder.insertRootc C s6 |d }|d }|d }| j |||�}| jj|� d S )Nr �publicId�systemId)� doctypeClassr? r# )r rU r rW rX Zdoctyper r r �
insertDoctype s
zTreeBuilder.insertDoctypec C s* |d kr| j d }|j| j|d �� d S )Nr/ r$ r1 )r; r# � commentClass)r rU r r r r �
insertComment s
zTreeBuilder.insertCommentc C s0 |d }|j d| j�}| j||�}|d |_|S )z.Create an element but don't insert it anywherer rJ r$ )�getr9 � elementClassr )r rU r rJ r6 r r r rT s
zTreeBuilder.createElementc C s | j S )N)�_insertFromTable)r r r r �_getInsertFromTable s z TreeBuilder._getInsertFromTablec C s || _ |r| j| _n| j| _dS )zsSwitch the function used to insert an element from the
normal one to the misnested table one and back againN)r_ �insertElementTablerL �insertElementNormal)r r r r r �_setInsertFromTable s
z TreeBuilder._setInsertFromTablec C sb |d }t |t�s td| ��|jd| j�}| j||�}|d |_| jd j|� | jj |� |S )Nr zElement %s not unicoderJ r$ r/ r1 )
rA r rC r] r9 r^ r r; r# r5 )r rU r rJ r6 r r r rb $ s
z TreeBuilder.insertElementNormalc C s` | j |�}| jd jtkr$| j|�S | j� \}}|dkrD|j|� n |j||� | jj|� |S )z-Create an element and insert it into the treer/ Nr1 ) rT r; r r rb � getTableMisnestedNodePositionr# r% r5 )r rU r6 r r% r r r ra . s
z TreeBuilder.insertElementTablec C sX |dkr| j d }| j s0| jr<| j d jtkr<|j|� n| j� \}}|j||� dS )zInsert text data.Nr/ r1 r1 )r; r= r r r& rd )r r$ r r% r r r 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 rd 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 )rf rg rh r r
ri rj rk )r; r � frozensetrP �generateImpliedEndTags)r �excluder r r r rm 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^ r[ rY rp r r: rH rO rQ rS rV rZ r\ rT r` rc �propertyr= rb ra r& rd rm ro rq rr r r r r r8 z s6
.
r8 )Z
__future__r r r Zpip._vendor.sixr Z constantsr r r r2 rl �setrD �objectr r r. r8 r r r r � s
K