3
Pfv6 @ s d dl mZmZmZ d dlmZ ddlmZmZm Z dZ
e edfe ee e d dfgB dfe ee e d d fe d d
fgB dfe e d dfe d d fgdfe e d d fe d d
fgdfdZ
G dd deZG dd deZG dd deZdS ) )absolute_importdivisionunicode_literals) text_type )scopingElementstableInsertModeElements
namespacesNFhtmlbuttonZolZultableoptgroupoptionT)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)nameparentvalue
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 )zInsert 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 )zInsert 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 )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
N)r r# )r Z newParentZchildr r r reparentChildrenL s zNode.reparentChildrenc C s t dS )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
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
nodesEqualremover append)r r" Z
equalCountelementr 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+ddZ d d
Z d d Z
d
d Zdd Zdd Zd,ddZdd Zdd Zdd ZeeeZdd 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)defaultNamespacereset)r ZnamespaceHTMLElementsr r r r s zTreeBuilder.__init__c C s. g | _ t | _d | _d | _d| _| j | _d S )NF) openElementsr. activeFormattingElementsZ headPointerZ formPointerinsertFromTable
documentClassdocument)r r r r r: s zTreeBuilder.resetc C s t |d}|s2t|tr$td |f}t|ts2tt| \}}xHt| jD ]:}|r^||kr^dS | rr|j |krrdS ||j |kA rJdS qJW dstd S )Nr7 r
TF)
hasattr
isinstancer r tupleAssertionErrorlistElementsMapreversedr; r7 )r targetZvariantZ exactNodeZ listElementsinvertr" 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 ientryZcloner6 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 )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/ 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 publicIdsystemId) 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 |ts 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/ dddtlir r
prprtr1 )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