U
    +}f'                     @   s   d Z ddlmZ ddlmZ ddlmZ ze W n e	k
rH   e
ZY nX ze W n e	k
rj   e
ZY nX G dd deZe ZdS )z9
The ``E`` Element factory for generating XML documents.
    )absolute_importN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	rPt|sHt|| _ntj| _ rft  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   z| d j pd| | d _ W n$ tk
r@   | jp4d| | _Y nX d S )N )tail
IndexErrortext)elemitem r   0/tmp/pip-unpacked-wheel-d1taxcn3/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                    sF   | j }| D ]2\}}t|tr*|||< q t| d |||< q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_makeelementETElementstrunicodeCDATA_typemap)selfr   	namespacensmapmakeelementr   r   r   r   r   r   __init__   s2    


zElementMaker.__init__c           
      O   s   | j }| jd k	r&|d dkr&| j| }| j|| jd}|rH|t || |D ]}t|r^| }|t|}|d krt	|r|
| qLt|jD ]}||}|d k	r qqtdt|j|f |||}	|	rL|t|	||	 qL|S )Nr   r   )r.   zbad argument type: %s(%r))r+   r    r%   r"   r!   r#   getr   r&   	iselementappend__mro__	TypeError__name__)
r,   tagchildrenr   r   r   r   tZbasetyper   r   r   r   __call__   s2    




zElementMaker.__call__c                 C   s
   t | |S r   r   )r,   r7   r   r   r   __getattr__   s    zElementMaker.__getattr__)NNNN)r6   
__module____qualname____doc__r0   r:   r;   r   r   r   r   r   ;   s   X      
7r   )r>   
__future__r   Z
lxml.etreeetreer&   	functoolsr   r   	NameErrorr(   r)   objectr   Er   r   r   r   <module>&   s   

 5