ruạṛ
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
cải xoăn