3
(h                 @   sb   d Z ddlmZmZmZ ddlmZ ddlmZ dddd	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 do
implement a tree walker object (called TreeWalker by convention) that
implements a 'serialize' method taking a tree as sole argument and
returning an iterator generating tokens.
    )absolute_importdivisionunicode_literals   )	constants)default_etreegetTreeWalkerpprintdometreegenshi
etree_lxmlNc             K   s   | j  } | 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j
| S )a  Get a TreeWalker class for various types of tree with built-in support

    Args:
        treeType (str): 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

        Implementation: A module implementing the tree type e.g.
            xml.etree.ElementTree or cElementTree (Currently applies to the
            "etree" tree type only).
    r
      )r
   r   )r   lxml)r   r   )r   N)lowertreeWalkerCache r
   
TreeWalkerr   r   r   r   getETreeModuleget)treeTypeimplementationkwargsr
   r   r   r    r   C/tmp/pip-install-q3hcpn_q/html5lib/html5lib/treewalkers/__init__.pyr      s"    c             c   sl   g }xL| D ]D}|d }|dkr.|j |d  q
|rHddj|dV  g }|V  q
W |rhddj|dV  d S )Ntype
CharactersSpaceCharactersdatar   )r   r   )r   r   )appendjoin)tokenspendingCharacterstokenr   r   r   r   concatenateCharacterTokens<   s    

r$   c             C   sl  g }d}xVt | D ]H}|d }|d kr&|d r~|d tjd kr~|d tjkrdtj|d  }n|d }d||d f }n|d }|jd	d
| |f  |d7 }|d }xdt|j D ]T\\}}	}
|r|tjkrtj| }n|}d||	f }n|	}|jdd
| ||
f  qW |dkr^|d8 }q|dkr:|d8 }q|dkr`|jdd
| |d f  q|dkr|d r|d r|jdd
| |d |d |d r|d ndf  nF|d r|jdd
| |d |d f  n|jdd
| |d f  n|jdd
| f  q|dkr8|jdd
| |d f  q|dkrRds^tdqtd| qW dj	|S )!zPretty printer for tree walkersr   r   StartTagEmptyTag	namespacehtmlz%s %snamez%s<%s> r   r   z	%s%s="%s"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&   )
r$   r   
namespacesprefixesr   sorteditemsAssertionError
ValueErrorr    )walkeroutputindentr#   r   nsr)   attrsr'   	localnamevaluer   r   r   r	   K   sd    












)N)__doc__
__future__r   r   r   r   r   _utilsr   __all__r   r   r$   r	   r   r   r   r   <module>	   s   
'