Current File : //opt/alt/python37/lib/python3.7/site-packages/clwpos/user/__pycache__/config.cpython-37.pyc
B
�4Lb�$ � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
d dl m Z d dl
mZmZmZ d dlmZ d dlmZ d d lmZ d d
lmZ G d d � d e�ZG d
d� de�ZdS )� )�absolute_importN)�Iterable�Optional)�ALL_OPTIMIZATION_MODULES)�
setup_logging)�get_relative_docroot� create_clwpos_dir_if_not_exists�is_run_under_user)�
is_wp_path)� constants)� WposError)�gettextc @ s e Zd ZdZdS )� ConfigErrorz^
Used for all exceptions during handling clwpos user config
in UserConfig methods
N)�__name__�
__module__� __qualname__�__doc__� r r �C/opt/alt/python37/lib/python3.7/site-packages/clwpos/user/config.pyr s r c @ s� e Zd ZdZej�dejej �Z
ej � d�Z i e d�Z dd� Z
dd� Zed �d
d �Zd d
� Zed�dd�Zd!eeeee ed�dd�Zeeedd�dd�Zeeedd�dd�Zdd� Zeee d �d d �Zd d � Z dS )"�
UserConfigzQ
Class to manage clwpos user config - read, write, set params in config.
z {homedir}Zmb)�docrootsZmax_cache_memoryc C sN t � sttd���|| _t�|�j| _| jj | jd�| _
t t �| _
t|� d S )Nz&Trying to use UserConfig class as root)�homedir)r r �_�username�pwd�getpwnam�pw_dirr � CONFIG_PATH�format� config_pathr r �_loggerr )�selfr r r r �__init__- s
zUserConfig.__init__c C s` y&t | jd��}t�|�� �S Q R X W n4 tk
rZ t�� }tt d�| j|d�d��Y nX dS )z4
Reads config from self.config_path
�rNz@Error while reading config %(config_path)s: %(exception_string)s)r Zexception_string)�message�context)
�openr �json�loads�read� Exception� traceback�
format_excr r )r! �fZ
exc_stringr r r � read_config7 s zUserConfig.read_config)�configc
C sr y6t j|ddd�}t| jd��}|�|� W dQ R X W n6 tk
rl } zttd�d|id��W dd}~X Y nX dS ) z=
Writes config (as json) to self.config_path
� T)�indentZ sort_keys�wNzDAttempt of writing to config file failed due to error:
%(exception)s� exception)r$ r% )r' �dumpsr& r �writer* r r )r! r/ Z config_jsonr- �er r r � write_configE s zUserConfig.write_configc C sL t j�| j�s| jS y | �� }W n tk
r6 | jS X t|t�rF|S | jS )zP
Returns default config or config content from self.config_path
) �os�path�existsr �DEFAULT_CONFIGr. r �
isinstance�dict)r! r/ r r r �
get_configS s zUserConfig.get_config)�paramsc C s4 | � � }x |�� D ]\}}|||< qW | �|� dS )a�
Set outer (not "docroots") params in config.
Example:
Old config:
{
"docroots": ...,
"max_cache_memory": "123mb",
}
Input params:
{
"max_cache_memory": "1024mb",
"param": "value"
}
New config:
{
"docroots": ...,
"max_cache_memory": "1024mb",
"param": "value"
}
N)r> �itemsr7 )r! r? r/ �key�valuer r r �
set_paramsc s zUserConfig.set_paramsN)�domain�wp_path�moduler/ �returnc
C s8 |p
| � � }yt|| j�}W nJ tk
rf } z,| jj|dd� ttd�|| jd�d��W d d }~X Y nX tt j
� | j||��s�ttd�d|id��|t kr�ttd�|t d �d��y$|d
}|�
|i ��
|g �}||kS tttfk
�r2 } z8| jjd | j� d |� �dd� ttd
�d| jid��W d d }~X Y nX d S )NT)�exc_infozDCan't find docroot for domain '%(domain)s' and homedir '%(homedir)s')rD r )r$ r% z)Wrong wordpress path '%(wp_path)s' passedrE z9Invalid module %(module)s, available choices: %(choices)s)rF �choicesr zconfig z is broken: z@Config is broken.
Repair %(config_path)s or restore from backup.r )r> r r r* r Zwarningr r r
r8 r9 �joinr �get�KeyError�AttributeError� TypeErrorr ) r! rD rE rF r/ �docrootr6 r Z module_infor r r �is_module_enabled} s2 " z UserConfig.is_module_enabled)rD rE rF rG c
C s yt || j�}W nB tk
rR } z$| j�|� ttd�d|id��W d d }~X Y nX ttj �
| j||��s~ttd�d|id��|t kr�ttd�|t d�d��| � � }| �
||||�s�d S |d | | �|� |d | | s�|d | |= |d | �s|d |= | �|� d S ) Nz,Docroot for domain '%(domain)s' is not foundrD )r$ r% z)Wrong wordpress path '%(wp_path)s' passedrE z9Invalid module %(module)s, available choices: %(choices)s)rF rI r )r r r* r r3 r r r
r8 r9 rJ r r> rP �remover7 )r! rD rE rF rO r6 r/ r r r �disable_module� s0
zUserConfig.disable_modulec
C s& yt || j�}W nB tk
rR } z$| j�|� ttd�d|id��W d d }~X Y nX ttj �
| j||��s~ttd�d|id��|t kr�ttd�|t d�d��| � � }| �
||||�r�d S d|kr�i |d< ||d kr�i |d |< ||d | k�rg |d | |< |d | | �|� | �|� d S ) Nz,Docroot for domain '%(domain)s' is not foundrD )r$ r% z)Wrong wordpress path '%(wp_path)s' passedrE z9Invalid module %(module)s, available choices: %(choices)s)rF rI r )r r r* r r3 r r r
r8 r9 rJ r r> rP �appendr7 )r! rD rE rF rO r6 r/ r r r �
enable_module� s4 zUserConfig.enable_modulec c sR xL| � � d �� D ]8\}}x.|�� D ]"\}}x|D ]}|||fV q2W q$W qW d S )Nr )r> r@ )r! �doc_root�
doc_root_inforE � module_names�namer r r �enabled_modules� s
zUserConfig.enabled_modules)� module_namerG c c s8 x2| � � D ]&\}}}||kr
tj�| j||�V q
W dS )zI
Return absolute WP paths with specified module enabled.
N)rY r8 r9 rJ r )r! rZ rU rE rX r r r � wp_paths_with_enabled_module� s z'UserConfig.wp_paths_with_enabled_modulec sZ d}xP| � � �di ��� D ]8\}}x.|�� D ]"\}� |t� fdd�|D ��7 }q,W qW |S )z<
Returns count of sites with enabled module
r r c 3 s | ]}|� kV qd S )Nr )�.0Zchecked_module_name)rW r r � <genexpr> s z@UserConfig.get_enabled_sites_count_by_modules.<locals>.<genexpr>)r> rK r@ �any)r! Zchecked_module_namesZ sites_countr rV r )rW r �"get_enabled_sites_count_by_modules� s
"z-UserConfig.get_enabled_sites_count_by_modules)N) r r r r r8 r9 rJ r Z
USER_WPOS_DIRZUSER_CLWPOS_CONFIGr ZDEFAULT_MAX_CACHE_MEMORYr; r" r. r= r7 r> rC �strr �boolrP rR rT rY r r[ r_ r r r r r % s
"('r ) Z
__future__r r' r8 r r+ �typingr r Zclwpos.optimization_modulesr Zclwpos.logsetupr Z clwpos.utilsr r r Zclcommon.clwpos_libr
Zclwposr Zclwpos.cl_wpos_exceptionsr r
r r �objectr r r r r �<module> s
Mini Shell