File: //usr/libexec/oracle-cloud-agent/plugins/osms/oci_agent_utils/config.pyc
a
���f� � @ sZ d Z ddlZddlZddlZddlmZ ddlZedddg�Zd dd�ZG d d
� d
�Z dS ) zD
Get a config object by calling config.from_file("/path/to/config")
� N)�Enum�OS�Windows�Linuxc C s
t | |�S �N)�Config)�filename�operating_system� r
�F/sparta/input/python_modules/oci_agent_utils/oci_agent_utils/config.py� from_file s r c @ sf e Zd ZdZi ZdZddd�Zdd� Zddd�Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
ddd�ZdS )r a�
This class loads the configuration from a yaml file and stores the
dictionary representation in memory. The main purpose is to separate
config from the implementation to allow alter settings after installation.
If no config file is specified, the config is assumed to be the file 'config.yml' in a platform-specific directory.
On Windows, the config file will be assumed located along with the agent
executable; on Linux, the config will be assumed located in: /etc/oracle-cloud-agent
The default config is packed accordingly in OS specific build. To update
config file location, please also update the build script accordingly.
NOTE: Use get_config function in the module, to avoid multiple copies of the same object.
Nc C s< |d u r(t �� dkr tj| _q.tj| _n|| _| �|� d S )Nr )�platform�systemr r � _platformr � load_config)�self� config_filer r
r
r �__init__' s
zConfig.__init__c
C s� |d u rXd}| � � r.tj�tj�tj�|�}n*| �� rJtj�ddd|�}ntd| j ��z@t
|d��"}| � t �
|��| _W d � n1 s�0 Y W n4 ty� } z td�|�|��W Y d }~n
d }~0 0 d S )Nz
config.yml�/�etc�oracle-cloud-agentzUnknown platform: �rz&Fail to load the configuration at {0}.)�
is_windows�os�path�join�dirname�sys�
executable�is_linux� RuntimeErrorr �open�_expand_log_filename�yaml� safe_load�_config� Exception�format)r r �default_config_file_name�
configFile�er
r
r r 2 s �(zConfig.load_configc C s | j �||�S r )r% �get)r �key�defaultr
r
r r+ C s z
Config.getc C s | j �|�S r )r% � __getitem__)r r, r
r
r r. F s zConfig.__getitem__c C s | j tjkS r )r r r �r r
r
r r I s zConfig.is_windowsc C s | j tjkS r )r r r r/ r
r
r r L s zConfig.is_linuxc C s | j j�� S r )r �name�lowerr/ r
r
r r
O s zConfig.platformc
C s( d|v �r$d|d v �r$| � � |d d v �r$tj�|d d | � � �}tj�|��szDt�|� | �� r�ddl}|�d�j }|�d�j
}t� |||� W nf t �y } zL| �
� r�td�|�|��n&| �� r�ddl}|�|jd�||�� W Y d}~n
d}~0 0 d |v �r$| �||d d �|d < |S )
a�
This function takes two special config settings to make the logging
filename config working across Windows and Linux. Specifically, if the
agent config has '[os]_log_dir' setting and if the 'filename' for
logging 'rotating_file' handler is not an absolute path, the 'filename'
will be assumed to be under 'log_dir' for the respective OS.
�agent�log_dirr Nr �admz(Fail to create the log directory at {0}.zMFail to create the log directory or set ownership at {0!s}. Exception: {1!r}.�logging)r
r r �
expandvars�exists�makedirsr �pwd�getpwnam�pw_uid�pw_gid�chownr& r r r' �syslog�LOG_INFO�#_Config__visit_and_replace_filename)r �configr3 r9 �uid�gidr* r> r
r
r r" R s< ��
��
��zConfig._expand_log_filenamec sn t |t�r&� �fdd�t|�� �D �S t |t�rD� �fdd�|D �S |dkrftj�|�sftj�� |�S |S dS )zDRecursively visit each node in config; expand the 'filename' settingc s i | ]\}}|�� � ||��qS r
�r@ )�.0�k�v�r3 r r
r �
<dictcomp>z � z7Config.__visit_and_replace_filename.<locals>.<dictcomp>c s g | ]}�� � |��qS r
rD )rE �itemrH r
r �
<listcomp>| rJ z7Config.__visit_and_replace_filename.<locals>.<listcomp>r N)�
isinstance�dict�list�itemsr r �isabsr )r r3 �noder, r
rH r � __visit_and_replace_filenamew s
z#Config.__visit_and_replace_filename)NN)N)N)�__name__�
__module__� __qualname__�__doc__r% r r r r+ r. r r r
r" r@ r
r
r
r r s
%r )NN)
rW r r
r �enumr r# r r r r
r
r
r �<module> s