HEX
Server: Apache
System: Linux insrv.serversdaddy.com 5.14.0-362.24.2.el9_3.x86_64 #1 SMP PREEMPT_DYNAMIC Sat Mar 30 14:11:54 EDT 2024 x86_64
User: jsaha (1066)
PHP: 8.3.6
Disabled: exec,passthru,shell_exec,system
$ pwd: /opt/alt/python37/lib/python3.7/site-packages/__pycache__/
Upload Files
File: //opt/alt/python37/lib/python3.7/site-packages/__pycache__/lvectllib.cpython-37.pyc
B

��\�j�d�	@s&ddlmZddlmZddlmZddlmZddlmZmZddl	m
Z
mZddlm
Z
mZmZmZmZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmmZ ddl!Z!dd	l"m#Z#ddl$Z$ddl%Z%dd
lm&Z&ddl'Z'ddl(Z(ddl)m*Z*ddl+m,Z,dd
l-m.Z.m/Z/m0Z0m1Z1ddl2m3Z3ddl4Z4ddl5m6Z6m7Z7ddl8m9Z9ddl:m;Z;m<Z<m=Z=ddl>m?Z?m@Z@ddl'mAZAmBZBmCZCmDZDmEZEddlFmGZGddlHmIZImJZJdZKdZLdZMdZNdZOdZPdZQdZRdZSeTejU�Vdd��ZWd ZXe1e,jY��s"dZZdZ[neBeWd!�ZZeAeZd"�Z[d#d$�Z\d%d&�Z]d'd(�Z^e3d)�d*d+�Z_�dd,d-�Z`d.d/�Zad0d1d0d2dd3dd4d5�Zbd3Zcd6d1ddddddd5�Zdd7d8d9d:d;d<d=d>gZed?afdagdahd@aid@ajiakd@ald@amd@and@aod@apd@aqd@ardAasd@atd@aud@avdBdC�ZwdDdE�ZxdFdG�Zy�ddHdI�ZzdJdK�Z{dLdM�Z|e{��p8e|�Z}�ddNdO�Z~dPdQ�ZdRdS�Z�dTdU�Z��ddVdW�Z�dXdY�Z�ej�dZd[��Z��dd\d]�Z�d^d_�Z��dd`da�Z�dbdc�Z�ddde�Z��ddfdg�Z�dhdi�Z�djdk�Z�dldm�Z�eTe�dn�dodp�Z�dqdr�Z�dsdt�Z�e
e�eTfdu�dvdw�Z�dxdy�Z��d d{d|�Z�d}d~�Z��d!dd��Z��d"d�d��Z�d�d��Z�d�d��Z�d�d��Z��d#d�d��Z��d$d�d��Z�d�d��Z�d�d��Z�d�d��Z��d%d�d��Z�d�d��Z�ia�d�Z�d�Z��d&d�d��Z�d�d��Z��d'd�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z��d(d�d��Z�e�fd�d��Z��d)d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z��d*d�d��Z�d�d��Z�d�d„Z�d�dĄZ�d�dƄZ�d�dȄZ�d�dʄZ�d�d̄Z�d�d΄Z�d�dЄZ�d�d҄Z�d�dԄZÐd+d�dքZ�d�d؄ZŐd,d�dڄZƐd-d�d܄Zǐd.d�dބZ�e�e�eeTdߜd�d�Z�d�d�Z�d�d�Z�d�d�Z�d�d�Z�d�d�Zΐd/d�d�Z�d�d�Z�d�d�Z�d�d�ZҐd0d�d��Z�d�d��Z�d�d��Z�d�d��Z�da�da�da�da�da�d�d��Zܐd1d��d�Zݐd�d�Zސd2�d�d�Zߐd�d�Z�d�d�Z�d	�d
�Z�d�d�Z�d
�d�Z�d�d�Z�d�d�Z�d�d�Z�d�d�Z�d�d�Z�dS(3�)�print_function)�absolute_import)�division)�range)�
basestring�unicode)�	iteritems�reraise)�Dict�List�Tuple�Optional�TextN)�partial)�PAGESIZE)�acquire_lock)�Feature)�admins�reseller_users�get_main_username_by_uid�is_panel_feature_supported)�
EncodingError)�reseller_limits_disabled_post�reseller_limits_enabled_post)�exec_utility)�get_xml_config�save_xml�BadVeConfigException)�setup_global_lock�LockFailedException)�Lve�PyLve�LVP_XML_TAG_NAME�
PyLveError�NameMap)�detect_panelclass)�write_file_via_tempfile�create_dir_securez%/usr/bin/getcontrolpaneluserspackagesz
/proc/cpuinfoi'Z
VE_DEFAULTFz/bin/umountz"/etc/container/exclude_mounts.conf�multiZsingleZPYLVE_DEBUG�plesk_id)�debug)�pycCst��rtj��SgSdS)z-Helper function for easy mocking in unittestsN)�lve�reseller_limit_supported�procZlvp_id_list�r/r/�lvectllib.py�lvp_listOs
r1cCst�}|��|��S)zf
    Get list of resellers with activated reseller limits

    :return: list of pairs (name, uid)
    )r$�
link_xml_nodeZload_from_node)�name_mapr/r/r0�get_active_resellersWsr4cCs|dd�t�D�kS)z]
    Check whether giver reseller has activated reseller limits or not
    :return: bool
    css|]\}}|VqdS)Nr/)�.0�name�uidr/r/r0�	<genexpr>gsz,is_active_reseller_limits.<locals>.<genexpr>)r4)�
reseller_namer/r/r0�is_active_reseller_limitsbsr:)�ecCs>t��st|�tr(tddt|�g�nt|�t�d�dS)a"
    Since cPanel user can corrupt config file for some user with wrong encodings,
    we want to notify him that he should fix encoding problems with the link to documentation.
    Print error message and exit with code 1 or raise given exception if it isn't cPanel.
    :return: None
    r(�ERROR�N)	�cldetectlib�	is_cpanelr	�JSON�json_format�str�print�sys�exit)r;r/r/r0�raise_cpanel_encoding_errorksrFcCs0yt|�Wntk
r*tdd�YnXdS)az
    ~~~~~~~~~~~~~~~~~~
    !!! DEPRECATED !!!
    ~~~~~~~~~~~~~~~~~~
    Please, use setup_global_lock instead if possible

    Wrapper over setup_global_lock. If lock cannot be set,
    it will write message and close app

    The only reason why it is here is legacy function
    check_result_and_exit that we use in TWO places
    :type write: bool
    :return: Nothing
    r=zcan`t get lockN)rr�check_result_and_exit)�writer/r/r0�get_global_lock}srIcCsN|tjkrJ|dkrJtr0ttddt|�g�ntdt|��t�|�dS)Nrr<z
lvectl: %szlvectl: Error: )	�errnoZENOSYSr@rA�MULTI_FORMATrBrCrDrE)�result�messager/r/r0rG�s
rG�r=�ii)�cpu�ncpu�io�ep�mem�pmem�nproc�iops�drQrPrRrTrUrVrWrS���falsecCs|adS)N)r@)Z	json_flagr/r/r0�set_json�sr\cCs|adS)N)�
BYTES_FLAG)Z
bytes_flagr/r/r0�	set_bytes�sr^c	Cs�trtdkrd}n tdkr d}ntdkr.d}nd}ddd	d
ddd
gddd	d
ddd
dgddd	d
ddd
gddd
dd
gd�|}tr�|�d�d}|d|�dg||d�S|S)N��noiops_8�8��6�4�ID�SPEED�PMEM�VMEM�EP�NPROC�IO�IOPS)r`rarcrdr=�CPU)�NOIOPS�LVE_VERSIONr@�index)�version�fieldsZ	speed_idxr/r/r0�
get_fields�s rscCsR|dkr.t��}|ddkr$td��|dan|at��}|sFtd��t��adS)Nrzget_lve_version failedzinit_lve() failed.)�clcommonZget_lve_version�	Exceptionro�pylve�
initialize�liblve_settings�lve_settings)Zlve_verZ
lve_statusr/r/r0�init�s
rzc
Csftj�d�}|dkrdSy
t�|�Sttfk
r`}ztdt|��t	�
d�Wdd}~XYnXdS)z-Get cpu information from environment veriableZCPU_DATANz.Invalid environment variable 'CPU_DATA' formatr=)�os�environ�get�json�loads�	TypeError�
ValueErrorrCrBrDrE)Zpacked_cpu_datar;r/r/r0�_get_cpu_data_from_envs
r�cCs�i}i}d}yttd�}Wn2tk
rLtdtt�d�t�d�YnXx||D]t}|��sz||d|<|d}i}qTt|�	d��dkr�|�	d�d��||�	d�d��<qTd	||�	d�d��<qTW||d
dgS)zF
    Parse /proc/cpuinfo

    return [NumProc, frequency in MHZ]

    r�rzlvectl: Error: Can`t open �.r=zproc%s�:�rZZproc0zcpu MHz)
�open�CPUINFO�IOErrorrCrBrDrE�strip�len�split)ZcpuinfoZprocinfoZnprocs�f�liner/r/r0�get_cpu_datas"
&r�cCs�t|�}|pd}t}t|d�}t�d|�}|dk	r~t|�}|dkr~|dkr~tttd||��}|dkrp|St|t|�SdS)z�
    Try converting to kernel format from old CPU format (percentage of whole
    cpu) and optionally the NCPU format. Return whichever is less.

    :param data: string presumably in old CPU format
    :param lncpu: integer number of cores limit
    rz
\d{1,2}0?$NrX)rB�CPUINFO_DATA�int�re�match�round�CORE_WEIGHT�min)�data�lncpu�cpu_datarQZcpu_percentZfrom_cpu_limitr/r/r0�convert_from_old_cpu9sr�cCstt|�}t}t|d�}t�d|�}|dk	rpt|�dd��}||dkrP|d}|dkrltttd|��SdSdS)zd
    Try converting cpu limit from SPEED in percentage of one CORE format to
    kernel format.
    rz\d+(?:\.\d+)?%$N�%rZrX)	rBr�r�r�r��float�replacer�r�)r�r�rQZpercentr/r/r0�convert_from_speed_percentOsr�cCs�t|�}t}t|d�}t|d�}t�dtj�}|�|�}|dk	r�|�d�}t|�d��}|�	�dkrp|d}|||kr�||}|dkr�tt
|t|��SdS)	zQ
    Try converting cpu limit from SPEED in mhz/gzh format to kernel format.
    rr=z,(?P<freq>\d+(?:\.\d+)?)(?P<suffix>mhz|ghz)+$N�suffix�freqZGHZi�)rBr�r�r�r��compile�
IGNORECASEr��group�upperr�r�)r�r�rQZcpu_freq�patternr�r�r�r/r/r0�convert_from_speed_hzas 

r�cCst|�pt|�S)z}
    Try converting cpu limit value from either SPEED limit format
    (percentage of CORE or mhz/ghz) to kernel format.
    )r�r�)r�r/r/r0�convert_from_speedvsr�cCsBt||�}|dk	r|St|�}|dk	r*|St|�}|dk	r>|SdS)z�
    Convert different variants of cpu limit to kmod ver 8 variant

    :param data: Value in old CPU format or SPEED with % or mhz/ghz.
    :param lncpu: Limit in old NCPU format.
    :return: CPU limit in kmod ver 8+ format or None for bad format
    N)r�r�r�)r�r�Zfrom_cpuZfrom_speed_percentZ
from_speed_hzr/r/r0�convert_to_kernel_format�s
r�cCs^t}t|d�}t|�}d|krBdtttt|�d��|���Stttt|�|���SdS)zm
    convert speed to old cpu format
    args: cpu limit in speed value
    return: old cpu limit format
    r�*N)r�r�rBr��lstrip)�speedr�rVr/r/r0�speed_to_old_cpu�s"r�c	csft��t��}yt�||�Wn(tk
rHt�tjd|��YnXz
|VWdt�|�XdS)zF
    Run subprocess in lve with pseudo-random id and given limits
    zKUnable to setup lve with id %i, something is wrong, check dmesg for detailsN)rvrwZget_available_lve_id�	lve_setupr#�syslog�	LOG_ALERT�lve_destroy)Zsettings�lve_idr/r/r0�
temporary_lve�s

r�cCsJt��}t||d�|_||_||_||_||_tt	�
d|t��|_|S)a
    Just a nice user-friendly constructor of liblve_settings object
    You can pass the following ls_cpu and ls_cpus values:
    - in percents of one core (just ls_cpu='75%', ls_cpus will be ignored)
    - in old 'CPU' format (two arguments, ls_cpu and ls_cpus required, both int)
    )r�g�?)
rvrxr��ls_cpu�ls_io�	ls_enters�ls_nproc�ls_iopsr��mathZceilr�
ls_memory_phy)r��ls_cpusr�r�r�r�r��sr/r/r0�make_liblve_settings�s	r�cCs0t|�d��}|s(|�d�}t�|�j}t|�S)N�id�user)rB�getAttribute�pwd�getpwnam�pw_uidr�)�ve_lve_elementZuser_uidZ	user_namer/r/r0�get_ve_lve_user_uid�s

r�cCsLdt|d�i}|tkr(t|d�|d<|dk	r:|�|�tt�|��dS)a�
    Print output in json as:
    {"status": "ERROR/OK", "msg": "Some Message", "ext1": "foo", "ext2": "bar"}
    where "status" and "msg" field are mandatory

    :param str error_type: Either MULTI_ERROR or SINGLE_ERROR
    :param list data: List with a status string and a message string
    :param dict extensions: Some additional fields for the final json object
    :return: None
    Zstatusrr=�msgN)rBrK�updaterCr~�dumps)Z
error_typer��
extensionsrLr/r/r0rA�s
rAc
Cs�y tt�|�d�d��||<Wn`tttfk
r�||||<|�|�}|�dt	||��y|�
|�WnYnXYnXdS)Nr�limit)r��
ve_default�getElementsByTagNamer�r��
IndexErrorr��
createElement�setAttributerB�appendChild)�xml�ve_defaults�ve_cfg�val�default�noder/r/r0�check_def_value�s 
r�cs�fdd�|jD�S)Ncs&g|]}t|tj�r|j�kr|�qSr/)�
isinstancer�ZElementZtagName)r5�_)�tagr/r0�
<listcomp>�sz"xml_filter_tag.<locals>.<listcomp>)Z
childNodes)r�r�r/)r�r0�xml_filter_tag�sr�cCsFx@t||�D]2}|dk	r$|�|�s$q|dk	r<|�|�|kr<q|SWdS)N)r�ZhasAttributer�)r�r��attrZattr_valZ
child_noder/r/r0�xml_filter_first�sr�cCs&t|||d�}|dkrt��|�|�S)N)r�r�r�)r�r�r�)r�r�r�Zfiltered_child_noder/r/r0�get_child_tag_atrrsr�cCs"t||�d}|�|t|��dS)z�
    Find in children nodes node with tag and setup attribute
    insted el.getElementsByTagName not search recursiveli in tree
    rN)r�r�rB)r�r�r�r�Zfirst_child_noder/r/r0�set_child_tag_atrr	sr�c
Csdyt�\aaWnPtk
r^}z2tr8ttdt|�g�ntt|��t	�
d�Wdd}~XYnXdS)zLoad config from ve.cfgr<r=N)rr��ve_lveconfigrr@rArKrBrCrDrE)r;r/r/r0�_load_config_wrappersr�)�lvp_id�lvp_defaultscCs�day\|sttd�datS|r.t�d�atSttt	dt
|��}|rT|�d�dan
t�t�atSt
k
r�tr�tdddg�t�d�ntd	�YnXtS)
z�Load default limits

    :param int lvp_id: lvp id
    :param bool lvp_defaults: load reseller's default limits instead of global
    :return: dict with default limits
    �true�defaultsrr�r(�WARNINGzdefault section error in ve.cfgr=z(warning: default section error in ve.cfg)�ubcr�r�r��LVE_DEFAULTr�r��LVP_DEFAULTr�r"rBr�Z	cloneNoder�r@rArDrErC)r�r�Zdefaults_root_noder/r/r0�_load_default_limits!s&


r�cCstdd�ttttttfD��S)Ncss|]}|dkVqdS)rZNr/)r5�xr/r/r0r8Hsz._all_config_elements_loaded.<locals>.<genexpr>)�all�ve_lve�ve_lvp�
ve_package�	ve_binary�ve_enter_by_name�ve_cfg_versionr/r/r/r0�_all_config_elements_loadedGsr�cCs�t�d�at�d�at�t�atjj�	t�t�d�}t
|�dkrL|dant�
d�at�t�t�d�at�d�}t
|�dkr�t|djj�ndadS)	z$Load all config elements from ve.cfgr,�packagez
enter-by-namer�binaryrqr=N)r�r�r�r�r"r�r,�mapr3r2r�r�r�r�r�r�r�Z
firstChildZ	nodeValuer�)Zenter_by_name_elemsZcfg_version_elemsr/r/r0�_load_config_elementsJs	








r�)�default_limitsc
Cs�iattdttt|d�}|dd�y,t�d�d�d�}t|tdd�td<Wnltt	t
fk
r�t|dtdd�td<t�d�}|�dt
|d��yt�|�WnYnXYnXy tt�d	�d�d
��td<Wnbtt	t
fk
�r>|dtd<t�d	�}|�d
t
|d��yt�|�WnYnXYnX|dd�td
k�r^|dd�n|ddtid�|dd�|dd�|dd�t�dS)z~Create ve_defaults dict with default values for all limits

    :param dict default_limits: default limits for lve or lvp
    N)r�r�r�r�r�rQ)r�rPrr�)r��other�
maxentryprocsrSrR�rT)r�r�rUrVrW)r�rr�r�r�r�r�r�r�r�r�r�r�rBr�r�ro�MEM_DEFAULT_CL5�_check_defaults_for_nones)r�Z
check_def_valr�rPrSr/r/r0�_load_ve_defaultsos>	

 

$


r�cCs`xZt��D]N\}}|dk	rq
d|�d�}tr<tdd|g�ntj�|�d��t�d�q
WdS)z*Check that all default values are not NoneNzERROR: Incorrect z default valuer(r<�
r=)r��itemsr@rArD�stderrrHrE)�key�value�err_msgr/r/r0r��sr�TcCs,t�t||�}|st�s t�t|�dS)z�
    :param bool lvp_defaults: load reseller's default limits instead of global
    :param int lvp_id: lvp id to load customise defaults
    N)r�r�r�r�r�)r�r��load_config_elementsr�r/r/r0�get_XML_cfg�s


rc
CsFytt||dd��}|||<|Stttfk
r@t||�SXdS)Nr�)r�r�)r�r�r�r�r�)r��elr��
setup_datarr/r/r0�check_value�sr	cCs*t��}|��|�|�}t||d�dS)a5
    This function is a pure workaround for our ugly globals-based API which
    should be fixed partially with LU-496, because there is no clean way
    to retrieve reseller's data from ve.cfg without touching globals
    :param reseller: reseller name
    :return: Nothing. It just updates some globals
    )rN)�lveapir$r2Zget_idr)�reseller�xml_config_load_elementsr3Zreseller_idr/r/r0�_load_resellers_xml_data�s	
r
c
s�t�t�a�dk	�r�t�}�dk	rF�|krF�|�krF��fdd�}n�t��r�|dk	r�y8t�|�}d|j�d�}t	|��}|�
�}WdQRXWntk
r�d}YnXd|kr�t�tj
d�|d	���fd
d�}x�tD]�}	||	�r�td|	tt�}
y&ttt|	dd
d�|
d��}|td<Wntttfk
�r4YnXtd|	tt�tdk�r^td|	tt�ndtd<ytt|	ddd��}|td<Wntttfk
�r�YnXtd|	tt�td|	tt�td|	tt�q�WdS)a
    Put limit values that will be applied later in a global variable `setup_ve`.

    :param plan_id: package
    :param reseller:
        If reseller is None we only inherit from admin packages.
        In that case we ignore all tags in ve.cfg with a "reseller" attribute.
    Ncs|�d��ko|�d��kS)Nr�r)r�)r)�plan_idrr/r0�<lambda>��z$prepare_setup_data.<locals>.<lambda>z"/usr/local/directadmin/data/users/z
/user.confrZzpackage=customzNPackage for user with id {} is incorrect, please recover it using Note from {}zYhttps://docs.cloudlinux.com/cloudlinux_os_components/#installation-enabling-and-disablingcs|�d��ko|�d�S)Nr�r)r�)r)rr/r0r�rrQrPr�)r�r�)r�rRr�rTrr�r�rSrVrUrW)�copyr�r�get_reseller_packages_mapr>�is_dar��getpwuid�pw_namer��readrur�r��formatr�r	r�r�r�r�r�r�r�)
rrr�Zres_pkg_dict�is_needed_planZuser_pwd�filenamer��textrr�rPrSr/)rrr0�prepare_setup_data�sP






rc	CsZy,tjtd|gtjtjd�}|��|jdkStk
rTtddtd|f�YnXdS)Nz-l)�stdoutrr���zfailed to run "%s %s %s")�
subprocess�Popen�UMOUNT�PIPEZcommunicate�
returncode�OSErrorrG)�path�pr/r/r0�
umount_dirs
r&c	CsBtj�t�sdSg}yDttd��0}x(|D] }|��}|r(|�t�|��q(WWdQRXWn"t	k
rzt
ddt�YnX|s�dSt�tj�y,tdd��}dd�|�
�D�}WdQRXWn"tt	fk
r�t
dd�YnXd	}xXt|�D]L}d
}x<|D]4}	x.|D]&}
|
�|	�}|�rt|	��p&|}P�qWq�W|s�Pq�WdS)zy
    Unmount all paths from /proc/mounts that match regular expressions from /etc/container/exclude_mounts.conf file
    Nr�rzfailed to read %sz/proc/mountscSsg|]}|��d�qS)r=)r�)r5�mr/r/r0r�7sz"prepare_mounts.<locals>.<listcomp>zfailed to parse /proc/mounts�
F)r{r$�isfile�EXCLUDE_MOUNTS_CONFr�r��appendr�r�r�rG�unshareZCLONE_NEWNS�	readlinesr�r�searchr&)Zreg_exp_listZconfr�r�r�ZmountsZATTEMPTSr��errorZmountZreg_expr'r/r/r0�prepare_mounts#s<
" 


r0cCsNd}ytj|ddd�Wntk
r6td|�YnXt�tjdd�dS)	zI
    Start LVE engine and initialize default mount namespace for LVE
    z,/bin/mount --make-rprivate / >/dev/null 2>&1Tz	/bin/bash)�shell�
executablezError: failed to executezCan`t init lve default settings)rN)rZcallr#rCr0rv�	lve_start)Z	MOUNT_CMDr/r/r0r3Gsr3cCstj|d�|�|d�dS)zs
    Create LVE container for given ID
    :type lve_id: int
    :type ignore_error: bool
    :return: Nothing
    z8lvectl: Can`t create lve with id {}; error code {{code}})r�ignore_errorN)rv�
lve_creater)r�r4r/r/r0r5Tsr5cCstj|d�|�|d�dS)zs
    Create LVP container for given ID
    :type lvp_id: int
    :type ignore_error: bool
    :return: Nothing
    z8lvectl: Can`t create lvp with id {}; error code {{code}})rr4N)rvZlve_lvp_creater)r�r4r/r/r0�
lvp_create`sr6cCs8t�}x,t�D]"}tj|d�|�d�|�|�qW|S)Nz9lvectl: Can`t destroy lvp with id {}; error code {{code}})r)�listr1rv�lve_lvp_destroyrr+)Zdestroyed_listr�r/r/r0�destroy_lvp_alllsr9cCs(|dkrt�ntj|d�|�d�dS)Nr�z9lvectl: Can`t destroy lvp with id {}; error code {{code}})r)r9rvr8r)r�r/r/r0�lvp_destroyusr:cCs�d�|�}d�|�}d}|dkrvt��r2tt��}tttj����dkrpx*tj��D]}tj	||d�d}qTWq�d}nJtj�
|�s�tj��r�tj�|�dk	r�tj	||d�d}ntj�
|�s�d}|r�t
r�ttdg�n"t
r�ttd	|g�ntd
�|��dS)Nz/Can't remove lve {} from kernel - error code -3z1Can`t destroy lve with id {}; error code {{code}}Fr�r)rTZOKZWARNzwarning: {})rr,r-�boolr9r�r7r.�lve_id_listr�Zcheck_inside_listZresellers_supported�detect_inside_lvpr@rA�
SINGLE_FORMATrKrC)r�Zcant_remove_msgZcant_destroy_msgZ	destroyedZid_r/r/r0r�~s0




r�cCsp|rtj�|�st|�t����Dt�d�y�ttd�t	_
ttd�t	_ttd�t	_ttd�t	_
ttd�t	_tdkr�ttd�t	_ttd	�t	_td
kr�ttd�t	_|r�|dkr�tj|t	d
�|�d�ntj|t	d�|�d�n0|dk�rtjt	dd�ntj|t	d�|�d�WnBtk
�r`}z"tddt|�dt|��Wdd}~XYnXWdQRXdS)Nr/rRrPrQrTrSr�rUrVrbrWrz'Can`t setup default settings for LVP {})rz/Can`t setup lvp with id {}; error code {{code}}zCan`t setup default settingsz/Can`t setup lve with id {}; error code {{code}}r=zCan`t setup lve z. RuntimeWarning excepted: )r,r.�	exist_lvpr6�warnings�catch_warnings�filterwarningsr�rryr�r�r�Z	ls_memoryr�ror�r�r�rv�lve_set_defaultrZ
lve_lvp_setupr��RuntimeWarningrGrB)r�r�Zrwr/r/r0r��s6

r�cCs�d}d}t}td�t}|at|t�r�y||d}|s<d}Wntk
rVd}YnXy||d}|snd}Wntk
r�d}YnX||fS)z�
    Get pair of package, reseller for lve_id
    :param lve_id: lve_id, UID with package, reseller
    :return: tuple of (package, reseller); Both can be None
    Nz
list-usersrr�)�packages_users�GetControlPanelUsersr��dict�KeyError)r�r�r�old_packages_usersZtemp_packager/r/r0�"get_package_and_reseller_by_lve_id�s(


rJz+/var/run/cloudlinux/effective-normal-limitsz;/opt/cloudlinux/flags/enabled-flags.d/burstable-limits.flagcCs�tj�t�sdSy8tj�t�}tj�|�sFtj�|�}t|ddd|�Wn8tk
r�}zt	d|�d|����Wdd}~XYnXt
t�d��}|��|r�i}n�y:tj�t�r�ttddd	��}t
�|�}WdQRXni}Wntt
jk
�r}zt	d
t�d|����Wdd}~XYn:tk
�rP}zt	dt�d|����Wdd}~XYnX|�t�ytt
�|�td�Wn8tk
�r�}zt	dt�d|���Wdd}~XYnXWdQRXdS)
ai
    Save calculated effective normal limits for an LVE to a cache file.

    The cache file is stored in /var/run (tmpfs) for faster operations.
    It disappears on reboot, but lvectl apply all runs in the lvectl systemd service anyway,
    so the file will always reappear if the service functions normally.

    If additional speed is desired, consider replacing the standard JSON lib
    with a faster one like rapidjson or orjson.

    :param reset: If True, recreate the cache file from scratch with data available in the dict,
    instead of updating it, defaults to False
    :type reset: bool, optional
    Ni�rz#Error: failed to create the folder z: z.lockr��utf8)�encodingz=Error: failed to parse the effective normal limit cache file z<Error: failed to read the effective normal limit cache file z<Error: failed to save the effective normal limit cache file )r{r$�exists�BURSTABLE_LIMITS_FLAG_FILE�dirname�EFFECTIVE_CACHE_FILE�isdirr'r#rCrr)r�r~�loadZJSONDecodeErrorr��CACHED_EFFECTIVE_LIMITSr&r�)�resetZ
effective_dirZvar_run_dirr;Zeffective_cache_lockZeffective_limitsZreadfiler/r/r0�write_effective_cache�s<
rUcCsttt|�<dS)a�
    Cache the calculated effective normal limits for an LVE to a dictionary.

    setup_data is a global dict that contains the LVE configuration last applied.
    Holds only one entry - during operations like lvectl apply all, we enter this
    function repeatedly with setup_data containing info for one processed LVE each time.

    These limits can be used by the Burstable Limits components
    to set/reset burst or normal limits without having to go
    through effective limit calculation or invoking lvectl as a middleman.
    Comparing the cached limits to current active limits (in kernel)
    also shows whether or not the LVE has burst limits active at the moment.

    This function gets called in lve_apply, which means it also runs within:
    * lvectl apply all
    * lvectl apply-many
    * lvectl set
    * and others, that also call lve_apply

    :param lve_id: LVE ID for which the limits are being saved.
    :type lve_id: int
    N)rrSrB)r�r/r/r0�cache_effective_limits1srVc
st}�dkr td��t|}atdkr2t|d�d}|dk	rD|}n$|rLtnt}	tt�fdd�|	�d�}|dkr�y|�d}Wntt	fk
r�d}YnXy|�d	}Wntt	fk
r�d}YnX|dk	�r�t
||d
�td|tt
�}
ytt|dd
d�|
d�t
d<Wntttfk
�r*YnXt
ddk�rFtdt
d<td|tt
�tdk�rntd|tt
�ndt
d<ytt|ddd��t
d<Wntttfk
�r�YnXtd|tt
�td|tt
�td|tt
�n t
||�d�tdk�r�dt
d<|�st��t�|d�dS)a

    Aplly limits to LVE lve_id
    :param lve_id: lve id
    :type lve_id: int
    :param plan_id: package for user with lve_id. deprecated
    :type plan_id: string
    :param result: if True = don't apply limits. only create setup_data with actual limits
    :type result: boolean
    :param reseller: if True = plan_id is resellers plan. deprecated
    :type reseller: boolean
    :param out_node: node with limits for lve_id
    :type out_node: xml_node
    :param lvp_id: reseller container id; host container if 0
    r�useridrZ)r�Ncst|d��kS)N)r�)r�)r�)r�r/r0rorzlve_apply.<locals>.<lambda>r�r)rrQrPr�)r�r�)r�rRr�rTr�r�rSrVrUrW)rr�r[)rErFr�rr�r��next�filter�	NameErrorrHrr	r�rr�r�r�r�r�r�r�rVr�)r�rrLr�out_noder�rIZnew_packages_usersr�	node_listr�r/)r�r0�	lve_applyNs`







r]cCst|�dS)a�
    Sometimes python is confused about which encoding
    we should use to print message. E.g. when we use
    "command | grep" to find something in the output.

    According to python docs, "Software should only work with Unicode
    strings internally, converting to a particular encoding on output."

    So let's convert all unicode into str using our "magic" byteify method
    :param string:
    :return:
    N)rC)�stringr/r/r0�_print_string_safe�sr_cGs2td�dd�|dd�D�d|dg��dS)zz
    Command to print data with only last column 30 symbols wide
    Useful to print data that contains package name.
    rZcSsg|]}d|�qS)z%8sr/)r5r�r/r/r0r��sz_pprint.<locals>.<listcomp>Nrz%30s)r_�join)rrr/r/r0�_pprint�sracGs<td�dd�|dd�D�d|dd|dg��dS)z�
    Command to print data with two last columns 30 symbols wide.
    Useful to print full data of every user with package name and reseller name.
    rZcSsg|]}d|�qS)z%8sr/)r5r�r/r/r0r��sz_pprint_f.<locals>.<listcomp>N���z%30sr)r_r`)rrr/r/r0�	_pprint_f�srccGs2td�d|dgdd�|dd�D���dS)ze
    Command to print data with first column 30 symbols wide.
    Useful to print packages data.
    rZz%30srcSsg|]}d|�qS)z%8sr/)r5r�r/r/r0r��sz_pprint_p.<locals>.<listcomp>r=N)r_r`)rrr/r/r0�	_pprint_p�srdcGs@td�d|dgdd�|dd�D�d|dg��dS)	z�
    Command to print data with first column and last column are 30 symbols wide.
    Useful to print data with user names and package names.
    rZz%30srcSsg|]}d|�qS)z%8sr/)r5r�r/r/r0r��sz_pprint_r.<locals>.<listcomp>r=rN)r_r`)rrr/r/r0�	_pprint_r�srecCsft|�}t|�ttfkr&|�d�r&dnd}|�dd�}|rLt|�}|d9}nd}|r^d�|�n|}|S)z�
    Convert pmem or vmem limits to bytes value

    :param value: pmem or vmem limits in kbytes value
    :return: bytes value of limit

    r�TFrZirz*{})rB�typer�
startswithr�r�r)r�was_changedr/r/r0�_pmem_vmem_to_bytes_value�s	"
ricCs�d}d}t|t�r,|�d�r,|dd�}d}yt|�}Wntk
rLdSX|rVd}|dd}|d	krxd
�||�}nd�||d�}|S)z�
    Convert amount of RAM to M format

    :param string value: amount of memory in KB
    :rtype: string
    :return: amount of memory in MB like "1234M"
    rZFr�r=NTrYirz{}{}Mz{}{}K)r�rrgr�r�r)rrLrh�vr/r/r0�_mb_mem�s rk�0r�c
st����dd�d<�fdd�}|d�dkr<t|d��nd}dd�}|t|d��t|�t|d��t||d���t||d	���t|d
��t|d��t|d��t|d
��|d��g}t�}|dk	r�||7}tr�d��fdd�|D��}	d|	g}n||�|�fdd�|D��|S)z^
    Generate header and default package data either as print to stdout or as json string
    rPrXcs��|d�S)NrZ)r})r)r�r/r0rrz_formatter.<locals>.<lambda>rZcSstrt|�St|�S)N)r]rirk)rr/r/r0rsrQrUrTrSrVrRrW)rerfrmZNCPUrgrhrirjrkrl�PACKAGEN�,c3s"|]}d|��|d�fVqdS)z	"%s":"%s"rZN)r})r5r�)�
fields_mapr/r0r8$sz_formatter.<locals>.<genexpr>z{%s}csg|]}��|d��qS)rZ)r})r5r�)ror/r0r�(sz_formatter.<locals>.<listcomp>)r�rr�rBrsr@r`)
�printer�
default_idZdefault_package�more_fields�get_dataZ_cpu�convert_mem_limits�resrrr�r/)r�ror0�
_formatter
s(rvcs��fdd�}|S)aA
    Generate inner function with closured fields names and printer function

    :param list fields: List of strings that represent names of fields in final output
    :param callable printer: Function to format and print data for every entry
    :rtype: callable
    :return: function to format data for every user
    csd�t|d}t|d}|dkr(d}tdkr>t|dd�n,|dk	rTt|dd�n
tdd	�t||d�t�t��t|||d
d��fdd
�}dd
�}t|��d<t|d���d<t|d���d<t||d����d<t||d����d<t|d���d<t|d���d<t|d���d<t|d���d<t	�r@t
|��d<n|�d<|dk�rft	�r\dnd�d<n|�d<d�dk�r�dtt�d�d��d��d<nt�d�d�d<�d�d<tt
�d���d<g}t	�rd��fdd��D��}d |g}n��fd!d"��D��|S)#z�
        :param string user: Find and format data for this User ID
        :rtype: list
        :return: List of given user's statistics data line or empty list
        rZr�rNr=)rF)r)rT)rrrLcs4t�|�tt|�kr(dtt|�St�|�S)Nr�)rBr)r)r�r/r0rSsz2_user_formatter.<locals>.wrapper.<locals>.<lambda>cSstrt|�St|�S)N)r]rirk)rr/r/r0rUsr�rPrQrUrT�vmemrSrRrVrWzN/Ar�rXr�rnc3s"|]}d|�|��fVqdS)z	"%s":"%s"N)�lower)r5r�)r�r/r0r8rsz3_user_formatter.<locals>.wrapper.<locals>.<genexpr>z{%s}csg|]}�|���qSr/)rx)r5r�)r�r/r0r�usz4_user_formatter.<locals>.wrapper.<locals>.<listcomp>)rEr�rr
rrrr]rBr@�_normalize_strr�r�r�r`)r�r�rZ
check_changedrtrur�)rrrp)r�r0�wrapper5sR


$z _user_formatter.<locals>.wrapperr/)rrrprzr/)rrrpr0�_user_formatter,s	Br{cCs�t�yBtd�|dk	r(|r(td|�n|dk	r>td|d�ntd�WnYnXdg}tt|d�}t�|}t|�}xtD]}|||�7}q�Wtr�tdd�	|�d	�dS)
Nz
list-usersrWzlist-reseller-users)rrm)rrz	{"data":[rnz]})
rrFrvrarsr{rEr@rCr`)rWrrrrLrr�	formatterr�r/r/r0�paneluserslimits|s$

r}cCsBtd�g}x0tt�D]$\}}|�t|�|d|df�qW|S)z@Get list of tuples[lve_id, reseller, package] from control panelz
list-usersrr�)rFrrEr+r�)rLZstr_uidZpayloadr/r/r0�paneluserslist�s
 r~cCs>ddlm}i}td�tt���||<td�|�t�|S)zBGet dict of pairs[provider, list[package_name]] from control panelr)�DEFAULT_PROVIDERz
list-packageszlist-resellers-packages)Z
clveconfigrrFr7rE�keysr�)r�packagesr/r/r0�panelpackagesdict�s
r�cCspt�td�ttddgd�}t�ddg}t|td�}xtD]}|||�7}q>Wtrltdd�	|�d�d	S)
z]
    Implements lvectl all-user-list command

    :return: None, prints result to stdout
    z
list-usersrm�RESELLER)rr)rpz	{"data":[rnz]}N)
rrFrvrcrsr{rEr@rCr`)rLrrr|r�r/r/r0�all_users_limits�s
r�cs6t��dkr2dd�t�D���fdd�t|�D�}|S)z�
    Filtering DirectAdmin's admins for `lvectl apply all` command

    :param ve_dict: dict with LVE
    :return: filtering dict

    ZDirectAdmincSsg|]}t�|�j�qSr/)r�r�r�)r5r�r/r/r0r��sz(_filtering_da_admins.<locals>.<listcomp>csi|]\}}|�kr||�qSr/r/)r5rr)�uids_da_adminsr/r0�
<dictcomp>�sz(_filtering_da_admins.<locals>.<dictcomp>)r>Z	getCPNamerr)�ve_dictr/)r�r0�_filtering_da_admins�s	r�c
s>y8t�tt�}|r6tjj����fdd�t��D�}Wnt�}YnX|dkr`t}t	�}nt
}tjj|d�}i}x |D]}|dd�|t
|d�<q|Wx"|D]}||kr�ddd�||<q�Wt|�dk�r:t|�}xf|D]^}||kr�d}n||d}||}t|�}	t|	�dk�r(||	dd�||<q�|dd�||<q�W|S)	Ncsi|]\}}|�kr||�qSr/r/)r5�krj)�cfg_lvp_id_listr/r0r��sz&prepare_apply_data.<locals>.<dictcomp>T)r�)r�r)r�rr�)rFrGrEr,r�r3�id_listrr�r1r�r.r<r�r�r��guess_reseller_by_package)
r�Zpackages_users_r\r�r�r�r�r7Zpkg�	resellersr/)r�r0�prepare_apply_data�s>


r�c
Csxt�td�t�}t��r(tt���ni}tjjdd��Fx>|�	�D]2}|�
|d�dkrFt|||d||dd�qFWWdQRXtdd�t��s�dStd�}tj�
�}x�|�	�D]�}t|dd	d
�t|||d||d|d�t|d	d�td|d
�tj
�|�}x`tj�|�D]P}t�|�j}|�
|d�|k�r^tjj||d|�d|�d�d�t||d��qWq�WdS)Nz
list-usersT)r4rr�r)r[r)r�F)r�r�r)r�r[rr�)r�r)r�r�zCan`t move lve_id=z to lvp_id=z; error code {code})r)r�r)rrFr�r,r-rGZlve_id_lvp_id_pairsr+Zcontext_ignore_errorr�r}r]r.r��get_reseller_namertZcpapirr�r�r��lve_lvp_move)r�Zlve_lvp_mapr�Zkernel_mappingZlvp_id_r9r��lve_id_r/r/r0�
lve_apply_all
s<,



r�c
Cs�td�t|d�x�tD]�}t|d�|krtj�|�}xd|dd�D]T}t�d|�yt	�
|�WqDtk
r�tj�
|�r�t�|�|�|�YqDXqDWt|�|j�|�tt�tdd�x|D]}t|�q�Wt|d�dSqWdS)z,Remove reseller from ve.cfg and from procfs.T)r�)r�NrF)rIrr�r�r,r.Zmap_lve_id_listrvr�r�rrHr+Z
lve_existsr��remover:�
parentNode�removeChildrr�r])r�r�usersr�r/r/r0�_remove_reseller9s,





r�cCsHt|�rtj|d�n.tr8tddd|g�t�d�ntd|�dS)z&Disable reseller limits and call hooks)rr(r�z!no configuration found for LVP %srz*warning: no configuration found for LVP %sN)r�r�throw_eventr@rArDrErC)r9r�r/r/r0�disable_reseller_limitsVsr�cCs�td�t�d}xPtD]H}t|d�|krd}t|�t|�|j�|�tt	�t�t
|�qW|s�tr�tddd|g�t
�d�ntd|�dS)	NTF)r�r(r�z no configuration found for VE %srz)warning: no configuration found for VE %s)rIrr�r�r�r5r�r�rr�r]r@rArDrErC)r��Deletedrr/r/r0�
lve_deletecs"
r�cCs6tj�d�s2tr tdddg�ntd�t�d�dS)Nz/proc/lve/enterr(r�zenter by name not supportedz$warning: enter by name not supportedr)r{r$rMr@rArCrDrEr/r/r/r0�lve_enter_checkxs
r�cCsDt�y,||��}tdd�}|�|�|��WnYnXdS)Nz/proc/lve/enter�w)r�r�r�rH�close)Zsignr�r�r�r/r/r0�enter_apply�s

r�cCs�t�trfd}d}x@tD]8}|�d�}|r@|d|d7}d}q|d|d7}qW|d7}t|�n$td�xtD]}t|�d��qtWdS)	Nz	{"data":[Tr$�"Fz,"z]}ZBinaries)rr@r�r�rC)rL�firstrr$r/r/r0�
list_binaries�s



r�cCs(t�xtD]}td|�d��qWdS)N�+r$)rr�r�r�)rr/r/r0�
load_binaries�s
r�cCsBt�tdd�}|��}|��x|D]}td|�q&Wt�dS)Nz/proc/lve/enterr��-)r�r�r-r�r�r�)r��lines�lr/r/r0�reload_binaries�s

r�cCs�td�t�t�d}xBtD]:}|�d�|krd}td|�|j�|�tt	�t�qW|s�t
rxtddd|g�ntd|�t
�d	�dS)
NTFr$r�r(r�zno configuration found for %sz&warning: no configuration found for %sr)rIr�rr�r�r�r�r�rr�r@rArCrDrE)r�Zdeletedrr/r/r0�
del_binary�s 


r�cCsjtd�t�xtD]}|�d�|krdSqWtd|�t�d�}|�d|�t�	|�t
t�t�dS)NTr$r�r�)rIrr�r�r�r�r�r�r�r�r)r�rZbin_xmlr/r/r0�
set_binary�s



r�c
sFy4|rt}n|rt}nt}��fdd�|D�d}Wntk
rHdSX|r�xB�dD]6}|dkrrt|dd�}nt||d	�}|rX|j�|�qXWdStt��d}xz|D]r}	|	dkr�t	|�
d��dkr�|	�kr�|�
d�d�d��|	<q�t	|�
|	��dkr�|	�kr�|�
|	�d�d	��|	<q�W|�r6t�d
�nt
�d
�dS)a�
    Set given lve or package to default values for given parameters

    :param dict set_data: Arguments of lvectl call
    :param bool package_flag: Should we delete package or lve with given id
    :param callable is_needed: Function that takes xml element and set_data dict and returns
                               whether current xml element contains info about needed ID from set_data
    csg|]}�|��r|�qSr/r/)r5r;)�	is_needed�set_datar/r0r��sz#lve_set_default.<locals>.<listcomp>rNzset-defaultrSr�r�r��ve_id)r�r�r�r�r�r�r��set�LIMITS_LIST_NAMEr�r�r��plan_deleter�)
r��package_flagr�r�r�rZtag_�nZto_keepr�r/)r�r�r0rC�s8	
rCcCsN|dkrdSt|�}|dkr dSyt|�}Wng}YnX||krJdSdS)a
    Checks is uid owned by reseller
    :param uid: uid for check
    :param reseller_name: Reseller name, None treats as root
    :return: True - valid reseller/user pair, False - else
        Special case:
          if reseller_name is None (root) - always valid
    )N�rootT)r�zN/AF)rr)r7r9ZusernameZreseller_users_listr/r/r0�_check_reseller_user_pairs	
r�cCs(|dkr*|�dd�}|d}t||�s*dStd�|rh||dkrhtj�|�r\t|dd�q�t|d�n2|dkr�t��r�ttj�|d�d�n
t|d�y
t	�WnYnXy4t
|d}t|�}|r�|dnd}t||d	�Wnt
aYnX|ddk�r�d}d
d�}d|k�r,t|d||d
�d|k�rDt|d�}	nt
d}	d}
d|k�r|t|d|	d�}|tdk�r|d}
|�r�t}nt}�x�|D�]�}
||
|��r�x�tD]�}||k�r�y4|dk�r�t|
dd||�nt|
|d||�Wn�tttfk
�r�|dk�r|
}nt|||k}|�s6|d�r�|dk�r`t�d�}|�dt||��nt�|�}|�dt||��|
�|�YnX�q�W|�dd��s�tt�d}|�r�||dk�r�tj�|��r�t|dd�n
t|d�n6|dk�rt���rttj�|d�d�n
t|d�t|d|d�n�q�W|�s$|d�r$|�rRt nd}t�|�}
|�r�|
�dt|��|
�d|d�|
�t!�n:|�d��r�|
�dt"�#|d�j$�n|
�dt|d��x�tD]�}||k�r�|dk�r�|
}nt|||k}|�s|d�r�|dk�r:t�d�}|�dt||��nt�|�}|�dt||��|
�|��q�Wd}x"t%D]}|j&�'|
|�d}P�qrW|�s�tj(�|
�|�dd��s�tt�|�rXtj�|�}t|dd�t|d|d�t)�|t)�*d��t|d�tj+�,|�}x2tj+j-|d�D] }tj.�/||�t||d	��q W|�r�t0j1|d	�n8t���rzttj�|d�d�n
t|d�t|d�n�x`tD]X}||k�r�|dk�r�t!�2d�d�dt||��nt!�2|�d�dt||���q�W|�dd��s
tt�t|d�t|d|d�dS)Nrr9r�FT)r�r�)r�rZ)rcSst|d�|dkS)N)r�r�)r�)rr�r/r/r0rQrzlve_set.<locals>.<lambda>zset-default)r�r�r�rQrP)r�rSr�r�r��savezskip-update-cfgr,r�r�z
save-username)3r}r�rIr,r.r?rr-r=rFrEr�rr�rrCr�r�r�r�r�r�r�r�r�r�r�r�rBr�rr]r"r�r�rrr�r�ZinsertBefore�	lastChildrvZlve_infor�r�Zlvp_lve_id_listr+r�rr�r�)r�r�r9r�r�r�rZhas_veZis_needed_userr�Zcpu_is_differentZ
setted_cpuZel_listrrZis_differentr�Zel_nameZaddedZel2Zenables_reseller_limitsr�r/r/r0�lve_set&s




























 $
r�cCsvtd�t�t|d�}t|�dkr,d}nt|�dkr@|d}|dk	rftdkrf||d<t|dd�nt|dd�dS)	zX
    Set package with some heuristic algorithm to simulate old package set behavior
    Tr�rNr=r9)�is_resellerF)rIrr�r�r��package_set_ext)r�r�Z
reseller_listrr/r/r0�package_set�sr�cCs�td�t�d}|r dd�}ndd�}d|kr>t|d|d��xtD�]}|||�rFx�tD]�}||kr\yJ|dkr�|�d	�d
�dt||��n|�|�d
�dt||��Wq\tt	t
fk
�r|dkr�t�d	�}|�dt||��nt�|�}|�dt||��|�
|�Yq\Xq\Wt���rTt|�d
d�|d�}|�tt|��d}qFW|�s>d}t�d�}|�d|d�|�r�|�d|d
�|d
}t���r�t||d�}|�tt|��xjtD]b}||k�r�|dk�rt�d	�}|�dt||��nt�|�}|�dt||��|�
|��q�Wtj�
|�tt�t�t|�d|k�rlt|d�}	ntd}	d|k�r�t|d|	d�|d<|�r�|d
nd}
t|d|
d�dS)NTFcSs$|�d�|dko"|�d�|dkS)Nr�r�rr9)r�)rr�r/r/r0rsz!package_set_ext.<locals>.<lambda>cSs|�d�|dko|�d�S)Nr�r�r)r�)rr�r/r/r0rszset-default)r�r�rSr�rr�r�r9rZr�r�r�rrQrP)r�)r)rIrrCr�r�r�r�rBr�r�r�r�r�r�r>Zis_plesk�_plesk_get_package_idr}�XML_PLESK_IDr�r�copy_package_settings_to_cpanelr�r�r��
plan_apply)r�r�Zhas_packagerrrr�r)Zpackage_resellerr�rr/r/r0r��sp


 "











r�)rr��returncsHt�}|��}y tt��fdd�|��}|dStk
rBdSXdS)z7
    Find the right package id from plesk DB query
    cs|d�dhko|d�kS)Nrr�r=r/)r�)r�rr/r0rLrz'_plesk_get_package_id.<locals>.<lambda>r�N)r%Zlist_domain_packages_with_idrXrY�
StopIteration)rr�Zpanelr�Zpackr/)r�rr0r�Dsr�cCst��}td�t}|a|S)z�
    Retrives resellers to packages map from panel using /usr/bin/getcontrolpaneluserspackages
    :return: Dictionary:
    { 'reseller1' -> ['pack1', 'pack2'], 'reseller2' -> ['pack'] }
    zlist-resellers-packages)rErrF)Zpackages_users_copy�reseller_packages_mapr/r/r0rTs
rcCsB|d}|d}t�}||kr:|||kr:t|dd�dSdSdS)z�
    Set reseller package limits
    :param set_data: input data dictionary
    :return: True - limits was set succesfully
            False - supplied provider has no supplied package
    r9r�T)r�FN)rr�)r�r9�package_namer�r/r/r0�reseller_package_setbsr�c
Cs\|d}t��sdSd|��}tj�|�s.dSt|���}|dd�}i}xx|D]p}|�d�r�|���	d�}|d�
dd���}|dd	kr�|d||<|tkr�|�|�|�d
�rPd|krPdSqPWx2dD]*}||kr�t
�||�}	|	p�t|||<q�Wt||��rdSt|t�}
x2tD]*}|
|}d|�d|�d
�}|�|��qWtd�|�|d�dS)zA
    Copy package limits from ve.cfg to cpanel packages data
    r�Nz/var/cpanel/packages/Zlve_�=rrZr=�DEFAULTZ_PACKAGE_EXTENSIONSr,)rUrTrwr�i�)r>r?r{r$r)r�r-rgr�r�r�r�r�rtZmemory_to_pager��is_limits_equals�create_cpanel_limitsr�r+r&r`)
r�r�Zpackage_pathZcpanel_package_dataZnew_cpanel_package_dataZold_cpanel_datar�Z
line_partsZ
limit_nameZmemory_page_valueZcpanel_dataZlimit_valueZ
limit_liner/r/r0r�xs>






r�c	CsVxP|��D]D}|dks
|dkr q
y||||kr6dSWq
tk
rLdSXq
WdS)zC
    check if new set of limits for package are equals to used
    r�r�FT)r�rH)Z
old_limitsZ
new_limitsrr/r/r0r��s
r�cCs�i}x�|D]�}|�d�|kr
x�tD]�}y�|dkrRt|�d�d�d����||<nT|dkr�tt�t|�|�d�d������||<n t|�|�d�d����||<Wq"tt	t
fk
r�d||<Yq"Xq"Wq
W|S)	z�
    create limits for cpanel package file
    use data from ve.cfg:
    limit = limit if found in ve.cfg or DEFAULT
    return dict
    r�rSr�rr�)rTrwrUr�r�)r�r�rBr�r�rtZpage_to_memoryr�r�r�r�)Z
package_idZxml_packagesZresult_datarr�r/r/r0r��s 

 &$r�cs�td�t�d}�dkr(�fdd�}n��fdd�}x.tD]&}||�r<d}|j�|�tt�Pq<W|s؈dkr�t��}t|�dkr�d}n|d}|dk	r�t	�|�dSt
r�tddd	�g�t�
d
�ntd��t�dS)NTFcs|�d��ko|�d�S)Nr�r)r�)r)rr/r0r�rzplan_delete.<locals>.<lambda>cs|�d��ko|�d��kS)Nr�r)r�)r)rr9r/r0r�rrr(r�z"no configuration found for plan %srz+warning: no configuration found for plan %s)rIrr�r�r�rr�r�r�r�r@rArDrErCr�)rr9r��is_needed_packagerZresellers_listrr/)rr9r0r��s8

r�cCst||d�dS)N)r9)r�)rr9r/r/r0�reseller_plan_delete�sr�c	CsTyt|�|�d�d��Sttfk
rN|dkr<t|Sd�t|d�SXdS)Nrr�rPz{}%rX)rBr�r�r�r�r�r)rrr/r/r0�
get_xml_limit�sr�cCsHdd�}|dkrdSt�d|i�}||dd�}|�d�}||d|�S)z�
    Normalize string for JSON output.
    Example:
     - Input string: -_&[{}]'"`te\s/t
     - Output string: -_&[{}]'"`te\\s/t\a
    :param data_str: String for normalize
    :return: Normalied string
    cSs:d}x0t|�D]$\}}||kr|d7}||kr|SqWdS)z�
        Get the index of the specified occurrence of character in string
        :param input_string: String
        :param char_to_search: Character to search
        :param ordinal: Required occurence number
        :return: Char index
        rr=r)�	enumerate)Zinput_stringZchar_to_searchZordinal�count�idxZchr/r/r0�_get_char_index
	sz'_normalize_str.<locals>._get_char_indexNrBr��r=)r~r��rfind)Zdata_strr�Zjson_strZtrd_idxZlast_idxr/r/r0ry	s	
rycs.�dk	r�n
�rtnt�d���fdd�	}|S)a�
    Generate inner function with closured fields names, is_reseller flag and printer function

    :param list fields: List of strings that represent names of fields in final output
    :param boolean is_reseller: Format output with info about reseller or not
    :param callable printer: Function to format and print data for every entry
    :rtype: callable
    :return: function to format data for every user
    Nc
s��r.t��t�t����d<��fdd�}n�fdd�}t�t��trPt��n��d<dd�}|�d��d<td	kr�|�d
��d
<x�tD]�}||�r�t|d�}|dkr�|n
ttd��d<tt	t|d
�t
�d�d�d��d<y$tt
|�d�d�d����d<Wnt
tfk
�r$YnX|t|d
���d
<|t|d���d<t|d��d<t|d��d<t|d��d<q�W��d�dk�r��d
d�d<tt�d���d
<g}t�r�d��fdd��D��}d|g}n��fdd��D��|S)a
        :param string package_name: Find and format data for this package name
        :param string reseller_name: reseller name, owner of supplied package
        :rtype: list
        :return: List of giver package's statistics data line or empty list
        rcs�|�d�ko|�d��kS)Nr�r)r�)r)r�r9r/r0rF	sz5_package_formatter.<locals>.wrapper.<locals>.<lambda>cs�|�d�ko|�d�S)Nr�r)r�)r)r�r/r0rJ	sr�cSstrt|�St|�S)N)r]rirk)rr/r/r0rP	srTrwrYrUrQrZrP)r�rXr�r�rr�rSrRrVrWNrnc3s&|]}d|��|��d�fVqdS)z	"%s":"%s"zN/AN)r}rx)r5r�)r�r/r0r8n	sz6_package_formatter.<locals>.wrapper.<locals>.<genexpr>z{%s}csg|]}��|��d��qS)rZ)r}rx)r5r�)r�r/r0r�q	sz7_package_formatter.<locals>.wrapper.<locals>.<listcomp>)r
rr�r@ryror�r�rBr�r�r�r�r�r�r}r�r`)r�r9r�rtrr�rur�)rrr�rp)r�r�r9r0rz8	sJ



" z#_package_formatter.<locals>.wrapper)N)rerd)rrr�rprzr/)rrr�rpr0�_package_formatter+	s;r�cCs�t�td�t��}td�t��}tttd�}tt�dtd�}x|D]}|||�7}qHWt	dkrttt�dtd�}xXt
|�D]L\}}t��r�|dkr�q~x0|D](}t	dkr�||||�7}q�|||�7}q�Wq~Wt
r�td	d
�|�d�dS)Nz
list-packageszlist-resellers-packages)rqF)r�rpr=TZadminz	{"data":[rnz]})rrFrErrvrd�DEFAULT_PACKAGEr�rsr�rr>rr@rCr`)r��reseller_packagesrLr|r�r9�
packages_list�reseller_packager/r/r0�get_packages_listw	s(

r�cCs�t�td�dg}t�|}tttd�}t|dd�}x0tt�D]$\}}x|D]}||||�7}qNWq@Wt	r�t
dd�|�d�dS)	Nzlist-resellers-packagesr�)rqT)r�z	{"data":[rnz]})rrFrsrvrdr�r�rrEr@rCr`)rrrrrLr|r9r�r�r/r/r0�get_resellers_packages_list�	s

r�c
Cs�t�td�t��}td�t��}dg}t�|}ttt|d�}t|dtd�}x|D]}|||�7}qXWt|dd�}x0t	|�D]$\}}x|D]}	|||	|�7}q�Wq�Wt
r�td	d
�|�d�dS)Nz
list-packageszlist-resellers-packagesr�)rqrrF)r�rpT)r�z
{"data": [rnz]})
rrFrErrsrvrer�r�rr@rCr`)
r�r�rrrrrLr|r�r9r�r�r/r/r0�get_all_packages_list�	s"


r�cCs t�}x|D]}|||<qW|S)z�
    Converts package list to internal format
    :param package_list: Package list. Example: ['BusinessPackage', 'Package2']
    :return: Package list as dictionary. Example: {'BusinessPackage': 'BusinessPackage', 'Package2': 'Package2'}
    )rG)�package_listZpackages_users_dictr�r/r/r0�_convert_packages_list�	s
r��list-allc
Cs|dkrdS|dkr |dkr dSddlm}m}m}m}m}m}m}	�y�|dkr�tdk	r�y"|t|d	t|d
d�ia	Wq�t
k
r�|ddd�ia	Yq�Xn@y||�\}
}Wntk
r�d}
}YnX|||
d�ia	dSn�|d	kr�||||�ia	dS|d
k�r,tdk�r$|�}t
|�a	t	anta	dS|dk�rTtdk�rL|�a	t	anta	dS|dk�rvtdk�rn|�ata	dS|dk�r�tdk�r�||�}
|
a|
a	nta	dS|dk�r�tdk�r�|	�ata	dSWn@tk
�r�}zt|�Wdd}~XYntk
�rYnXdS)a�
    Parse output from GET_CP_PACKAGE_SCRIPT and get package and lve relations

    :param option: option for GET_CP_PACKAGE_SCRIPT.
     Option is one from the following possible values:  'userid', 'package', 'list-packages', 'list-resellers-packages'
    :type option: string
    :param lve_package_id: lve_id or package_name
    :type lve_package_id: string or int
    :param reseller:
    :type reseller: string
    )zlist-allrWr�z
list-packageszlist-resellers-packagesz
list-userszlist-reseller-usersF)rWr�rZr)�
list_users�admin_packages�reseller_package_by_uid�get_uids_list_by_package�resellers_packages�get_reseller_users�list_allrWNr�r)r�rTz
list-packageszlist-resellers-packagesz
list-userszlist-reseller-userszlist-all)�clcommon.cpapir�r�r�r�r�r�r��cached_usersrErHr��cached_list_packagesr��cached_resellers_packages�cached_reseller_users�cached_defaultrrFr#)ZoptionZlve_package_idrr�r�r�r�r�r�r�r9r�r�Zreseller_users_dictr;r/r/r0rF�	st$










rFcCst�tt�S)z2
    Retrieves panel users count
    :return:
    )rFr�rEr/r/r/r0�get_panel_users_countE
sr�cCs>td�td||d�r:x"t|D]}tt|�||d�q WdS)Nz
list-usersr�)r)rFrEr]r�)rrr7r/r/r0r�O
sr�cCsfx`|D]X}|�dd�}|����}x:|D]2}t|�dkr(yt|�}t|�Wq(Yq(Xq(WqWdS)Nr�rZr)r�r�r�r�r�r�)�
users_listr�r�r�r/r/r0�destroy_manyY
s

r�cCs�t�y
t�WnYnXx`|D]X}|�dd�}|����}x:|D]2}t|�dkrFyt|�}t|�WqFYqFXqFWq$WdS)Nr�rZr)rrFr�r�r�r�r�r])r�r�r�r�r/r/r0�
apply_manyg
s


r�cCs6tjt|�t|�|dt|�dt|�dd�dS)NzCan`t put proccess with pid z in lve z; error code {code})r)rvZlve_enter_pid_flagsr�rB)ZLVEid�PID�flagsr/r/r0�	limit_pid{
sr�cCstjt|�dt|�d�dS)NzCan`t release process with pid )r)rvZ
lve_leave_pidr�rB)r�r/r/r0�release_pid�
sr�c	Csttttttttd�S)N)r�r�r�r�r�r�r�r�)r�r�r�r�r�r�r�r�r/r/r/r0�get_globals�
s
r�cCs`g}t��}td�t��}|��ax6t|�D]*\}}x |D]}||kr<|�|g�q<Wq.W|S)Nzlist-resellers-packages)rErrFr�extend)r�rZ
pkg_users_oldr�r9r�Zpackage_name_in_keyr/r/r0r��
s
r�cCs|tjtjddd�}|rxtj�|�rxt||�\}}|dkrxd�|�}trVt	dd|g�n"d|}t
j�d	|�t
�
|�d
S)z�
    Call Endurance's custom script

    :param args: list of arguments for pass to Endurance's custom script
    :return: None

    ZENDURANCE_CUSTOM_SCRIPTr�)ZfileNameZ	paramNameZ	separatorrz2Error while executing Endurance's custom script
{}r(r<z	error: %sz%s
N)r>Zget_param_from_fileZCL_CONFIG_FILEr{r$r)rrr@rArDrrHrE)�argsZendurance_custom_scriptZret_codeZstd_outrMZerr_messager/r/r0�call_endurance_custom_script�
s	
r�cCstrtt|tj��St|�S)a
    Convert page value to human-readable value or bytes, depending on BYTES_FLAG;
    E.g.
    >>> _page_to_memory_or_bytes(1233254)  # BYTES_FLAG=False
    '100M'
    >>> _page_to_memory_or_bytes(1233254)  # BYTES_FLAG=True
    654321

    :type value: int
    :rtype: str | int
    )r]r�r��mmaprrk)rr/r/r0�_page_to_memory_or_bytes�
sr�cCs�t�}ttj���}t�xVt�D]L}y"tj�|�}||krF|�|�Wq"tt	t
fk
rl|�|�Yq"Xq"Wx|D]}t|�qxWdS)zV
    Remove from LVE all resellers, which are absent from panel
    :return: None
    N)r7r,r�r�rr1r�r+rHr#r�r�)Zreseller_id_list_for_deleteZcpapi_resellers_listZlve_reseller_idZlve_reseller_nameZreseller_id_for_deleter/r/r0�remove_absent_resellers�
s
r�cCs�x�ttj���D]�\}}yt�|�Wqtk
r�yt�|�WnDtk
r�ytj	�
|�t�|�Wntk
r�YnXYnXYqXqWdS)zM
    Remove from LVE all users, which absent in system
    :return: None
    N)rr,r.r�r�rrHr�r#r+r5)r�r�r/r/r0�remove_absent_users�
sr�)F)N)r)r)rrrrrrr)N)NN)rFT)T)NNN)F)F)r)F)NFNNr)rlr�N)NN)r)r)r)F)F)N)FN)r�rZN)N)�Z
__future__rrr�builtinsrZ
past.builtinsrrZfuture.utilsrr	�typingr
rrr
r�
contextlibr�rr@rJr{r�r�r�rDrZxml.dom.minidomZdomZminidomr�r~�	functoolsrr,r�rr
rtZ
clcommon.lockrZclcommon.constrr�rrrrZclcommon.cpapi.cpapiexceptionsrr>ZcleventsrrZcllimits.librZclveconfig.ve_configrrrZclveconfig.ve_lockrrr r!r"r#r$Zclcontrollibr%Zsecureior&r'ZGET_CP_PACKAGE_SCRIPTr�r�r�rnr r*rKr>r�r|r}ZIS_DEBUGr�ZLVErvr,r1r4r:rFrIrGr�r�r�r�ror@r]ryrrEr�r�r�r�r�r�r�r�r�r�r�r\r^rsrzr�r�r�r�r�r�r�r�r��contextmanagerr�r�r�rAr�r�r�r�r�r�r;r�r�r�rBr�r�rr	r
rr&r0r3r5r6r9r:r�r�rJrSrPrNrUrVr]r_rarcrdrerirkrvr{r}r~r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rCr�r�r�r�r�rr�r�r�r�r�r�r�ryr�r�r�r�r�r�r�r�r�r�rFr�r�r�r�r�r�r�r�r�r�r�r�r/r/r/r0�<module>
s�

	


		&%.	D
$
		#%&AZP
2/
	
.#>H-&'L
p


@ Telegram