U
��.e4 � @ s� d dl mZmZmZ d dlmZ ddlmZmZm Z ddddd d
d d d
g Z
ej Z ej
ZejZejZejZejZdZd�e �Z G dd � d e�ZG dd
� d
e�ZdS )� )�absolute_import�division�unicode_literals)�Node� )�
namespaces� voidElements�spaceCharacters�DOCUMENT�DOCTYPE�TEXT�ELEMENT�COMMENT�ENTITY�UNKNOWN�
TreeWalker�NonRecursiveTreeWalkerz <#UNKNOWN#>� c @ sl e Zd ZdZdd� Zdd� Zdd� Zdd d
�Zd d � Zd
d� Z dd� Z
dd� Z ddd�Z dd� Z
dd� ZdS ) r z}Walks a tree yielding tokens
Tokens are dicts that all have a ``type`` field specifying the type of the
token.
c C s
|| _ dS )zCCreates a TreeWalker
:arg tree: the tree to walk
N)�tree)�selfr � r �I/usr/lib/python3.8/site-packages/pip/_vendor/html5lib/treewalkers/base.py�__init__ s zTreeWalker.__init__c C s t �d S �N��NotImplementedError)r r r r �__iter__# s zTreeWalker.__iter__c C s
d|d�S )z�Generates an error token with the given message
:arg msg: the error message
:returns: SerializeError token
ZSerializeError��type�datar )r �msgr r r �error& s zTreeWalker.errorFc c s$ d|||d�V |r | � d�V dS )ar Generates an EmptyTag token
:arg namespace: the namespace of the token--can be ``None``
:arg name: the name of the element
:arg attrs: the attributes of the element as a dict
:arg hasChildren: whether or not to yield a SerializationError because
this tag shouldn't have children
:returns: EmptyTag token
ZEmptyTag�r �name� namespacer zVoid element has childrenN�r! )r r$ r# �attrs� hasChildrenr r r �emptyTag0 s �zTreeWalker.emptyTagc C s d|||d�S )z�Generates a StartTag token
:arg namespace: the namespace of the token--can be ``None``
:arg name: the name of the element
:arg attrs: the attributes of the element as a dict
:returns: StartTag token
ZStartTagr" r )r r$ r# r& r r r �startTagE s
�zTreeWalker.startTagc C s d||d�S )z�Generates an EndTag token
:arg namespace: the namespace of the token--can be ``None``
:arg name: the name of the element
:returns: EndTag token
ZEndTag)r r# r$ r )r r$ r# r r r �endTagV s
�zTreeWalker.endTagc c sx |}|� t�}|dt|�t|� � }|r6d|d�V |}|�t�}|t|�d� }|rdd|d�V |rtd|d�V dS )at Generates SpaceCharacters and Characters tokens
Depending on what's in the data, this generates one or more
``SpaceCharacters`` and ``Characters`` tokens.
For example:
>>> from html5lib.treewalkers.base import TreeWalker
>>> # Give it an empty tree just so it instantiates
>>> walker = TreeWalker([])
>>> list(walker.text(''))
[]
>>> list(walker.text(' '))
[{u'data': ' ', u'type': u'SpaceCharacters'}]
>>> list(walker.text(' abc ')) # doctest: +NORMALIZE_WHITESPACE
[{u'data': ' ', u'type': u'SpaceCharacters'},
{u'data': u'abc', u'type': u'Characters'},
{u'data': u' ', u'type': u'SpaceCharacters'}]
:arg data: the text data
:returns: one or more ``SpaceCharacters`` and ``Characters`` tokens
NZSpaceCharactersr Z
Characters)�lstripr �len�rstrip)r r Zmiddle�left�rightr r r �textd s
zTreeWalker.textc C s
d|d�S )zdGenerates a Comment token
:arg data: the comment
:returns: Comment token
�Commentr r )r r r r r �comment� s zTreeWalker.commentNc C s d|||d�S )z�Generates a Doctype token
:arg name:
:arg publicId:
:arg systemId:
:returns: the Doctype token
ZDoctype)r r# �publicId�systemIdr )r r# r3 r4 r r r �doctype� s
�zTreeWalker.doctypec C s
d|d�S )zjGenerates an Entity token
:arg name: the entity name
:returns: an Entity token
ZEntity)r r# r )r r# r r r �entity� s zTreeWalker.entityc C s | � d| �S )zHandles unknown node typeszUnknown node type: r% )r ZnodeTyper r r �unknown� s zTreeWalker.unknown)F)NN)�__name__�
__module__� __qualname__�__doc__r r r! r( r) r* r0 r2 r5 r6 r7 r r r r r s
&
c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) r c C s t �d S r r �r Znoder r r �getNodeDetails� s z%NonRecursiveTreeWalker.getNodeDetailsc C s t �d S r r r< r r r �
getFirstChild� s z$NonRecursiveTreeWalker.getFirstChildc C s t �d S r r r< r r r �getNextSibling� s z%NonRecursiveTreeWalker.getNextSiblingc C s t �d S r r r< r r r �
getParentNode� s z$NonRecursiveTreeWalker.getParentNodec c s� | j }|d k �r�| �|�}|d |dd � }}d}|tkrJ| j|� V n�|tkrj| j|� D ]
}|V q\n�|tkr�|\}}}}|r�|td kr�|tkr�| � ||||�D ]
}|V q�d}n| �
|||�V nV|t kr�| � |d �V n<|t
k�r | �|d �V n |tk�rd}n| �|d �V |�r2| �|�} nd } | d k �rF| }q|d k r| �|�}|d |dd � }}|tk�r�|\}}}}|�r�|td k�s�|tk�r�| �||�V | j |k�r�d }q| �|�}
|
d k �r�|
}qn
| �|�}�qFqd S )Nr � FZhtmlT)r r= r r5 r r0 r
r r r( r) r r2 r r6 r
r7 r> r* r? r@ ) r Z currentNodeZdetailsr r' �tokenr$ r# Z
attributesZ
firstChildZ nextSiblingr r r r � s\
�
z NonRecursiveTreeWalker.__iter__N)r8 r9 r: r= r> r? r@ r r r r r r � s
N) Z
__future__r r r Zxml.domr Z constantsr r r �__all__Z
DOCUMENT_NODEr
ZDOCUMENT_TYPE_NODEr Z TEXT_NODEr Z ELEMENT_NODEr
Z COMMENT_NODEr Z ENTITY_NODEr r �join�objectr r r r r r � s" �
!