Current File : //opt/alt/python37/lib/python3.7/site-packages/clconfig/__pycache__/config.cpython-37.pyc
B
�2Lb�] � @ sP d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlmZm Z d dl
m Z d dl m
Z
ddlmZmZ d d lmZmZ d d
lmZ d d lmZmZ dd lmZ dd
lm Z m Z m Z m Z m Z m!Z!m"Z"m#Z# d dl$m%Z% d dl&m'Z' ddl(m)Z) d dl*m+Z+ d dl,m-Z- G dd� de%�Z.G dd� de�Z/G dd� de/�Z0G dd� de/�Z1dS )� )�print_function)�absolute_importN)�ABC�abstractmethod)�
ClQuotaLib)�cpapi� )�
ui_config_lib�db_governor_lib)�is_active_reseller_limits�lve)�CloudlinuxLicenseLib)�print_dictionary�replace_params)� parse_cloudlinux_config_opts)�boolean_to_0_1�boolean_to_on_off�boolean_to_yes_no�is_int� is_string�is_string_or_none�time_unit_to_letter�is_positive_int)�FormattedException)�is_admin)� MysqlInfo)�is_litespeed_running)�Featurec @ s e Zd ZdS )�ValidateDataExceptionN)�__name__�
__module__� __qualname__� r" r" �@/opt/alt/python37/lib/python3.7/site-packages/clconfig/config.pyr # s r c @ s> e Zd Zdd� Zedd� �Zddd�Zdd
d �Zdd
d�Zd S )�
ClConfingBasec C s d| _ i | _d S )NF)�_is_json�_opts)�selfr" r" r# �__init__) s zClConfingBase.__init__c C s dS )zM
Return list of pairs (input_dict_keys, validator_function).
Nr" )r' r" r" r# �_get_data_list- s z ClConfingBase._get_data_listF�successc C s2 t |t�r |�|t�� d�� t|| jp*|� dS )z�
Output data wrapper
:param: `dict` data - data for output to stdout
:param: `bool` force_json - always output json format
)�result� timestampN)�
isinstance�dict�update�timer r% )r' �dataZ
force_jsonr+ r" r" r# � _print_data4 s
zClConfingBase._print_datar c
C s� | j r$|�dt�� i� t|d� nXy tt|d �|�di � � W n6 tk
rz } ztdt|�|f � W dd}~X Y nX t� |� dS )z�
Print error and exit
:param message: Dictionary with keys "result" as string and optional "context" as dict
r, Tr+ �contextzError: %s [%s]N)
r% r/ r0 r �print�str�get�KeyError�sys�exit)r' �messageZ
error_code�er" r" r# �_error_and_exit> s &z ClConfingBase._error_and_exitNc C s� | � � }xz|D ]r}yX|\}}|�dd��dd��dd��d�}td| �}||�dkrhtdd |id
���W q tk
r~ Y qX qW dS ) z4
Validate config single parameters.
�[� �'�]�/z data_dict%sNz*Value error in JSON key(s) options/%(key)s�key)r: r3 )r) �replace�rstrip�evalr r7 ) r' � data_dict�resellerZcontrol_data_listZcontrol_data_itemZinput_dict_keysZvalidator_functionZkey_name_for_exceptionZ input_valuer" r" r# �_validate_parameters_singleM s
z)ClConfingBase._validate_parameters_single)Fr* )r )N) r r r! r( r r) r2 r< rH r" r" r" r# r$ ' s
r$ c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� ClConfigSolozC
Perform cloudlinux-config utility for CloudLinux OS Solo.
c C sf t � }|�� s"d| _| �ddi�S d|k| _| �|�}t|�| _| j�d�dkrZ| �� n| � � d S )NTr+ z Cloudlinux license isn't validz--json�command�set)
r
�get_license_statusr% r< � _parse_args�varsr& r6 �_set_run�_get_run)r' �args�licence�optsr" r" r# �runk s
zClConfigSolo.runc
C sJ i }yt �| jd �d }W n& ttfk
rB | �ddi� Y nX y| �|� W n8 tk
r� } z| �ddt|� i� W dd}~X Y nX dt � � i d�}t
� }d |kr�|d |d < yt � |� W nL t j
k
�r } z*|d |jd
7 < |d �|j� W dd}~X Y nX |d �s$d |d< t|d �d
k�r<|d = t|d� dS )zA
Set parameters from self._opts
:return:
r1 �optionsr+ z/ERROR: Invalid JSON string passed in --data keyz Error: %sNr> )r+ r, r3 �
uiSettingsz; r3 r* r T)�json�loadsr& �
ValueErrorr7 r< �_validate_input_dictionaryr r5 r0 r. r �
set_ui_config�UIConfigExceptionr: r/ r3 �lenr )r' rF r; r+ �
ui_parametersr" r" r# rO y s. ( "
zClConfigSolo._set_runc C sv dt � � d�}i }|�t�� � ||d< dd� t�� �� D �|d< |d }t� rftj |krf|�
tj � t || j � dS )z#
Run 'get' command
r* )r+ r, rU c S s g | ]\}}|r|�qS r" r" )�.0� component� is_supportedr" r" r# �
<listcomp>� s z)ClConfigSolo._get_run.<locals>.<listcomp>�supported_cl_featuresN)
r0 r/ r �
get_ui_configr �get_supported_cl_features�itemsr r �LSAPI�remover r% )r' r+ � full_config�supported_cl_features_listr" r" r# rP � s zClConfigSolo._get_runc C sZ t jddd�}|jddd�}|jddd � |jd
d d �}|jd d
dd� |�|�\}}|S )z%
Parse CLI arguments
zcloudlinux-configz'Utility to get/set Cloudlinux UI config)�prog� descriptionZcommandsrJ )�title�destr6 z
get UI config)�helprK z
set UI configz--dataTz data to set)Zrequiredro )�argparse�ArgumentParserZadd_subparsersZ
add_parser� add_argumentZparse_known_args)r' �argv�parserZ
subparsersZ
set_subparserrS �_r" r" r# rM � s
zClConfigSolo._parse_argsc C s | � |� dS )z�
Check data validity in data_dict.
For CL OS Solo we check only single parameters.
:param data_dict: Dictionary with input data
:return: None, throws ValidateDataException if any error detected
N)rH )r' rF r" r" r# rZ � s z'ClConfigSolo._validate_input_dictionaryc C s. dt fdt fdt fdt fdt fdt fdt fgS )Nz ['uiSettings']['hideRubyApp']z ['uiSettings']['hidePythonApp']z ['uiSettings']['hideNodeJsApp']z#['uiSettings']['hidePHPextensions']z!['uiSettings']['hideLVEUserStat']z ['uiSettings']['hideXrayApp']z ['uiSettings']['hideWPOSApp'])r )r' r" r" r# r) � s zClConfigSolo._get_data_listN)
r r r! �__doc__rT rO rP rM rZ r) r" r" r" r# rI f s $ rI c @ sJ e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd
d� Z dd� Z
d
S )�ClConfigzT
Perform cloudlinux-config utility for CloudLinux OS Shared and Shared Pro.
c C sx t � }|�� s"d| _| �ddi�S d|k| _| �|�| _| jd rXt�� sX| �ddi� | jd rl| �� n| � � d S )NTr+ z Cloudlinux license isn't validz--jsonz--for-resellerzLERROR: reseller limits are not supported, can not use --for-reseller option.r6 )
r
rL r% r< rM r& r Zreseller_limit_supportedrP rO )r' rQ rR r" r" r# rT � s
z ClConfig.runc
C sD | j �dd�}|dk r:t|�s:t|�s:| �dd|id�� | j d r�|dk rZ| �ddi� t� }|�� snd }n d
}|�� t|t � � d �d � dS | j d
s�| �ddi� i }yt
� | j d
�d }W n | �ddi� Y nX y| j ||d� W n: t
k
�r. } z| �ddt|� i� W dd}~X Y nX dt � � i d�}d\}}}} |dk �r�ddlm}
||| h} t| �t|���dk�r�| �dd�|�i� n ddlm}
t� } ||k�r�|| | |< ||k�r�|| | |< yt�| � W nL tjk
�r2 } z*|d |jd 7 < |d �|j � W dd}~X Y nX ||k�r�y|
j || |d� W nL |
j k
�r� } z*|d |jd 7 < |d �|j � W dd}~X Y nX | |k�r yt � || � W nL t j!k
�r
} z*|d |jd 7 < |d �|j � W dd}~X Y nX |d �s d
|d< t|d �dk�r6|d = t|d � dS ) zA
Set parameters from self._opts
:return:
z--for-resellerNz5ERROR: reseller limits for %(reseller)s are disabled.rG )r+ r3 z--reset-inodes-limitr+ z2ERROR: cannot reset inode limits for reseller userz ERROR: quota isn't activatedr* )r+ r, Tz--dataz)ERROR: You must provide parameters to setrU z/ERROR: Invalid JSON string passed in --data key)rG z Error: %sr> )r+ r, r3 )Z inodeLimitsrV ZfaultsNotificationZmySQLGovSettingsr )�lve_stats2_reseller_libr z)Error: can only configure {} for reseller)�lve_stats2_libz; r3 )"r& r6 r r r< r Zis_clquota_activatedZreset_inodes_limitsr r0 rW rX rZ r r5 r> rx r] � intersectionrK �formatry r. r r[ r\ r: r/ r3 Zset_notificationZLveStats2Exceptionr
Zset_db_gov_config�DBGovernorException)
r' rG Zcl_quotar+ rF r; Z inode_limitsZ ui_settingsZfaults_notificationZ mysql_gov�lve_stats2_notifZadmin_only_settingsr^ r" r" r# rO � s�
(
"
"
"
zClConfig._set_runc
C sF dt � � d�}i }d}| jd r8| jd }ddlm} n ddlm} |dkr�|�|�� � |�t�� � y|�t �
� � W nF t j k
r� } z&d|d < t � |d
< t
|�|d
d < W dd}~X Y nX |�d t� �� i� n|�|j|d d
�� ||d< dd� t�� �� D �|d< |d }t� �r6tj|k�r6|�tj� t|| j� dS )z#
Run 'get' command
r* )r+ r, Nz--for-resellerr )rx )ry �errorZmySqlGov�errorsZdbInfoT)rG �defaultsrU c S s g | ]\}}|r|�qS r" r" )r_ r` ra r" r" r# rb r s z%ClConfig._get_run.<locals>.<listcomp>rc )r0 r& r> rx ry r/ Zget_notificationr rd r
Zget_db_gov_configr| r. r5 r r6 r re rf r r rg rh r r% )r' r+ ri rG r} r; rj r" r" r# rP R s2
" zClConfig._get_runc C s& t || j�\}}|s"| �t|�� |S )z%
Parse CLI arguments
)r r% r< r )r' rs Zstatusr1 r" r" r# rM { s zClConfig._parse_argsNc C s | � ||� | �|� dS )z�
Checks data validity in data_dict
:param data_dict: Dictionary with input data
:param str reseller: Reseller name to validate to config reseller
:return: None, throws ValidateDataException if any error detected
N)rH �_validate_parameters_pairs)r' rF rG r" r" r# rZ � s z#ClConfig._validate_input_dictionaryc F C s� dt fdt fdt fdt fdt fdt fdt fdt fd tfd
tfd tfd tfd
tfdtfdtfdtfdtfdtfdtfdtfdtfdtfdtfdtfdtfdtfdtfd tfd tfd
tfd tfd tfd
tfdtfd tfd tfd tfd tfd!tfd"tfd#tfd$tfdtfdtfd%tfd&tfd'tfd(tfd)tfd*tfd+tfd,tfd-tfd.tfd/tfd0tfd1tfd2tfd3tfd4tfd5tfd6tfd7tfd8tfd9tfd:tfd;tfd<tfd=tfgES )>Nz&['inodeLimits']['showUserInodesUsage']z ['uiSettings']['hideRubyApp']z ['uiSettings']['hidePythonApp']z ['uiSettings']['hideNodeJsApp']z#['uiSettings']['hidePHPextensions']z!['uiSettings']['hideLVEUserStat']z ['uiSettings']['hideXrayApp']z ['uiSettings']['hideWPOSApp']z0['faultsNotification']['faultsToInclude']['cpu']z2['faultsNotification']['faultsToInclude']['nproc']z/['faultsNotification']['faultsToInclude']['io']z0['faultsNotification']['faultsToInclude']['mem']zB['faultsNotification']['faultsToInclude']['concurrentConnections']z1['faultsNotification']['faultsToInclude']['iops']z3['faultsNotification']['notify']['admin']['period']z7['faultsNotification']['notify']['admin']['unitOfTime']z2['faultsNotification']['notify']['user']['period']z6['faultsNotification']['notify']['user']['unitOfTime']z%['faultsNotification']['notifyAdmin']z@['faultsNotification']['minimumNumberOfFaultsToNotify']['admin']z?['faultsNotification']['minimumNumberOfFaultsToNotify']['user']z)['faultsNotification']['notifyCustomers']z)['faultsNotification']['notifyResellers']z1['faultsNotification']['notifyResellerCustomers']z2['faultsNotification']['email']['notifyFromEmail']z0['faultsNotification']['email']['notifySubject']z2['faultsNotification']['email']['reportAdminMail']z0['faultsNotification']['email']['notifyCharset']z6['faultsNotification']['notify']['reseller']['period']z:['faultsNotification']['notify']['reseller']['unitOfTime']z6['faultsNotification']['notify']['customer']['period']z:['faultsNotification']['notify']['customer']['unitOfTime']z3['faultsNotification']['notifyResellerOnCustomers']zC['faultsNotification']['minimumNumberOfFaultsToNotify']['reseller']zC['faultsNotification']['minimumNumberOfFaultsToNotify']['customer']z(['faultsNotification']['notifyReseller']z'['mySQLGovSettings']['modeOfOperation']z"['mySQLGovSettings']['scriptPath']z*['mySQLGovSettings']['userMaxConnections']z1['mySQLGovSettings']['logRestrictedUsersQueries']z2['mySQLGovSettings']['gatherDataForDetailedStats']z.['mySQLGovSettings']['restrictLog']['logPath']z-['mySQLGovSettings']['restrictLog']['format']z+['mySQLGovSettings']['errorLog']['logPath']z)['mySQLGovSettings']['errorLog']['level']z.['mySQLGovSettings']['slowQueries']['logPath']z+['mySQLGovSettings']['slowQueries']['kill']z.['mySQLGovSettings']['slowQueries']['timeout']z,['mySQLGovSettings']['restrictType']['mode']z-['mySQLGovSettings']['unlimit']['unitOfTime']z)['mySQLGovSettings']['unlimit']['period']zE['mySQLGovSettings']['restrictedTimePeriods']['level1']['unitOfTime']zA['mySQLGovSettings']['restrictedTimePeriods']['level1']['period']zE['mySQLGovSettings']['restrictedTimePeriods']['level2']['unitOfTime']zA['mySQLGovSettings']['restrictedTimePeriods']['level2']['period']zE['mySQLGovSettings']['restrictedTimePeriods']['level3']['unitOfTime']zA['mySQLGovSettings']['restrictedTimePeriods']['level3']['period']zE['mySQLGovSettings']['restrictedTimePeriods']['level4']['unitOfTime']zA['mySQLGovSettings']['restrictedTimePeriods']['level4']['period']zF['mySQLGovSettings']['restrictedTimePeriods']['timeout']['unitOfTime']zB['mySQLGovSettings']['restrictedTimePeriods']['timeout']['period'])r r r r r r r r )r' r" r" r# r) � s� zClConfig._get_data_listc C s� ddddddddg}x�|D ]�}|\}}d \}}yt d
| �}W n tk
rR Y nX yt d
| �}W n tk
rx Y nX |d kr�|d k s�|d k r|d krtd d
i��qW d S )zK
Validate config pairs parameters (period and unitOfTime).
)z3['faultsNotification']['notify']['admin']['period']z7['faultsNotification']['notify']['admin']['unitOfTime'])z2['faultsNotification']['notify']['user']['period']z6['faultsNotification']['notify']['user']['unitOfTime'])z=['mySQLGovSettings']['restrictType']['unlimit']['unitOfTime']z9['mySQLGovSettings']['restrictType']['unlimit']['period'])zE['mySQLGovSettings']['restrictedTimePeriods']['level1']['unitOfTime']zA['mySQLGovSettings']['restrictedTimePeriods']['level1']['period'])zE['mySQLGovSettings']['restrictedTimePeriods']['level2']['unitOfTime']zA['mySQLGovSettings']['restrictedTimePeriods']['level2']['period'])zE['mySQLGovSettings']['restrictedTimePeriods']['level3']['unitOfTime']zA['mySQLGovSettings']['restrictedTimePeriods']['level3']['period'])zE['mySQLGovSettings']['restrictedTimePeriods']['level4']['unitOfTime']zA['mySQLGovSettings']['restrictedTimePeriods']['level4']['period'])zF['mySQLGovSettings']['restrictedTimePeriods']['timeout']['unitOfTime']zB['mySQLGovSettings']['restrictedTimePeriods']['timeout']['period'])NNz data_dict%sNr: z4Keys 'period' and 'unitOfTime' should be passed both)rE r7 r )r' rF Zpairs_data_listZpairs_data_itemZkeys1Zkeys2Zval1Zval2r" r" r# r� � s,
z#ClConfig._validate_parameters_pairs)N) r r r! rv rT rO rP rM rZ r) r� r" r" r" r# rw � s b)
Trw )2Z
__future__r r rp r0 r8 rW �abcr r Zcllimits.clquota_libr Zclcommonr r> r r
Z lvectllibr r Z cllicenser
Z cli_utilsr r Zcl_config_arg_parser Zclconfig_utilsr r r r r r r r Zclcommon.clexceptionr Zclcommon.cpapir Z db_info_libr Zclcommon.utilsr Zclcommon.constr r r$ rI rw r" r" r" r# �<module>
s. ( ?r
Mini Shell