U
    TdF2                     @   s   d Z ddlZddlZddlZejee	e
ef Zejeje ejeef ejeef ef Zejeef ZG dd dZejejeejejejeef   f dddZejeddd	Zejedd
dZdejeedddZejeedddZdS )z

uritemplate.variable
====================

This module contains the URIVariable class which powers the URITemplate class.

What treasures await you:

- URIVariable class

You see a hammer in front of you.
What do you do?
>

    Nc                   @   s  e Zd ZdZdZdZedddZeddd	Zedd
dZ	ddddZ
ddddZeeeeje eje dddZeeeeje eje dddZeeeeje eje dddZeeeeje eje dddZdeje ejeef dddZdS )URIVariablea  This object validates everything inside the URITemplate object.

    It validates template expansions and will truncate length as decided by
    the template.

    Please note that just like the :class:`URITemplate <URITemplate>`, this
    object's ``__str__`` and ``__repr__`` methods do not return the same
    information. Calling ``str(var)`` will return the original variable.

    This object does the majority of the heavy lifting. The ``URITemplate``
    object finds the variables in the URI and then creates ``URIVariable``
    objects.  Expansions of the URI are handled by each ``URIVariable``
    object. ``URIVariable.expand()`` returns a dictionary of the original
    variable and the expanded value. Check that method's documentation for
    more information.

    )
+#./;?&|!@z:/?#[]@!$&'()*+,;=)varc                 C   s8   || _ d| _d| _g | _g | _i | _|   |   d S )N )originaloperatorsafe	variablesvariable_namesdefaultsparse
post_parse)selfr    r   8/tmp/pip-unpacked-wheel-qt7p1f5q/uritemplate/variable.py__init__6   s    zURIVariable.__init__)returnc                 C   s   d|  S )NzURIVariable(%s)r   r   r   r   r   __repr__I   s    zURIVariable.__repr__c                 C   s   | j S N)r   r   r   r   r   __str__L   s    zURIVariable.__str__Nc           	      C   s  | j }| j d tjkr0| j d | _| j dd }| jtjdd krLtj| _|d}|D ]}d}|}d|krt|dd\}}d}|drd	}|dd
 }d}d|krt|dd\}}t	|}|r|| j
|< | j|||df qZdd | jD | _dS )zParse the variable.

        This finds the:
            - operator,
            - set of safe characters,
            - variables, and
            - defaults.

        r      N   ,=F*T:)explodeprefixc                 S   s   g | ]\}}|qS r   r   ).0varname_r   r   r   
<listcomp>z   s     z%URIVariable.parse.<locals>.<listcomp>)r   r   	operatorsr   reservedr   splittupleendswithintr   r   appendr   )	r   Zvar_list_strZvar_listr   Zdefault_valnamer'   r(   Z
prefix_strr   r   r   r   O   s4    



zURIVariable.parsec                 C   sp   d| _ | j | _| _| jdkr$d| _| jdkr4d| _| jdkrDd| _| jdkrZd| _d| _| jdkrltj| _ d	S )
zSet ``start``, ``join_str`` and ``safe`` attributes.

        After parsing the variable, we need to set up these attributes and it
        only makes sense to do it in a more easily testable way.
        r   r   )r   r   r   r"   r   r   r	   )r   r   N)r   r   startjoin_strr   r.   r   r   r   r   r   |   s    




zURIVariable.post_parse)r4   valuer'   r(   r   c                    sP  |dkrdS t |\}}| jt|r|s|s2dS ttjt |}|rd| j fdd|D S dfdd|D }  d| S t	|s|r
|sdS ttj
ttf |}|pt| }|r| jfdd|D S dfdd|D }  d| S |rHttj|}|r0|d| n|}  dt| S  d S )	z/Expansion method for the '?' and '&' operators.Nc                 3   s"   | ]}  d t | V  qdS r#   Nquoter)   vr4   r   r   r   	<genexpr>   s    z/URIVariable._query_expansion.<locals>.<genexpr>r"   c                 3   s   | ]}t | V  qd S r   r9   r;   r   r   r   r>      s     r#   c                 3   s,   | ]$\}}t |  d t |  V  qdS r8   r9   r)   kr<   r?   r   r   r>      s    c                 3   s,   | ]$\}}t |  d t |  V  qdS )r"   Nr9   r@   r?   r   r   r>      s    )is_list_of_tuplesr   	list_testtcastSequenceScalarVariableValuer6   join	dict_testMappingstrsorteditemsTextr:   r   r4   r7   r'   r(   tuplesrM   r   r=   r   _query_expansion   s@    zURIVariable._query_expansionc           
         s"  | j }| j|dks2t|ttttfs6t|dkr6dS t|\}}t	|r|s|sVd}t
t
jt |}fdd|D }|r||S dS t|s|rt
t
jttf |}|pt| }d |sd d}| fdd	|D }	|	r|	S dS t
t
j|}|r|d| n|}t|S )
zSLabel and path expansion method.

        Expands for operators: '/', '.'

        Nr   r"   c                    s   g | ]}|d k	rt | qS r   r9   r;   r?   r   r   r,      s      z5URIVariable._label_path_expansion.<locals>.<listcomp>%s=%s%s,%sc                 3   s2   | ]*\}}|d k	r t |t |f V  qd S r   r9   r@   
format_strr   r   r   r>      s   z4URIVariable._label_path_expansion.<locals>.<genexpr>)r6   r   
isinstancerK   r2   floatcomplexlenrB   rC   rD   rE   rF   rG   rH   rI   rJ   rL   rM   rN   r:   )
r   r4   r7   r'   r(   r6   rP   rM   Z	fragmentsexpandedr   rT   r   _label_path_expansion   s:    
z!URIVariable._label_path_expansionc           	         sX  | j }| j|dkrdS | jdkr&d}t|\}}t|r|sttjt |}|rz|	 fdd|D }|rv|S dS d	fdd|D }  d| S t
|s|rttjttf |}|pt| }|r|	fd	d|D S d	fd
d|D }  d| S ttj|}|r6|d| n|}|rT  dt| S  S )z"Expansion method for ';' operator.Nr   r	   c                 3   s*   | ]"}|d k	r  dt | V  qd S Nr#   r9   r;   r=   r   r   r>     s     z3URIVariable._semi_path_expansion.<locals>.<genexpr>r"   c                 3   s   | ]}t | V  qd S r   r9   r;   r?   r   r   r>     s     r#   c                 3   s4   | ],\}}|d k	rt |  dt |  V  qd S r\   r9   r@   r?   r   r   r>     s   c                 3   s4   | ],\}}|d k	rt |  dt |  V  qd S )Nr"   r9   r@   r?   r   r   r>     s   )r6   r   r   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   r:   )	r   r4   r7   r'   r(   r6   rP   rM   rZ   r   r=   r   _semi_path_expansion   s@    
z URIVariable._semi_path_expansionc                    s   |d krd S t |\}}t|rN|sNttjt |}dfdd|D S t|sZ|rttjt	tf |}|p~t
| }|rdnd d fdd|D S ttj|}|r|d | n|}t|jS )Nr"   c                 3   s   | ]}t | jV  qd S r   r:   r   r;   r   r   r   r>   7  s     z0URIVariable._string_expansion.<locals>.<genexpr>rR   rS   c                 3   s.   | ]&\}} t |jt |jf V  qd S r   r^   r@   rU   r   r   r   r>   >  s   )rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   r:   r   rO   r   r_   r   _string_expansion)  s     zURIVariable._string_expansion)var_dictr   c                 C   s   g }|dkr| j | j iS | jD ]\}}||d}|sR|dkrR|| jkrR| j| }|dkr\qd}| jdkrr| j}n*| jdkr| j}n| jdkr| j}n| j}||||d |d }|dk	r|	| qd}|r| j
| j| }| j |iS )a1  Expand the variable in question.

        Using ``var_dict`` and the previously parsed defaults, expand this
        variable and subvariables.

        :param dict var_dict: dictionary of key-value pairs to be used during
            expansion
        :returns: dict(variable=value)

        Examples::

            # (1)
            v = URIVariable('/var')
            expansion = v.expand({'var': 'value'})
            print(expansion)
            # => {'/var': '/value'}

            # (2)
            v = URIVariable('?var,hello,x,y')
            expansion = v.expand({'var': 'value', 'hello': 'Hello World!',
                                  'x': '1024', 'y': '768'})
            print(expansion)
            # => {'?var,hello,x,y':
            #     '?var=value&hello=Hello%20World%21&x=1024&y=768'}

        Nr   )r   r   )r   r	   r   r'   r(   )r   r   getr   r   r[   rQ   r]   r`   r3   r5   r6   rH   )r   ra   Zreturn_valuesr4   optsr7   rZ   Z	expansionr   r   r   expandG  s0    



zURIVariable.expand)N)__name__
__module____qualname____doc__r-   r.   rK   r   r   r   r   r   VariableValueboolrD   Optionalr2   rQ   r[   r]   r`   VariableValueDictrJ   rd   r   r   r   r   r      sJ   -149 r   )r7   r   c                 C   s0   | r$t | ttfr$tdd | D s(dS d| fS )Nc                 s   s$   | ]}t |tot|d kV  qdS )r!   N)rV   r0   rY   )r)   rD   r   r   r   r>     s     z$is_list_of_tuples.<locals>.<genexpr>)FNT)rV   listr0   allr7   r   r   r   rB     s    rB   c                 C   s   t | ttfS r   )rV   rm   r0   ro   r   r   r   rC     s    rC   c                 C   s   t | ttjjfS r   )rV   dictcollectionsabcMutableMappingro   r   r   r   rI     s    rI   utf-8)r7   encodingr   c                 C   s   t | tr| |S | S r   )rV   rK   encode)r7   ru   r   r   r   _encode  s    

rw   )r7   r   r   c                 C   s(   t | ttfst| } tjt| |S r   )rV   rK   bytesurllibr   r:   rw   )r7   r   r   r   r   r:     s    r:   )rt   )rh   collections.abcrq   typingrD   urllib.parsery   Unionr2   rW   rX   rK   rG   rF   rJ   Tupleri   Dictrl   r   Anyrj   rk   rB   rC   rI   AnyStrrx   rw   r:   r   r   r   r   <module>   s,     i"