3
(h                 @   s|   d Z ddljZejZddlmZ ye W n e	k
r@   e
ZY nX ye W n e	k
rb   e
ZY nX G dd dZe ZdS )z9
The ``E`` Element factory for generating XML documents.
    N)partialc               @   s*   e Zd ZdZd	ddZdd Zdd ZdS )
ElementMakerac  Element generator factory.

    Unlike the ordinary Element factory, the E factory allows you to pass in
    more than just a tag and some optional attributes; you can also pass in
    text and other elements.  The text is added as either text or tail
    attributes, and elements are inserted at the right spot.  Some small
    examples::

        >>> from lxml import etree as ET
        >>> from lxml.builder import E

        >>> ET.tostring(E("tag"))
        '<tag/>'
        >>> ET.tostring(E("tag", "text"))
        '<tag>text</tag>'
        >>> ET.tostring(E("tag", "text", key="value"))
        '<tag key="value">text</tag>'
        >>> ET.tostring(E("tag", E("subtag", "text"), "tail"))
        '<tag><subtag>text</subtag>tail</tag>'

    For simple tags, the factory also allows you to write ``E.tag(...)`` instead
    of ``E('tag', ...)``::

        >>> ET.tostring(E.tag())
        '<tag/>'
        >>> ET.tostring(E.tag("text"))
        '<tag>text</tag>'
        >>> ET.tostring(E.tag(E.subtag("text"), "tail"))
        '<tag><subtag>text</subtag>tail</tag>'

    Here's a somewhat larger example; this shows how to generate HTML
    documents, using a mix of prepared factory functions for inline elements,
    nested ``E.tag`` calls, and embedded XHTML fragments::

        # some common inline elements
        A = E.a
        I = E.i
        B = E.b

        def CLASS(v):
            # helper function, 'class' is a reserved word
            return {'class': v}

        page = (
            E.html(
                E.head(
                    E.title("This is a sample document")
                ),
                E.body(
                    E.h1("Hello!", CLASS("title")),
                    E.p("This is a paragraph with ", B("bold"), " text in it!"),
                    E.p("This is another paragraph, with a ",
                        A("link", href="http://www.python.org"), "."),
                    E.p("Here are some reserved characters: <spam&egg>."),
                    ET.XML("<p>And finally, here is an embedded XHTML fragment.</p>"),
                )
            )
        )

        print ET.tostring(page)

    Here's a prettyprinted version of the output from the above script::

        <html>
          <head>
            <title>This is a sample document</title>
          </head>
          <body>
            <h1 class="title">Hello!</h1>
            <p>This is a paragraph with <b>bold</b> text in it!</p>
            <p>This is another paragraph, with <a href="http://www.python.org">link</a>.</p>
            <p>Here are some reserved characters: &lt;spam&amp;egg&gt;.</p>
            <p>And finally, here is an embedded XHTML fragment.</p>
          </body>
        </html>

    For namespace support, you can pass a namespace map (``nsmap``)
    and/or a specific target ``namespace`` to the ElementMaker class::

        >>> E = ElementMaker(namespace="http://my.ns/")
        >>> print(ET.tostring( E.test ))
        <test xmlns="http://my.ns/"/>

        >>> E = ElementMaker(namespace="http://my.ns/", nsmap={'p':'http://my.ns/'})
        >>> print(ET.tostring( E.test ))
        <p:test xmlns:p="http://my.ns/"/>
    Nc                s   |d k	rd| d nd | _ |r&t|nd | _|d ks@t|s@t|d k	rL|ntj| _ r`t ni  dd }dd }t kr| t< t	 kr| t	< tj
 kr| tj
<  fdd}t kr| t<  | _d S )	N{}c             S   sF   y| d }W n$ t k
r0   | jp$d| | _Y nX |jp:d| |_d S )N    )
IndexErrortexttail)elemitemZ
last_child r   ./tmp/pip-install-q3hcpn_q/lxml/lxml/builder.pyadd_text   s
    z'ElementMaker.__init__.<locals>.add_textc             S   s   | j rtd| j  || _ d S )Nz<Can't add a CDATA section. Element already has some text: %r)r
   
ValueError)r   cdatar   r   r   	add_cdata   s    z(ElementMaker.__init__.<locals>.add_cdatac                sJ   | j }x>|j D ]2\}}t|tr,|||< q t| d |||< qW d S )N)attribitems
isinstance
basestringtype)r   r   r   kv)typemapr   r   add_dict   s
    

z'ElementMaker.__init__.<locals>.add_dict)
_namespacedict_nsmapcallableAssertionErrorETElement_makeelementstrunicodeCDATA_typemap)selfr   	namespacensmapmakeelementr   r   r   r   )r   r   __init__   s"    

zElementMaker.__init__c       
      O   s  | j }t|t r$t|tr$|j}n | jd k	rD|d dkrD| j| }| j|| jd}|rf|t || x|D ]}t	|r~| }|j
t|}|d krtj|r|j| qlx<t|jD ]}|j
|}|d k	rP qW tdt|j|f |||}	|	rl|j
t|	||	 qlW |S )Nr   r   )r+   zbad argument type: %s(%r))r(   r   r%   _QNamer
   r   r$   r   r   r    getr   r"   	iselementappend__mro__	TypeError__name__)
r)   tagchildrenr   r   r   r   tZbasetyper   r   r   r   __call__   s4    





zElementMaker.__call__c             C   s
   t | |S )N)r   )r)   r5   r   r   r   __getattr__   s    zElementMaker.__getattr__)NNNN)r4   
__module____qualname____doc__r-   r8   r9   r   r   r   r   r   ;   s
   W 
+#r   )r<   Z
lxml.etreeetreer"   QNamer.   	functoolsr   r   	NameErrorr%   r&   r   Er   r   r   r   <module>(   s   


 .