File: //lib/python3/dist-packages/awscli/customizations/configure/__pycache__/writer.cpython-310.pyc
o
.&�a�" � @ s0 d dl Z d dlZddlmZ G dd� de�ZdS )� N� )�SectionNotFoundErrorc @ sb e Zd Ze�d�Ze�d�Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z ddd�Z
dd� ZdS )�ConfigFileWriterz^\s*\[(?P<header>[^]]+)\]z7(?P<option>[^:=][^:=]*)\s*(?P<vi>[:=])\s*(?P<value>.*)$c C s� |� dd�}tj�|�s| �|� | �|||� dS t|d�� }|�� }W d � n1 s.w Y z)| �|||� t|d��}|� d�
|�� W d � W dS 1 sUw Y W dS t ym | �|||� Y dS w )a� Update config file with new values.
This method will update a section in a config file with
new key value pairs.
This method provides a few conveniences:
* If the ``config_filename`` does not exist, it will
be created. Any parent directories will also be created
if necessary.
* If the section to update does not exist, it will be created.
* Any existing lines that are specified by ``new_values``
**will not be touched**. This ensures that commented out
values are left unaltered.
:type new_values: dict
:param new_values: The values to update. There is a special
key ``__section__``, that specifies what section in the INI
file to update. If this key is not present, then the
``default`` section will be updated with the new values.
:type config_filename: str
:param config_filename: The config filename where values will be
written.
� __section__�defaultN�r�w� ) �pop�os�path�isfile� _create_file�_write_new_section�open� readlines�_update_section_contents�write�joinr )�self�
new_values�config_filename� section_name�f�contents� r �H/usr/lib/python3/dist-packages/awscli/customizations/configure/writer.py�
update_config s
� &� �z ConfigFileWriter.update_configc C sj t j�|�d }t j�|�st �|� t �t �|t jt jB d�d�� W d � d S 1 s.w Y d S )Nr i� r ) r r �split�isdir�makedirs�fdopenr �O_WRONLY�O_CREAT)r r �dirnamer r r r F s
��"�z ConfigFileWriter._create_filec C sb t |d��"}|�d| � g }| jd||d� |�d�|�� W d � d S 1 s*w Y d S )N�az[%s]
r �� line_numberr r r )r r �_insert_new_valuesr )r r r r r r r r r r O s �"�z#ConfigFileWriter._write_new_sectionc C sZ t t|��D ]"}|| }|�� �d�rq| j�|�}|d ur(| �||�r(| S qt|��)N)�#�;)�range�len�strip�
startswith�
SECTION_REGEX�search�_matches_sectionr )r r r �i�line�matchr r r �_find_section_startX s ��z$ConfigFileWriter._find_section_startc
C s8 |� � }| �||�}|}|d }|t|�k r�|| }| j�|�d ur,| j|||d� d S | j�|�}|d urv|}|�d��� } | |v rvt || t
�s\|| }
d| |
f } | ||< || = n| � |||| t|�d��t|�d�� � � �}d S |d7 }|t|�k s|r�|d �
d�s�|�d� | j|d ||d� d S d S )Nr r& z%s = %s
����
)�copyr5 r, r/ r0 r( � OPTION_REGEX�groupr- �
isinstance�dict�_update_subattributes�lstrip�endswith�append) r r r r �section_start_line_num�last_matching_line�jr3 r4 �key_name� option_value�new_liner r r r d sN � �
� �
��z)ConfigFileWriter._update_section_contentsc C s� |d7 }t |t|��D ]X}|| }| j�|�}|d urJt|�d��t|�d��� � }|�d��� } | |v rJ|| }
dd| | |
f } | ||< || = ||ksV| j�|�d urc| �|d ||d� |S q ||krp| �|||d� |S )Nr �
%s%s = %s
� � ) r+ r, r9 r0 r: r> r- r/ r( ) r �indexr �values�starting_indentr2 r3 r4 �current_indentrD rE rF r r r r= � s6 ����z&ConfigFileWriter._update_subattributesr c C s� g }t |�� �D ]9\}}t|t�r4|d }|�d||f � t |�� �D ]\} }
|�d|| |
f � q$n
|�d|||f � ||= q|�|d d�|�� d S )NrI z%s%s =
rG r r )�list�itemsr; r< r@ �insertr ) r r' r r �indent� new_contents�key�value� subindent�subkey�subvalr r r r( � s
��z#ConfigFileWriter._insert_new_valuesc C s\ |� d�}|�d�d| k}t|�dkr,|�d�d|d d�|dd � �f k}|p+|S |S )NrH r z[%s]r z [%s "%s"])r r: r, r )r r4 r �parts�unquoted_match� quoted_matchr r r r1 � s
�z!ConfigFileWriter._matches_sectionN)r )�__name__�
__module__� __qualname__�re�compiler/ r9 r r r r5 r r= r( r1 r r r r r s
�+ 1
r )r r^ r r �objectr r r r r �<module> s