3 T]4 @sddlmZmZmZddlmZddlmZmZm Z ddddd d d d d g Z ej Z ej ZejZejZejZejZdZdje 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)selfrrF/tmp/pip-install-_1y4nff5/pip/pip/_vendor/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 SerializeError)typedatar)rmsgrrrerror&szTreeWalker.errorFccs$d|||dV|r |jdVdS)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 EmptyTag)rname namespacerzVoid element has childrenN)r)rr"r!attrs 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 StartTag)rr!r"rr)rr"r!r#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 EndTag)rr!r"r)rr"r!rrrendTagVs zTreeWalker.endTagccsx|}|jt}|dt|t|}|r6d|dV|}|jt}|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 NSpaceCharacters)rr 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 Doctype)rr!publicIdsystemIdr)rr!r5r6rrrdoctypes zTreeWalker.doctypecCs d|dS)zjGenerates an Entity token :arg name: the entity name :returns: an Entity token Entity)rr!r)rr!rrrentityszTreeWalker.entitycCs|jd|S)zHandles unknown node typeszUnknown node type: )r)rZnodeTyperrrunknownszTreeWalker.unknown)F)NN)__name__ __module__ __qualname____doc__rrrr%r'r)r1r3r7r9r:rrrrrs &  c@s4eZdZddZddZddZddZd d Zd S) rcCstdS)N)r)rnoderrrgetNodeDetailssz%NonRecursiveTreeWalker.getNodeDetailscCstdS)N)r)rr?rrr getFirstChildsz$NonRecursiveTreeWalker.getFirstChildcCstdS)N)r)rr?rrrgetNextSiblingsz%NonRecursiveTreeWalker.getNextSiblingcCstdS)N)r)rr?rrr getParentNodesz$NonRecursiveTreeWalker.getParentNodec cs|j}x|dk r|j|}|d|dd}}d}|tkrN|j|Vn|tkrrx|j|D] }|VqbWn|tkr|\}}}}| s|tdkr|tkrx|j ||||D] }|VqWd}n|j |||VnV|t kr|j |dVn<|t kr|j|dVn |tkrd}n|j|dV|r@|j|} nd} | dk rT| }q x|dk r|j|}|d|dd}}|tkr|\}}}}|r|tdks|tkr|j||V|j|krd}P|j|} | dk r| }Pn |j|}qVWq WdS)NrFhtmlT)rr@r r7r r1r rrr%r'rr3rr9r r:rAr)rBrC) r currentNodedetailsrr$tokenr"r! attributesZ firstChildZ nextSiblingrrrrsZ                 zNonRecursiveTreeWalker.__iter__N)r;r<r=r@rArBrCrrrrrrs N) __future__rrrxml.domr constantsrrr __all__Z DOCUMENT_NODEr ZDOCUMENT_TYPE_NODEr Z TEXT_NODEr Z ELEMENT_NODEr Z COMMENT_NODErZ ENTITY_NODErrjoinobjectrrrrrrs  !