File: //lib/python3.6/site-packages/awscli/customizations/configure/__pycache__/writer.cpython-36.pyc
3
�T_�" � @ s0 d dl Z d dlZddlmZ G dd� de�ZdS )� N� )�SectionNotFoundErrorc @ sb e Zd Zejd�Zejd�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� |j dd�}tjj|�s4| j|� | j|||� dS t|d��}|j� }W dQ R X y8| j|||� t|d��}|j dj
|�� W dQ R X W n" t k
r� | j|||� Y nX dS )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.
Z __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 � /usr/lib/python3.6/writer.py�
update_config s
z ConfigFileWriter.update_configc C sT t jj|�d }t jj|�s&t j|� t jt j|t jt jB d�d�� W d Q R X 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 sL t |d��8}|jd| � g }| jd||d� |jdj|�� W d Q R X 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 s^ xPt t|��D ]@}|| }|j� jd�r*q| jj|�}|d k r| j||�r|S qW t|��d S )N�#�;)r' r( )�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 s2 |j � }| j||�}|}|d }x�|t|�k r�|| }| jj|�d k rZ| j|||d� d S | jj|�}|d k r�|}|jd�j� } | |kr�t || t
�s�|| }
d| |
f } | ||< || = n4| j |||| t|jd��t|jd�j � � �}d S |d7 }q"W |�r.|d j
d��s|jd� | j|d ||d� d S )Nr )r% r r z%s = %s
�
���)�copyr3 r* r- r. r& � OPTION_REGEX�groupr+ �
isinstance�dict�_update_subattributes�lstrip�endswith�append) r r r r Zsection_start_line_numZlast_matching_line�jr1 r2 �key_name� option_value�new_liner r r r d sB
"
z)ConfigFileWriter._update_section_contentsc C s� |d7 }x�t |t|��D ]�}|| }| jj|�}|d k r�t|jd��t|jd�j� � }|jd�j� } | |kr�|| }
dd| | |
f } | ||< || = ||ks�| jj|�d k r| j|d ||d� P qW ||kr�| j|||d� |S )Nr z
%s%s = %s
� z ) r) r* r7 r. r8 r< r+ r- r& ) r �indexr �valuesZstarting_indentr0 r1 r2 Zcurrent_indentr@ rA rB r r r r; � s*
z&ConfigFileWriter._update_subattributesr c C s� g }x�t |j� �D ]v\}}t|t�rn|d }|jd||f � xBt |j� �D ] \} }
|jd|| |
f � qLW n|jd|||f � ||= qW |j|d dj|�� d S )Nz z%s%s =
z
%s%s = %s
r r )�list�itemsr9 r: r> �insertr ) r r% r r �indentZ new_contents�key�valueZ subindentZsubkeyZsubvalr r r r& � s
z#ConfigFileWriter._insert_new_valuesc C s\ |j d�}|jd�d| k}t|�dkrX|jd�d|d dj|dd � �f k}|pV|S |S )NrC r z[%s]r z [%s "%s"])r r8 r* r )r r2 r �partsZunquoted_matchZ quoted_matchr r r r/ � s
z!ConfigFileWriter._matches_sectionN)r )�__name__�
__module__� __qualname__�re�compiler- r7 r r
r r3 r r; r& r/ r r r r r s
+ 1
r )r
rP r r �objectr r r r r �<module>
s