Edit File: config.cpython-37.opt-1.pyc
B 5/9e� � @ 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 � py/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