o
    ;/b                     @   s   d Z ddlmZ ddlmZ ejZddlm	Z	 ze
 W n ey&   eZ
Y nw ze W n ey6   eZY nw 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 ur
d| d nd | _ |rt|nd | _|d u s t|s J |d ur&|ntj| _ r0t ni  dd }dd }t vrB| t< t vrJ| t< tj	 vrT| tj	<  fdd}t vrb| t<  | _
d S )	N{}c                 S   sF   z| d }W n t y   | jpd| | _Y d S w |jpd| |_d S )N )
IndexErrortexttail)elemitemZ
last_child r   ./usr/lib/python3/dist-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                    sF   | j }| D ]\}}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ETZElement_makeelementstrunicodeZCDATA_typemap)selfr   	namespacensmapZmakeelementr   r   r   r   r   r   __init__   s"   


zElementMaker.__init__c           
      O   s
  | j }t|tst|tr|j}n| jd ur!|d dkr!| j| }| j|| jd}|r2|t || |D ]N}t	|r=| }|
t|}|d u rqt|rS|| q4t|jD ]}|
|}|d ure nqXtdt|j|f |||}	|	r|
t|	||	 q4|S )Nr   r   )r*   zbad argument type: %s(%r))r'   r   r%   _QNamer   r   r$   r!   r    r"   getr   r#   Z	iselementappend__mro__	TypeError__name__)
r(   tagchildrenr   r   r   r   tZbasetyper   r   r   r   __call__   s:   




zElementMaker.__call__c                 C   s
   t | |S r   r   )r(   r2   r   r   r   __getattr__   s   
zElementMaker.__getattr__)NNNN)r1   
__module____qualname____doc__r+   r5   r6   r   r   r   r   r   <   s    X
,#r   )r9   Z
__future__r   Z
lxml.etreeZetreer#   ZQNamer,   	functoolsr   r   	NameErrorr%   r&   objectr   Er   r   r   r   <module>   s$   % 
.