|
Server : Apache System : Linux cvar2.toservers.com 3.10.0-962.3.2.lve1.5.73.el7.x86_64 #1 SMP Wed Aug 24 21:31:23 UTC 2022 x86_64 User : njnconst ( 1116) PHP Version : 8.4.18 Disable Function : NONE Directory : /opt/alt/python37/lib/python3.7/site-packages/clwpos/user/__pycache__/ |
Upload File : |
B
A�co% � @ 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
� Zdd� Zed�dd�Zd#eeeee ed�dd�Zeeedd�dd�Zeeedd�dd�Zd d � Zeee d �d d �Z d!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 s t j�| j� S )z?
Checks if user customized his config already.
)�os�path�existsr )r! r r r �is_default_configS s z UserConfig.is_default_configc C sF | � � r| jS y | �� }W n tk
r0 | jS X t|t�r@|S | jS )zP
Returns default config or config content from self.config_path
)r; �DEFAULT_CONFIGr. r �
isinstance�dict)r! r/ r r r �
get_configY 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_paramsi 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')rE r )r$ r% z)Wrong wordpress path '%(wp_path)s' passedrF z9Invalid module %(module)s, available choices: %(choices)s)rG �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! rE rF rG r/ �docrootr6 r Z module_infor r r �is_module_enabled� s2 " z UserConfig.is_module_enabled)rE rF rG rH 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 foundrE )r$ r% z)Wrong wordpress path '%(wp_path)s' passedrF z9Invalid module %(module)s, available choices: %(choices)s)rG rJ r )r r r* r r3 r r r
r8 r9 rK r r? rQ �remover7 )r! rE rF rG rP 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 foundrE )r$ r% z)Wrong wordpress path '%(wp_path)s' passedrF z9Invalid module %(module)s, available choices: %(choices)s)rG rJ r )r r r* r r3 r r r
r8 r9 rK r r? rQ �appendr7 )r! rE rF rG rP 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? rA )r! �doc_root�
doc_root_inforF � module_names�namer r r �enabled_modules� s
zUserConfig.enabled_modules)� module_namerH c c s8 x2| � � D ]&\}}}||kr
tj�| j||�V q
W dS )zI
Return absolute WP paths with specified module enabled.
N)rZ r8 r9 rK r )r! r[ rV rF rY 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)rX r r � <genexpr>
s z@UserConfig.get_enabled_sites_count_by_modules.<locals>.<genexpr>)r? rL rA �any)r! Zchecked_module_namesZ sites_countr rW r )rX r �"get_enabled_sites_count_by_modules s
"z-UserConfig.get_enabled_sites_count_by_modules)N) r r r r r8 r9 rK r Z
USER_WPOS_DIRZUSER_CLWPOS_CONFIGr ZDEFAULT_MAX_CACHE_MEMORYr< r" r. r> r7 r; r? rD �strr �boolrQ rS rU rZ 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