Current File : //opt/alt/python37/lib/python3.7/site-packages/clwizard/config/__pycache__/config.cpython-37.pyc
B
�2Lb� � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
d dl m Z m
Z
mZ d dlmZmZ ddlmZmZ G d d
� d
e�ZdS ) � )�absolute_importN)� iteritems)�Dict�Any�Optional)� ModuleStatus�MODULES_STATUS_FILE�
MAIN_LOG_PATH)� atomic_write� setup_logger� )� NoSuchModule�MalformedConfigErrorc @ s� e Zd ZdZdd� Zdd� Zedd� �Zdd � Zd
d � Z d d
� Z
dd� Z edd� �Z e j
dd� �Z dd� Zdd� Zdd� Zdd� Zedd � �Zd S ) �Configz;
Low-level logic of interaction with 'states' file
c C s, G dd� d�}|| _ tdt�| _| �� d S )Nc @ s e Zd ZdZi ZdS )z Config.__init__.<locals>.CfgN)�__name__�
__module__� __qualname__�
worker_pid�modules� r r �G/opt/alt/python37/lib/python3.7/site-packages/clwizard/config/config.py�Cfg s r z
wizard.config)r r r �log�reload)�selfr r r r �__init__ s zConfig.__init__c C s> i | j _x0t|�D ]$\}}tj|�di �d�| j j|< qW dS )a�
Forget about the previous config and create a new one with specified
modules and instructions.
{'cagefs': {'options': {'enable_for_new_users': True, 'enable_for_existing_users': True}},
'mod_lsapi': {'options': {'use_beta_for_da': True}},
'mysql_governor': {'options': {'mode': 'single'}},
'php': {'options': {'install_versions': ['5.2', '5.3'],
'install_modules': False, 'default_version': '5.3'}},
'nodejs': {'options': {'versions': ['7.4', '8.9'], 'default_version': '8.9'}},
'python': {'options': {'versions': ['2.7', '3.6']}},
'ruby': {'options': {'versions': ['2.7']}}}
�options)�statusr N)r r r r ZPENDING�get)r r � module_nameZ instructionsr r r � set_modules( s zConfig.set_modulesc C s dd� t | jj�D �S )z$Get dictionary with modules statusesc S s i | ]\}}|d |�qS )r r )�.0�moduler r r r �
<dictcomp>B s z#Config.statuses.<locals>.<dictcomp>)r r r )r r r r �statuses> s zConfig.statusesc C s6 y| j j| �di �S tk
r0 t|��Y nX dS )z Get modules options (dictionary)r N)r r r �KeyErrorr
)r r r r r �get_module_optionsD s zConfig.get_module_optionsc C s2 y| j j| d S tk
r, t|��Y nX dS )z/Get modules states in format (see ModuleStatus)r N)r r r% r
)r r r r r �get_module_statusL s zConfig.get_module_statusc C s4 y| j j| �d�S tk
r. t|��Y nX dS )z/Get modules states in format (see ModuleStatus)� status_timeN)r r r r% r
)r r r r r �get_module_status_timeT s z Config.get_module_status_timec C s< || j jkrt|��|| j j| d< t�� | j j| d< dS )zSet new module stater r( N)r r r
�time)r r Z new_stater r r �set_module_status\ s zConfig.set_module_statusc C s | j jS )z Get background worker process id)r r )r r r r r d s zConfig.worker_pidc C s || j _dS )z$Set new background worker process idN)r r )r Znew_pidr r r r j s c C s> x0| j j�� D ] \}}|�d�st| j |d� qW i | j _dS )z
Reset self.Cfg object to default values before it will be loaded
from file as a part of self.reload()
�__N)r �__dict__�items�
startswith�setattrr )r �k�vr r r �
_reset_cfgp s
zConfig._reset_cfgc C sl | � � | �� }|dkr dS |d | j_x>t|d �D ].\}}|d |�d�|�di �d�| jj|< q6W dS )z�
Reset config object and load data from json config.
:raises MalformedConfigError: cannot parse json config
N�pidr r r( r )r r( r )r3 �_read_json_configr r r r r )r Z json_datar �infor r r r { s z
Config.reloadc C s | j j| j jd�}| �|� dS )z Dump python object state to file)r4 r N)r r r �_write_json_config)r �stater r r �save� s z Config.savec
C s� t j�t�sdS y tt�� }t�|�S Q R X W nx ttfk
rf } z| j �
dt |�� dS d}~X Y nB t t
fk
r� } z | j �
dt |�� ttd��W dd}~X Y nX dS )z{
Load state config and parse it using json
:raises MalformedConfigError: cannot parse json config
Nz*Unable to load config file due to error %sz#Unable to load json config file, %s)Z config_path)�os�path�existsr �open�json�load�IOError�OSErrorr �error�str� TypeError�
ValueErrorr )r �f�er r r r5 � s
zConfig._read_json_configc C s( t t��}tj| |dd� W dQ R X dS )z%Write data to file using atomic write� )�indentN)r
r r> �dump)ZschemarF r r r r7 � s
zConfig._write_json_configN)r r r �__doc__r r �propertyr$ r& r' r) r+ r �setterr3 r r9 r5 � staticmethodr7 r r r r r s
r )Z
__future__r r> r: r* Z future.utilsr �typingr r r Zclwizard.constantsr r r Zclwizard.utilsr
r �
exceptionsr
r �objectr r r r r �<module> s
Mini Shell