Edit File: config.cpython-37.opt-1.pyc
B 5/9eq^ � @ s\ 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- d dl&m.Z. G dd� de%�Z/G dd� de�Z0G dd� de0�Z1G dd� de0�Z2dS ) � )�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)�Feature)�is_panel_feature_supportedc @ s e Zd ZdS )�ValidateDataExceptionN)�__name__� __module__� __qualname__� r# r# �py/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_data5 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& r0 r1 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; r4 )r* �replace�rstrip�evalr r8 ) r( � data_dict�resellerZcontrol_data_listZcontrol_data_itemZinput_dict_keysZvalidator_functionZkey_name_for_exceptionZ input_valuer# r# r$ �_validate_parameters_singleN s z)ClConfingBase._validate_parameters_single)Fr+ )r )N) r r! r" r) r r* r3 r= rI r# r# r# r$ r% ( s r% c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�ClConfigWithoutLvezC 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' r7 �_set_run�_get_run)r( �args�licence�optsr# r# r$ �runl s zClConfigWithoutLve.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: r2 �optionsr, z/ERROR: Invalid JSON string passed in --data keyz Error: %sNr? )r, r- r4 � uiSettingsz; r4 r+ r T)�json�loadsr' � ValueErrorr8 r= �_validate_input_dictionaryr r6 r1 r/ r � set_ui_config�UIConfigExceptionr; r0 r4 �lenr )r( rG r< r, � ui_parametersr# r# r$ rP z s. ( " zClConfigWithoutLve._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- rV c S s g | ]\}}|r|�qS r# r# )�.0� component� is_supportedr# r# r$ � <listcomp>� s z/ClConfigWithoutLve._get_run.<locals>.<listcomp>�supported_cl_featuresN) r1 r0 r � get_ui_configr �get_supported_cl_features�itemsr r �LSAPI�remover r& )r( r, � full_config�supported_cl_features_listr# r# r$ rQ � s zClConfigWithoutLve._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)�progZ descriptionZcommandsrK )�title�destr7 z get UI config)�helprL z set UI configz--dataTz data to set)Zrequiredro )�argparse�ArgumentParserZadd_subparsersZ add_parser� add_argumentZparse_known_args)r( �argv�parserZ subparsersZ set_subparserrT �_r# r# r$ rN � s z ClConfigWithoutLve._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)rI )r( rG r# r# r$ r[ � s z-ClConfigWithoutLve._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']['hideAccelerateWPApp'])r )r( r# r# r$ r* � s z!ClConfigWithoutLve._get_data_listN) r r! r" �__doc__rU rP rQ rN r[ r* r# r# r# r$ rJ g s $ rJ 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.r7 ) r rM r& r= rN r' r Zreseller_limit_supportedrQ rP )r( rR rS r# r# r$ rU � 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.rH )r, r4 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 setrV z/ERROR: Invalid JSON string passed in --data key)rH z Error: %sr? )r, r- r4 )Z inodeLimitsrW ZfaultsNotificationZmySQLGovSettingsr )�lve_stats2_reseller_libr z)Error: can only configure {} for reseller)�lve_stats2_libz; r4 )"r' r7 r r r= r Zis_clquota_activatedZreset_inodes_limitsr r1 rX rY r[ r r6 r? rx r^ � intersectionrL �formatry r/ r r\ r] r; r0 r4 Zset_notificationZLveStats2Exceptionr Zset_db_gov_config�DBGovernorException) r( rH Zcl_quotar, rG r< Z inode_limitsZ ui_settingsZfaults_notificationZ mysql_gov�lve_stats2_notifZadmin_only_settingsr_ r# r# r$ rP � s� ( " " " zClConfig._set_runc C s\ dt � � d�}i }d}| jd r8| jd }ddlm} n ddlm} |dkr�ttj�rd|�|� � � |�t � � � y|�t � � � W nF t jk r� } z&d|d < t� |d < t|�|d d < W dd}~X Y nX |�d t� �� i� n ttj��r|�|j |d d �� ||d< dd� t�� �� D �|d< |d }t� �rLtj|k�rL|�tj� t|| j� dS )z# Run 'get' command r+ )r, r- Nz--for-resellerr )rx )ry �errorZmySqlGov�errorsZdbInfoT)rH �defaultsrV c S s g | ]\}}|r|�qS r# r# )r` ra rb r# r# r$ rc u s z%ClConfig._get_run.<locals>.<listcomp>rd )r1 r' r? rx ry r r ZLVEr0 Zget_notificationr re r Zget_db_gov_configr| r/ r6 r r7 r rf rg r rh ri r r& )r( r, rj rH r} r< rk r# r# r$ rQ S s6 " zClConfig._get_runc C s& t || j�\}}|s"| �t|�� |S )z% Parse CLI arguments )r r& r= r )r( rs Zstatusr2 r# r# r$ rN ~ 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)rI �_validate_parameters_pairs)r( rG rH r# r# r$ r[ � 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']['hideAccelerateWPApp']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)rF r8 r )r( rG Zpairs_data_listZpairs_data_itemZkeys1Zkeys2Zval1Zval2r# r# r$ r� � s, z#ClConfig._validate_parameters_pairs)N) r r! r" rv rU rP rQ rN r[ r* r� r# r# r# r$ rw � s b+ Trw )3Z __future__r r rp r1 r9 rX �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 r% rJ rw r# r# r# r$ �<module> s0 ( ?r