U
    <gW                     @   sZ   d Z ddlmZmZmZ ddlmZ ddlmZ ddgZ	i Z
dd	dZd
d Zdd ZdS )a  A collection of modules for iterating through different kinds of
tree, generating tokens identical to those produced by the tokenizer
module.

To create a tree walker for a new type of tree, you need to
implement a tree walker object (called TreeWalker by convention) that
implements a 'serialize' method which takes a tree as sole argument and
returns an iterator which generates tokens.
    )absolute_importdivisionunicode_literals   )	constants)default_etreegetTreeWalkerpprintNc                 K   s   |   } | tkr| dkr0ddlm} |jt| < np| dkrPddlm} |jt| < nP| dkrpddlm} |jt| < n0| dkrdd	lm} |d
krt}|j	|f|jS t
| S )a;  Get a TreeWalker class for various types of tree with built-in support

    :arg str treeType: the name of the tree type required (case-insensitive).
        Supported values are:

        * "dom": The xml.dom.minidom DOM implementation
        * "etree": A generic walker for tree implementations exposing an
          elementtree-like interface (known to work with ElementTree,
          cElementTree and lxml.etree).
        * "lxml": Optimized walker for lxml.etree
        * "genshi": a Genshi stream

    :arg implementation: A module implementing the tree type e.g.
        xml.etree.ElementTree or cElementTree (Currently applies to the "etree"
        tree type only).

    :arg kwargs: keyword arguments passed to the etree walker--for other
        walkers, this has no effect

    :returns: a TreeWalker class

    dom   )r
   genshi)r   lxml)
etree_lxmletree)r   N)lowertreeWalkerCache r
   
TreeWalkerr   r   r   r   getETreeModuleget)treeTypeimplementationkwargsr
   r   r   r    r   A/tmp/pip-unpacked-wheel-jqcau075/html5lib/treewalkers/__init__.pyr      s"    c                 c   sh   g }| D ]D}|d }|dkr,| |d  q|rFdd|dV  g }|V  q|rddd|dV  d S )Ntype)
CharactersSpaceCharactersdatar   r   )r   r   )appendjoin)tokenspendingCharacterstokenr   r   r   r   concatenateCharacterTokensA   s    r$   c              	   C   sb  g }d}t | D ]D}|d }|dkr|d rz|d tjd krz|d tjkr`tj|d  }n|d }d||d f }n|d }|dd	| |f  |d
7 }|d }t| D ]T\\}}	}
|r|tjkrtj| }n|}d||	f }n|	}|dd	| ||
f  q|dkrV|d
8 }q|dkr2|d
8 }q|dkrX|dd	| |d f  q|dkr
|d r|d r|dd	| |d |d |d r|d ndf  nF|d r|dd	| |d |d f  n|dd	| |d f  n|dd	| f  q|dkr0|dd	| |d f  q|dkrJdsVtdqtd| qd	|S ) zPretty printer for tree walkers

    Takes a TreeWalker instance and pretty prints the output of walking the tree.

    :arg walker: a TreeWalker instance

    r   r   )StartTagEmptyTag	namespacehtmlz%s %snamez%s<%s> r   r   z	%s%s="%s"r&   EndTagCommentz%s<!-- %s -->DoctypepublicIdz%s<!DOCTYPE %s "%s" "%s">systemIdr   z%s<!DOCTYPE %s "" "%s">z%s<!DOCTYPE %s>z%s<!DOCTYPE >r   z%s"%s"r   FzBconcatenateCharacterTokens should have got rid of all Space tokenszUnknown token type, %s
)
r$   r   
namespacesprefixesr   sorteditemsAssertionError
ValueErrorr    )walkeroutputindentr#   r   nsr)   attrsr'   	localnamevaluer   r   r   r	   P   sn    













)N)__doc__
__future__r   r   r   r   r   _utilsr   __all__r   r   r$   r	   r   r   r   r   <module>   s   

,