B n&=[4 @sddlmZmZmZddlmZddlmZmZm Z ddddd d d d d g Z ej Z ej ZejZejZejZejZdZde Z Gdd d eZGdd d eZdS))absolute_importdivisionunicode_literals)Node) namespaces voidElementsspaceCharactersDOCUMENTDOCTYPETEXTELEMENTCOMMENTENTITYUNKNOWN TreeWalkerNonRecursiveTreeWalkerz <#UNKNOWN#>c@sleZdZdZddZddZddZdd d Zd d Zd dZ ddZ ddZ dddZ ddZ ddZdS)rz}Walks a tree yielding tokens Tokens are dicts that all have a ``type`` field specifying the type of the token. cCs ||_dS)zCCreates a TreeWalker :arg tree: the tree to walk N)tree)selfrr8lib/python3.7/site-packages/html5lib/treewalkers/base.py__init__szTreeWalker.__init__cCstdS)N)NotImplementedError)rrrr__iter__#szTreeWalker.__iter__cCs d|dS)zGenerates an error token with the given message :arg msg: the error message :returns: SerializeError token ZSerializeError)typedatar)rmsgrrrerror&szTreeWalker.errorFccs$d|||dV|r |dVdS)arGenerates 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)rname namespacerzVoid element has childrenN)r)rr rattrs hasChildrenrrremptyTag0s  zTreeWalker.emptyTagcCsd|||dS)zGenerates 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 ZStartTag)rrr rr)rr rr!rrrstartTagEs zTreeWalker.startTagcCs d||dS)zGenerates an EndTag token :arg namespace: the namespace of the token--can be ``None`` :arg name: the name of the element :returns: EndTag token ZEndTag)rrr r)rr rrrrendTagVs zTreeWalker.endTagccsx|}|t}|dt|t|}|r6d|dV|}|t}|t|d}|rdd|dV|rtd|dVdS)atGenerates 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 NZSpaceCharacters)rrZ Characters)lstripr lenrstrip)rrZmiddleleftrightrrrtextds    zTreeWalker.textcCs d|dS)zdGenerates a Comment token :arg data: the comment :returns: Comment token Comment)rrr)rrrrrcommentszTreeWalker.commentNcCsd|||dS)zGenerates a Doctype token :arg name: :arg publicId: :arg systemId: :returns: the Doctype token ZDoctype)rrpublicIdsystemIdr)rrr.r/rrrdoctypes zTreeWalker.doctypecCs d|dS)zjGenerates an Entity token :arg name: the entity name :returns: an Entity token ZEntity)rrr)rrrrrentityszTreeWalker.entitycCs|d|S)zHandles unknown node typeszUnknown node type: )r)rZnodeTyperrrunknownszTreeWalker.unknown)F)NN)__name__ __module__ __qualname____doc__rrrr#r$r%r+r-r0r1r2rrrrrs &  c@s4eZdZddZddZddZddZd d Zd S) rcCstdS)N)r)rnoderrrgetNodeDetailssz%NonRecursiveTreeWalker.getNodeDetailscCstdS)N)r)rr7rrr getFirstChildsz$NonRecursiveTreeWalker.getFirstChildcCstdS)N)r)rr7rrrgetNextSiblingsz%NonRecursiveTreeWalker.getNextSiblingcCstdS)N)r)rr7rrr getParentNodesz$NonRecursiveTreeWalker.getParentNodec cs|j}x|dk r||}|d|dd}}d}|tkrN|j|Vn|tkrrx|j|D] }|VqbWn|tkr|\}}}}|r|tdkr|tkrx| ||||D] }|VqWd}n| |||VnV|t kr| |dVn<|t kr ||dVn |tkrd}n||dV|r>||} nd} | dk rR| }q x|dk r||}|d|dd}}|tkr|\}}}}|r|tdks|tkr|||V|j|krd}P||} | dk r| }Pn ||}qTWq WdS)NrFZhtmlT)rr8r r0r r+r rrr#r$rr-rr1r r2r9r%r:r;) rZ currentNodeZdetailsrr"tokenr rZ attributesZ firstChildZ nextSiblingrrrrsZ                 zNonRecursiveTreeWalker.__iter__N)r3r4r5r8r9r:r;rrrrrrs N)Z __future__rrrZxml.domrZ constantsrrr __all__Z DOCUMENT_NODEr ZDOCUMENT_TYPE_NODEr Z TEXT_NODEr Z ELEMENT_NODEr Z COMMENT_NODErZ ENTITY_NODErrjoinobjectrrrrrrs  !