B
    |/\'                 @   s   d Z ddlmZ ddlmZ ddlmZ ye W n e	k
rH   e
ZY nX y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   y| d j pd| | d _ W n$ tk
r@   | jp4d| | _Y nX d S )N )tail
IndexErrortext)elemitem r   +lib/python3.7/site-packages/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   Zcdatar   r   r   	add_cdata   s    z(ElementMaker.__init__.<locals>.add_cdatac                sJ   | j }x>| 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_makeelementETZElementstrunicodeZCDATA_typemap)selfr   	namespacensmapZ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 || x|D ]}t|r`| }|t|}|d krt	|r|
| qNx<t|jD ]}||}|d k	rP qW tdt|j|f |||}	|	rN|t|	||	 qNW |S )Nr   r   )r(   zbad argument type: %s(%r))r%   r   r!   r   r   r   getr   r"   Z	iselementappend__mro__	TypeError__name__)
r&   tagZchildrenr   r   r   r   tZbasetyper   r   r   r   __call__   s0    





zElementMaker.__call__c             C   s
   t | |S )N)r   )r&   r/   r   r   r   __getattr__   s    zElementMaker.__getattr__)NNNN)r.   
__module____qualname____doc__r)   r1   r2   r   r   r   r   r   ;   s
   W 
6r   )r5   Z
__future__r   Z
lxml.etreeZetreer"   	functoolsr   r   	NameErrorr#   r$   objectr   Er   r   r   r   <module>(   s   

 5