File: //usr/lib/python3.9/site-packages/pip/_vendor/toml/__pycache__/encoder.cpython-39.pyc
a
?�f�& � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ ejdkr>eZ ddd�Z
d dd�Z d d
� Z d d � Z
d
d� ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS ) � N)�Decimal)�InlineTableDict�� c C s( |j std��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
�</usr/lib/python3.9/site-packages/pip/_vendor/toml/encoder.py�dump s
r c C s� d}|du rt | j�}|�| d�\}}||7 }t| �g}|r�dd� |D �}|D ]}||v rNtd��qN||7 }|�� }|D ]z} |�|| | �\}}
|s�|s�|
s�|r�|dd� dkr�|d7 }|d | d
7 }|r�||7 }|
D ]} |
| || d | < q�qx|}q8|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 |��qS r
)�id)�.0�sectionr
r
r �
<listcomp>@ � zdumps.<locals>.<listcomp>zCircular reference detected���z
�
�[�]
�.)� TomlEncoder� __class__�
dump_sectionsr �
ValueError�get_empty_table) r
r �retvalZ addtoretvalZsectionsZ
outer_objsZ section_idsZ outer_objZ newsectionsr Z
addtosections�sr
r
r r " s6
� r c C sX t jdk r(t| d�r(t| t�r(| �d�} d| } | d dkrH| dd � } | �d�}|s`| �d �rl| dd
� } |r�| �d d�} | �d d �} | �d
�} t | �dk�rDd
}| d s�| dd � } | d �dd�| d< | d | dk}| d d |� �r| d | dk�r| }|d8 }q�|�r d}nd}| d | | d g| dd � } q�t
d | d d �S )Nr �decodezutf-8z%rr �u� �'�"���z\'z\"z\xz\\�\�xZu00� ) �sys� version_info�hasattr�
isinstance�strr" �
startswith�replace�split�len�unicode)�vZ singlequote�iZjoinxZjoinerr
r
r � _dump_strV s4
$
$r7 c C s d� | ��dd��dd�S )N�{}ze+0ze+ze-0ze-)�formatr1 �r5 r
r
r � _dump_floatu s r; c C s( | � � }|d u r| �� S | �� d d� S )Ni����)� utcoffset� isoformat)r5 r< r
r
r �
_dump_timey s r> c @ sB e Zd Zedfdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )r Fc C sV || _ || _ttttt| jtdd� tdd� t t
t t
t j dd� t j
tt jdd� i
| _d S )Nc S s t | ��� S �N)r4 �lowerr: r
r
r �<lambda>� r z&TomlEncoder.__init__.<locals>.<lambda>c S s | S r? r
r: r
r
r rA � r c S s | � � �dd�S )Nz+00:00�Z)r= r1 r: r
r
r rA � r c S s | � � S r? )r= r: r
r
r rA � r )�_dict�preserver/ r7 r4 �list� dump_list�bool�int�floatr; r �datetime�timer> �date�
dump_funcs)�selfrC rD r
r
r �__init__� s
�zTomlEncoder.__init__c C s | � � S r? )rC �rN r
r
r r � s zTomlEncoder.get_empty_tablec C s4 d}|D ] }|dt | �|�� d 7 }q|d7 }|S )Nr � �,�])r4 �
dump_value)rN r5 r r# r
r
r rF � s
zTomlEncoder.dump_listc C sl d}t |t�rZg }|�� D ]$\}}| �|�}|�|d | � q|dd�|� d 7 }|S t| �|��S dS )z�Preserve 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 }
N)r. �dict�items�dump_inline_table�append�joinr4 rT )rN r r Zval_list�kr5 �valr
r
r rX � s
z TomlEncoder.dump_inline_tablec C sJ | j �t|��}|d u r,t|d�r,| j t }|d ur<||�S | j t |�S )N�__iter__)rM �get�typer- rE r/ )rN r5 Zdump_fnr
r
r rT � s
zTomlEncoder.dump_valuec C s d}|dkr |d dkr |d7 }| � � }d}|D �]�}t|�}|}t�d|�sVt|�}t|| t��s�d}t|| t�r�|| D ]} t| t�r�d}q�|�r�|| D ]�} d}
|d| | d 7 }| �| || �\} } | r�| d
d kr�|
| 7 }
n|| 7 }| �r~| � � }
| D ]p}| �| | || d | �\}}|�rT|
d | | d | d 7 }
|
|7 }
|D ]}|| |
|d | < �qX�q|
} q�||
7 }q�n0|| d u�r�||d
t| � || �� d 7 }q0| j
�r�t|| t ��r�||d
| � || � 7 }q0|| ||< q0||7 }||fS )Nr r' r z^[A-Za-z0-9_-]+$FTr z[[z]]
r r r rU )
rC r4 �re�matchr7 r. rV rE r rT rD r rX )rN r
ZsupZretstrZretdictZarraystrr ZqsectionZ
arrayoftables�aZ arraytabstrr! r ZnewdZdsec�s1Zd1r
r
r r � sz
���
��� �� �zTomlEncoder.dump_sectionsN)
�__name__�
__module__� __qualname__rV rO r rF rX rT r r
r
r
r r � s r c s e Zd Zef� fdd� Z� ZS )� TomlPreserveInlineDictEncoderc s t t| ��|d� d S )NT)�superrg rO )rN rC �r r
r rO � s z&TomlPreserveInlineDictEncoder.__init__�rd re rf rV rO �
__classcell__r
r
ri r rg � s rg c s, e Zd Zeddf� fdd� Zdd� Z� ZS )�TomlArraySeparatorEncoderFrR c sD t t| ��||� |�� dkr(d| }n|�d�r:td��|| _d S )Nr rR z
,z Invalid separator for arrays)rh rl rO �stripr � separator)rN rC rD rn ri r
r rO � s
z"TomlArraySeparatorEncoder.__init__c C s� g }d}|D ]}|� | �|�� q |g krtg }|D ]:}t|t�rV|D ]}|� |� qDq2|dt|� | j 7 }q2|}q"|d7 }|S )Nr rQ rS )rY rT r. rE r4 rn )rN r5 �tr r# r! �rr
r
r rF � s
z#TomlArraySeparatorEncoder.dump_list)rd re rf rV rO rF rk r
r
ri r rl � s rl c s* e Zd Zedf� fdd� Zdd� Z� ZS )�TomlNumpyEncoderFc sl dd l }tt| ��||� t| j|j< t| j|j< t| j|j< | j | j|j
< | j | j|j < | j | j|j < d S )Nr )
Znumpyrh rq rO r; rM Zfloat16Zfloat32Zfloat64� _dump_intZint16Zint32Zint64)rN rC rD Znpri r
r rO s zTomlNumpyEncoder.__init__c C s d� t|��S )Nr8 )r9 rH �rN r5 r
r
r rr s zTomlNumpyEncoder._dump_int)rd re rf rV rO rr rk r
r
ri r rq s
rq c s" e Zd Zedf� fdd� Z� ZS )�TomlPreserveCommentEncoderFc s4 ddl m} tt� ��||� � fdd�� j|< d S )Nr )� CommentValuec s | � � j�S r? )r rT r: rP r
r rA # r z5TomlPreserveCommentEncoder.__init__.<locals>.<lambda>)�pip._vendor.toml.decoderru rh rt rO rM )rN rC rD ru ri rP r rO s z#TomlPreserveCommentEncoder.__init__rj r
r
ri r rt s rt c s$ e Zd Zdd� Z� fdd�Z� ZS )�TomlPathlibEncoderc C s t t|��S r? )r7 r/ rs r
r
r �_dump_pathlib_path( s z%TomlPathlibEncoder._dump_pathlib_pathc s6 dt jkr&dd l}t||j�r&t|�}tt| ��|�S )N)r � r ) r+ r, �pathlibr. �PurePathr/ rh rw rT )rN r5 rz ri r
r rT + s
z TomlPathlibEncoder.dump_value)rd re rf rx rT rk r
r
ri r rw & s rw )N)N)rJ r` r+ Zdecimalr rv r r, r/ r4 r r r7 r; r> �objectr rg rl rq rt rw r
r
r
r �<module> s"
4 k