o V=^W@sZdZddlmZmZmZddlmZddlmZddgZ iZ d d dZ d d Z d dZ dS)aA 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_importdivisionunicode_literals) constants) default_etree getTreeWalkerpprintNcKs|}|tvrR|dkrddlm}|jt|<n:|dkr(ddlm}|jt|<n*|dkr8ddlm}|jt|<n|dkrRdd lm}|d urHt}|j |fi|jSt |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 Zlxml) etree_lxmletree)rN) lowertreeWalkerCacher Z TreeWalkerr r rrZgetETreeModuleget)ZtreeTypeimplementationkwargsr r r rr?/usr/lib/python3/dist-packages/html5lib/treewalkers/__init__.pyrs"        ccsng}|D]"}|d}|dvr||dq|r$dd|dVg}|Vq|r5dd|dVdSdS)Ntype) CharactersSpaceCharactersdatarr)rr)appendjoin)tokensZpendingCharacterstokenrrrrconcatenateCharacterTokensAsrc CsFg}d}t|D]}|d}|dvr|dr<|dtjdkr<|dtjvr/tj|d}n|d}d||df}n|d}|dd ||f|d 7}|d }t|D]*\\}} } |ru|tjvrltj|}n|}d|| f}n| }|d d ||| fqY|d kr|d 8}q|dkr|d 8}q|dkr|dd ||d fq|dkr|dr|dr|dd ||d|d|dr|dndfq|dr|dd ||d|dfq|dd ||dfq|dd |fq|dkr|dd ||d fq|dkrJdtd|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 rr)ZStartTagEmptyTag namespaceZhtmlz%s %snamez%s<%s> rrz %s%s="%s"r ZEndTagCommentz %sZDoctypeZpublicIdz%sZsystemIdrz%sz%sz %srz%s"%s"rFzBconcatenateCharacterTokens should have got rid of all Space tokenszUnknown token type, %s ) rrZ namespacesprefixesrsorteditems ValueErrorr) Zwalkeroutputindentrrnsr"Zattrsr!Z localnamevaluerrrr Psp         )N)__doc__Z __future__rrrrrZ_utilsr__all__rrrr rrrrs   ,