W d dl Z d dlZd dlZd dlmZmZmZ d dlmZm Z m
Z
d dl m Z m
Z
ddlmZmZ G d d ZdS ) N)DictAnyOptional) ModuleStatusMODULES_STATUS_FILE
MAIN_LOG_PATH) atomic_write setup_logger ) NoSuchModuleMalformedConfigErrorc e Zd ZdZd Zd Zed Zd Zd Z d Z
d Z ed Z e j
d
Z d Zd Zd
Zd Zed ZdS )Configz;
Low-level logic of interaction with 'states' file
c G d d }|| _ t dt | _ | d S )Nc e Zd ZdZi ZdS ) Config.__init__..CfgN)__name__
__module__ __qualname__
worker_pidmodules py/clwizard/config/config.pyCfgr s JGGGr r z
wizard.config)r r
r logreload)selfr s r __init__zConfig.__init__ sU ??
r c i | j _ | D ]5\ }}t j | di d| j j |< 6dS )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 itemsr PENDINGget)r r module_name instructionss r set_moduleszConfig.set_modules& sf )0 %K &.'++Ir::- -DH [ ) ) r c R d | j j D S )z$Get dictionary with modules statusesc & i | ]\ }}||d S )r" r ).0moduler! s r
z#Config.statuses..@ s# ZZZofg)ZZZr )r r r# r s r statuseszConfig.statuses< s* [ZAQAWAWAYAYZZZZr c | j j | di S # t $ r}t | |d}~ww xY w)z Get modules options (dictionary)r! Nr r r% KeyErrorr r r& es r get_module_optionszConfig.get_module_optionsB sV 38#K044YCC C 3 3 3 {++ 2 3s %(
AAAc x | j j | d S # t $ r}t | |d}~ww xY w)/Get modules states in format (see ModuleStatus)r" N)r r r2 r r3 s r get_module_statuszConfig.get_module_statusJ sK 38#K0: : 3 3 3 {++ 2 3s
949c | j j | d S # t $ r}t | |d}~ww xY w)r7 status_timeNr1 r3 s r get_module_status_timez Config.get_module_status_timeR sT 38#K044]CC C 3 3 3 {++ 2 3s $'
AAAc || j j vrt | || j j | d< t j | j j | d< dS )zSet new module stater" r: N)r r r time)r r& new_states r set_module_statuszConfig.set_module_statusZ sV dh . . . {++ +2;%h/7;y{{%m444r c | j j S )z Get background worker process idr r r. s r r zConfig.worker_pidb s x""r c || j _ dS )z$Set new background worker process idNrA )r new_pids r r zConfig.worker_pidh s &r c | j j D ]0\ }}| d st | j |d 1i | 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__r#
startswithsetattrr )r kvs r
_reset_cfgzConfig._reset_cfgn sa H%++-- + +DAq<<%%
+!T*** r c B | | }|dS |d | j _ |d D ]E\ }}|d | d | di d| j j |< FdS )z
Reset config object and load data from json config.
:raises MalformedConfigError: cannot parse json config
Npidr r" r: r! )r" r: r! )rK _read_json_configr r r# r% r )r json_datar& infos r r z
Config.reloady s
**,, F ' .!*9!5!;!;!=!= K x. #xx
6 6 88Ir22- -DH [ ) ) r c b | j j | j j d}| | dS )z Dump python object state to file)rM r N)r r r _write_json_config)r states r savez Config.save s< 8&x'
&&&&&r c
t j t sdS t t d 5 }t j | cddd S # 1 swxY w Y dS # t t f$ r3}| j
dt | Y d}~dS d}~wt t f$ rC}| j
dt | t t |d}~ww xY w)z{
Load state config and parse it using json
:raises MalformedConfigError: cannot parse json config
Nzutf-8)encodingz*Unable to load config file due to error %sz#Unable to load json config file, %s) config_path)ospathexistsr openjsonloadIOErrorOSErrorr errorstr TypeError
ValueErrorr
)r fr4 s r rN zConfig._read_json_config sD w~~ 122 4 O)G<<<
$y||
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$! HNNGQ P P P44444:& O O O HNN@#a&& I I I&3FGGGQ N OsF A, A A, A##A, &A#'A, ,D=(B++D?>C==Dc t t 5 }t j | |d ddd dS # 1 swxY w Y dS )z%Write data to file using atomic write )indentN)r r r\ dump)schemard s r rR zConfig._write_json_config s -
.
. +! Ifa * * * * + + + + + + + + + + + + + + + + + +s
:>>N)r r r __doc__r r( propertyr/ r5 r8 r; r? r setterrK r rT rN staticmethodrR r r r r r s- , [ [ X[
3 3 33 3 33 3 3C C C # # X#
& & &
(' ' 'O O O( + + \+ + +r r )r\ rX r= typingr r r clwizard.constantsr r r clwizard.utilsr r
exceptionsr r
r r r r rr s & & & & & & & & & & O O O O O O O O O O 5 5 5 5 5 5 5 5 W+ W+ W+ W+ W+ W+ W+ W+ W+ W+r