Current File : //opt/alt/python37/lib/python3.7/site-packages/clcommon/__pycache__/sysctl.cpython-37.pyc
B
E!wd� � @ s� d dl mZ d dl mZ d dl mZ d dlZd dlmZ d dlmZm Z m
Z
d dl m Z m
Z
dZd ZG d
d � d �ZG d d
� d
�ZdS )� )�absolute_import)�division)�print_functionN)� ConfigParser)� run_command�get_file_lines�ExternalProgramFailed)�AnyStr�Listz /etc/sysctl.d/90-cloudlinux.confz/etc/sysctl.confc @ st e Zd ZdZdZedfdd�Zdd� Zedd � �Z d
d � Z
e d d
� �Z dd� Z
dd� Zdd� Zddd�Zdd� ZdS )�
SysCtlConfz(
For reading params from sysctl
z /sbin/sysctlTc C s || _ d�| j �| _|| _dS )z�
:param config_file: path to user defined systcl config file
:param mute_errors: T/F value to define should we skip errors or not (used in cldiag checker)
z{}.tmpN)� config_file�format�config_tmp_file� mute_errors)�selfr r � r �@/opt/alt/python37/lib/python3.7/site-packages/clcommon/sysctl.py�__init__ s zSysCtlConf.__init__c C s8 | j dg}y t|� W n tk
r2 | js.� Y nX dS )z>
Apply all params from sysctl.d & sysctl.conf
z--systemN)�
SYSCTL_BINr r r )r �cmdr r r �
_apply_all' s zSysCtlConf._apply_allc C s, | j dd|g}t|dd�\}}}|�� }|S )zM
Read sysctl param
:param name: name of sysctl param
z-bz-nT)r Zreturn_full_output)r r �strip)�cls�namer Zret_codeZstd_outZstd_in�valuer r r �_read_sysctl_param9 s z SysCtlConf._read_sysctl_paramc C sV t | jd��0}d�|�}|�|� |�� t�|�� � W dQ R X t�| j| j � dS )zm
Write sysctl params to sysctl.conf
:param lines: content for writing to sysctl.conf
�w� N)
�openr �join�write�flush�os�fsync�fileno�renamer )r �linesZ sysctl_confr r r �_write_params_to_fileO s
z SysCtlConf._write_params_to_filec C s | � d�d �� S )N�=r )�splitr )�liner r r �_get_param_name_from_line\ s z$SysCtlConf._get_param_name_from_linec C s t | j�}|S )zW
Read content from sysctl.conf
:return: lines from sysctl.conf
)r r )r �resultr r r �_read_sysctl_confb s
z SysCtlConf._read_sysctl_confc s$ �� � }t� �fdd�|D ��}|S )Nc 3 s | ]}� �� |�kV qd S )N)r+ )�.0r* )�
param_namer r r � <genexpr>q s z+SysCtlConf.has_parameter.<locals>.<genexpr>)r- �any)r r/ Z
file_linesr, r )r/ r r �
has_parameterm s zSysCtlConf.has_parameterc C s | � � | �|�}|S )z{
Get sysctl param by name
:param name: name of sysctl param
:return: value of sysctl param
)r r )r r r r r r �gett s
zSysCtlConf.getc
C s� d� ||�}t| �� �}d}x6t|�D ]*\}}|�d�r:q&| �|�} || kr&|}q&W |dkrh|�|� n |rt|||< | �|� | �� dS )z�
Set sysctl param by name
:param overwrite: overwrite value of existed parameter
:param name: name of sysctl param
:param value: value of sysctl param
z{} = {}
����#N) r
�listr- � enumerate�
startswithr+ �appendr' r )
r r r � overwriteZparam�sysctl_conf_outputZ idx_param�ir* �keyr r r �set� s
zSysCtlConf.setc C sn | � � t| �� �}g }x2t|�D ]&\}}| �|�}||kr"|�d|� q"W x|D ]
}||= qRW | �|� dS )z[
Remove systcl param from config
:param name: name of sysctl param
r N)r r6 r- r7 r+ �insertr' )r r r; Zidx_listr< r* r= r r r �remove� s
zSysCtlConf.removeN)T)�__name__�
__module__� __qualname__�__doc__r � SYSCTL_FILEr r � classmethodr r' � staticmethodr+ r- r2 r3 r> r@ r r r r r s
r c @ sB e Zd ZdZdZd�e�ZdZdd� Zdd� Z d d
� Z
d d � Z d
S )�
SysCtlMigratezp
Class for migrating of sysctl parameter from /etc/sysctl.conf to /etc/sysctl.conf.d/90-cloudlinux.conf
z/var/lve/cl-sysctl.migratez{}.tmp�mainc C s8 t td�| _t td�| _td dd�| _| j�| j� d S )N)r F)Z
interpolation�strict) r rE � _src_conf�SYSCTL_CL_CONF_FILE� _dst_confr �_migrate_config�read�MIGRATE_CONFIG_PATH)r r r r r � s zSysCtlMigrate.__init__c C s6 d}| j �| j�r2| j �| j|�r2| j �| j|�}|S )NF)rN � has_section� MAIN_SECTIONZ
has_optionZ
getboolean)r r/ r, r r r �_is_migration_done� s
z SysCtlMigrate._is_migration_donec C s| | j �| j�s | j �| j� | j �| j|d� t| jd��(}| j �|� |�� t �
|� � � W d Q R X t � | j| j
� d S )N�truer )rN rQ rR Z add_sectionr> r �MIGRATE_CONFIG_TMP_PATHr r! r" r# r$ r% rP )r r/ Z
config_tmpr r r � _set_migration_state_to_done� s z*SysCtlMigrate._set_migration_state_to_donec C st | j |d�rdS | j�|�r*| j�|�}n|}| j�|� |dk rd|dkrNdnd}| jj|||d� | j|d� dS )a%
Migrate sysctl parameter from one config to another
in conformity with presence of parameter in source config and default value.
All cases of using you can see in doc:
https://docs.google.com/spreadsheets/d/1H_q3TA3CMFCj1YwAOn7G1LZgxcS1F4h90B6jCgiTUpo
)r/ NTF)r: )rS rK r2 r3 r@ rM r> rV )r r/ Z
default_valuer r: r r r �migrate� s
zSysCtlMigrate.migrateN) rA rB rC rD rP r
rU rR r rS rV rW r r r r rH � s
rH )Z
__future__r r r r" Z configparserr Zclcommon.utilsr r r �typingr r
rL rE r rH r r r r �<module> s
Mini Shell