B
    a&                 @   s   d dl Z d dlZd dlZd dlmZ d dlmZ ejdkr>eZ	dddZ
dddZd	d
 Zdd Zdd ZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZdS )    N)Decimal)InlineTableDict)   c             C   s(   |j stdt| |d}| | |S )a  Writes out dict as toml to a file

    Args:
        o: Object to dump into toml
        f: File descriptor where the toml should be stored
        encoder: The ``TomlEncoder`` to use for constructing the output string

    Returns:
        String containing the toml corresponding to dictionary

    Raises:
        TypeError: When anything other than file descriptor is passed
    z0You can only dump an object to a file descriptor)encoder)write	TypeErrordumps)ofr   d r   0/tmp/pip-unpacked-wheel-gkvjc_mv/toml/encoder.pydump   s
    
r   c             C   s  d}|dkrt | j}|| d\}}||7 }t| g}x|rdd | D }x|D ]}||krXtdqXW ||7 }| }x|D ]~}	|||	 |	\}}
|s|s|
s|r|dd dkr|d7 }|d	|	 d
 7 }|r||7 }x |
D ]}|
| ||	d | < qW qW |}q:W |S )a  Stringifies input dict as toml

    Args:
        o: Object to dump into toml
        encoder: The ``TomlEncoder`` to use for constructing the output string

    Returns:
        String containing the toml corresponding to dict

    Examples:
        ```python
        >>> import toml
        >>> output = {
        ... 'a': "I'm a string",
        ... 'b': ["I'm", "a", "list"],
        ... 'c': 2400
        ... }
        >>> toml.dumps(output)
        'a = "I'm a string"
b = [ "I'm", "a", "list",]
c = 2400
'
        ```
     Nc             S   s   g | ]}t |qS r   )id).0sectionr   r   r   
<listcomp>@   s    zdumps.<locals>.<listcomp>zCircular reference detectedz


[z]
.)TomlEncoder	__class__dump_sectionsr   values
ValueErrorget_empty_table)r	   r   retvalZaddtoretvalsectionsZ
outer_objsZsection_idsZ	outer_objZnewsectionsr   Zaddtosectionssr   r   r   r   "   s4    




r   c             C   s`  t jdk r(t| dr(t| tr(| d} d|  } | d dkrH| dd  } | d}|s`| d	rl| dd
 } |r| dd} | d	d} | d} xt	| dkrJd
}| d s| dd  } | d dd| d< | d | dk}x6| d d | r| d | dkr| }|d8 }qW |r"d}nd}| d | | d  g| dd   } qW t
d	| d  d	 S )N)r   decodezutf-8z%rr   u   '"z\'z\"z\xz\\\xZu00   )sysversion_infohasattr
isinstancestrr!   
startswithreplacesplitlenunicode)vZsinglequoteiZjoinxZjoinerr   r   r   	_dump_strV   s4    


&&r6   c             C   s   d | ddddS )Nz{}ze+0ze+ze-0ze-)formatr0   )r4   r   r   r   _dump_floatu   s    r8   c             C   s(   |   }|d kr|  S |  d d S )Ni)	utcoffset	isoformat)r4   r9   r   r   r   
_dump_timey   s    r;   c               @   sB   e Zd ZedfddZdd Zdd Zdd	 Zd
d Zdd Z	dS )r   Fc             C   sV   || _ || _ttttt| jtdd tdd t	t
tt
tjdd tjttjdd i
| _d S )Nc             S   s   t |  S )N)r3   lower)r4   r   r   r   <lambda>       z&TomlEncoder.__init__.<locals>.<lambda>c             S   s   | S )Nr   )r4   r   r   r   r=      r>   c             S   s   |   ddS )Nz+00:00Z)r:   r0   )r4   r   r   r   r=      r>   c             S   s   |   S )N)r:   )r4   r   r   r   r=      r>   )_dictpreserver.   r6   r3   list	dump_listboolintfloatr8   r   datetimetimer;   date
dump_funcs)selfr@   rA   r   r   r   __init__   s    
zTomlEncoder.__init__c             C   s   |   S )N)r@   )rK   r   r   r   r      s    zTomlEncoder.get_empty_tablec             C   s8   d}x&|D ]}|dt | | d 7 }q
W |d7 }|S )Nr    ,])r3   
dump_value)rK   r4   r   r"   r   r   r   rC      s
    
zTomlEncoder.dump_listc             C   sp   d}t |tr^g }x0| D ]$\}}| |}||d |  qW |dd| d 7 }|S t| |S dS )zPreserve inline table in its compact syntax instead of expanding
        into subsection.

        https://github.com/toml-lang/toml#user-content-inline-table
        r   z = z{ z, z }
N)r-   dictitemsdump_inline_tableappendjoinr3   rP   )rK   r   r   Zval_listkr4   valr   r   r   rS      s    

zTomlEncoder.dump_inline_tablec             C   sJ   | j t|}|d kr,t|dr,| j t }|d k	r<||S | j t |S )N__iter__)rJ   gettyper,   rB   r.   )rK   r4   Zdump_fnr   r   r   rP      s    
zTomlEncoder.dump_valuec             C   s*  d}|dkr |d dkr |d7 }|   }d}x|D ]}t|}|}td|sZt|}t|| tsd}t|| trx|| D ]}	t|	trd}qW |rx|| D ]}	d}
|d| | d	 7 }| |	|| \}}|r|d
 dkr|
|7 }
n||7 }x|r|   }x||D ]t}| || || d | \}}|rb|
d| | d | d 7 }
|
|7 }
x"|D ]}|| ||d | < qhW qW |}q W ||
7 }qW n0|| d k	r||d t| 	||  d 7 }q4| j
r
t|| tr
||d | ||  7 }q4|| ||< q4W ||7 }||fS )Nr   r&   r   z^[A-Za-z0-9_-]+$FTr   z[[z]]
r   r   z]
z = )r@   r3   rematchr6   r-   rQ   rB   r   rP   rA   r   rS   )rK   r	   supZretstrZretdictZarraystrr   ZqsectionZarrayoftablesaZarraytabstrr    r   ZnewdZdsecs1Zd1r   r   r   r      s`    





 
"zTomlEncoder.dump_sectionsN)
__name__
__module____qualname__rQ   rL   r   rC   rS   rP   r   r   r   r   r   r      s   r   c                   s    e Zd Zef fdd	Z  ZS )TomlPreserveInlineDictEncoderc                s   t t| |d d S )NT)superrc   rL   )rK   r@   )r   r   r   rL      s    z&TomlPreserveInlineDictEncoder.__init__)r`   ra   rb   rQ   rL   __classcell__r   r   )r   r   rc      s   rc   c                   s,   e Zd Zeddf fdd	Zdd Z  ZS )TomlArraySeparatorEncoderFrN   c                sD   t t| || | dkr(d| }n|dr:td|| _d S )Nr   rN   z 	
,zInvalid separator for arrays)rd   rf   rL   stripr   	separator)rK   r@   rA   rh   )r   r   r   rL      s    

z"TomlArraySeparatorEncoder.__init__c             C   s   g }d}x|D ]}| | | qW x\|g krg }xF|D ]>}t|trbx.|D ]}| | qNW q:|dt| | j 7 }q:W |}q(W |d7 }|S )Nr   rM   rO   )rT   rP   r-   rB   r3   rh   )rK   r4   tr   r"   r    rr   r   r   rC      s    




z#TomlArraySeparatorEncoder.dump_list)r`   ra   rb   rQ   rL   rC   re   r   r   )r   r   rf      s   rf   c                   s*   e Zd Zedf fdd	Zdd Z  ZS )TomlNumpyEncoderFc                sl   dd l }tt| || t| j|j< t| j|j< t| j|j< | j	| j|j
< | j	| j|j< | j	| j|j< d S )Nr   )Znumpyrd   rk   rL   r8   rJ   Zfloat16Zfloat32Zfloat64	_dump_intZint16Zint32Zint64)rK   r@   rA   np)r   r   r   rL     s    zTomlNumpyEncoder.__init__c             C   s   d t|S )Nz{})r7   rE   )rK   r4   r   r   r   rl     s    zTomlNumpyEncoder._dump_int)r`   ra   rb   rQ   rL   rl   re   r   r   )r   r   rk     s   
rk   c                   s"   e Zd Zedf fdd	Z  ZS )TomlPreserveCommentEncoderFc                s4   ddl m} tt ||  fdd j|< d S )Nr   )CommentValuec                s   |   jS )N)r   rP   )r4   )rK   r   r   r=   #  r>   z5TomlPreserveCommentEncoder.__init__.<locals>.<lambda>)toml.decoderro   rd   rn   rL   rJ   )rK   r@   rA   ro   )r   )rK   r   rL      s    z#TomlPreserveCommentEncoder.__init__)r`   ra   rb   rQ   rL   re   r   r   )r   r   rn     s   rn   c                   s$   e Zd Zdd Z fddZ  ZS )TomlPathlibEncoderc             C   s   t t|S )N)r6   r.   )rK   r4   r   r   r   _dump_pathlib_path(  s    z%TomlPathlibEncoder._dump_pathlib_pathc                s6   dt jkr&dd l}t||jr&t|}tt| |S )N)r      r   )	r*   r+   pathlibr-   PurePathr.   rd   rq   rP   )rK   r4   rt   )r   r   r   rP   +  s
    
zTomlPathlibEncoder.dump_value)r`   ra   rb   rr   rP   re   r   r   )r   r   rq   &  s   rq   )N)N)rG   r[   r*   decimalr   rp   r   r+   r.   r3   r   r   r6   r8   r;   objectr   rc   rf   rk   rn   rq   r   r   r   r   <module>   s"   


4k