.

https://t.me/AnonymousX5
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 : /proc/self/root/opt/alt/python37/lib/python3.7/site-packages/__pycache__/
Upload File :
Current File : //proc/self/root/opt/alt/python37/lib/python3.7/site-packages/__pycache__/lvectllib.cpython-37.pyc
B

��cQq�	@s|ddlmZddlmZddlmZddlmZddlmZmZddl	m
Z
mZddlm
Z
mZmZmZmZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmmZ ddl!Z!dd	l"m#Z#ddl$Z$ddl%Z%dd
lm&Z&ddl'Z'ddl(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.m/Z/dd
l0m1Z1ddl2Z2ddl3m4Z4m5Z5ddl6m7Z7ddl8m9Z9m:Z:m;Z;ddl<m=Z=m>Z>ddl'm?Z?m@Z@mAZAmBZBmCZCddlDmEZEddlFmGZGdZHdZIdZJdZKdZLdZMdZNdZOdZPeQejR�Sdd��ZTdZUe/e*jV��sdZWdZXne@eTd �ZWe?eWd!�ZXd"d#�ZYd$d%�ZZd&d'�Z[e1d(�d)d*�Z\�dd+d,�Z]d-d.�Z^d/d0d/d1dd2dd3d4�Z_d2Z`d5d0ddddddd4�Zad6d7d8d9d:d;d<d=gZbd>acdaddaed?afd?agiahd?aid?ajd?akd?ald?amd?and?aod@apd?aqd?ardAdB�ZsdCdD�ZtdEdF�Zu�ddGdH�ZvdIdJ�ZwdKdL�Zxew��p$ex�Zy�ddMdN�ZzdOdP�Z{dQdR�Z|dSdT�Z}�ddUdV�Z~dWdX�Zej�dYdZ��Z��dd[d\�Z�d]d^�Z��dd_d`�Z�dadb�Z�dcdd�Z��d	dedf�Z�dgdh�Z�didj�Z��d
dkdl�Z�dmdn�Z�dodp�Z��ddqdr�Z�dsdt�Z�dudv�Z�dwdx�Z��ddydz�Z��d
d{d|�Z�d}d~�Z�dd��Z�d�d��Z��dd�d��Z�d�d��Z��dd�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z��dd�d��Z�e�fd�d��Z��dd�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z��dd�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��dd�d��Z�d�dZ��dd�dĄZ��dd�dƄZ��dd�dȄZ�e�e�eeQdɜd�d˄Z�d�d̈́Z�d�dτZ�d�dфZ�d�dӄZ�d�dՄZ��dd�dׄZ�d�dلZ�d�dۄZ�d�d݄Zdd�d߄Z�d�d�Z�d�d�Z�d�d�Z�da�da�da�da�da�d�d�Z̐dd�d�Z�d�d�Zΐdd�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(�)�print_function)�absolute_import)�division)�range)�
basestring�unicode)�	iteritems�reraise)�Dict�List�Tuple�Optional�TextN)�partial)�PAGESIZE)�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_tempfilez%/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-�:/opt/alt/python37/lib/python3.7/site-packages/lvectllib.py�lvp_listPs
r/cCst�}|��|��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-r.�get_active_resellersXsr2cCs|dd�t�D�kS)z]
    Check whether giver reseller has activated reseller limits or not
    :return: bool
    css|]\}}|VqdS)Nr-)�.0�name�uidr-r-r.�	<genexpr>hsz,is_active_reseller_limits.<locals>.<genexpr>)r2)�
reseller_namer-r-r.�is_active_reseller_limitscsr8)�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)r9r-r-r.�raise_cpanel_encoding_errorlsrDcCs0yt|�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-r.�get_global_lock~srGcCsN|tjkrJ|dkrJtr0ttddt|�g�ntdt|��t�|�dS)Nrr:z
lvectl: %szlvectl: Error: )	�errnoZENOSYSr>r?�MULTI_FORMATr@rArBrC)�result�messager-r-r.rE�s
rE�r;�ii)�cpu�ncpu�io�ep�mem�pmem�nproc�iops�drOrNrPrRrSrTrUrQ���falsecCs|adS)N)r>)Z	json_flagr-r-r.�set_json�srZcCs|adS)N)�
BYTES_FLAG)Z
bytes_flagr-r-r.�	set_bytes�sr\c	Cs�trtdkrd}n tdkr d}ntdkr.d}nd}ddd	d
ddd
gddd	d
ddd
dgddd	d
ddd
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^r_rarbr;�CPU)�NOIOPS�LVE_VERSIONr>�index)�version�fieldsZ	speed_idxr-r-r.�
get_fields�s rqcCsR|dkr.t��}|ddkr$td��|dan|at��}|sFtd��t��adS)Nrzget_lve_version failedzinit_lve() failed.)�clcommonZget_lve_version�	Exceptionrm�pylve�
initialize�liblve_settings�lve_settings)Zlve_verZ
lve_statusr-r-r.�init�s
rxc
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�
ValueErrorrAr@rBrC)Zpacked_cpu_datar9r-r-r.�_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�:�rXZproc0zcpu MHz)
�open�CPUINFO�IOErrorrAr@rBrC�strip�len�split)ZcpuinfoZprocinfoZnprocs�f�liner-r-r.�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?$NrV)r@�CPUINFO_DATA�int�re�match�round�CORE_WEIGHT�min)�data�lncpu�cpu_datarOZcpu_percentZfrom_cpu_limitr-r-r.�convert_from_old_cpu9sr�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�%rXrV)	r@r�r�r�r��float�replacer�r�)r�r�rOZpercentr-r-r.�convert_from_speed_percentOsr�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�)r@r�r�r�r��compile�
IGNORECASEr��group�upperr�r�)r�r�rOZcpu_freq�patternr�r�r�r-r-r.�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-r.�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-r.�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�r@r��lstrip)�speedr�rTr-r-r.�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)rtruZget_available_lve_id�	lve_setupr"�syslog�	LOG_ALERT�lve_destroy)Zsettings�lve_idr-r-r.�
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�?)
rtrvr��ls_cpu�ls_io�	ls_enters�ls_nproc�ls_iopsr��mathZceilr�
ls_memory_phy)r��ls_cpusr�r�r�r�r��sr-r-r.�make_liblve_settings�s	r�cCs0t|�d��}|s(|�d�}t�|�j}t|�S)N�id�user)r@�getAttribute�pwd�getpwnam�pw_uidr�)�ve_lve_elementZuser_uidZ	user_namer-r-r.�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)r@rI�updaterAr|�dumps)Z
error_typer��
extensionsrJr-r-r.r?�s
r?c
Cs�y tt�|�d�d��||<Wn`tttfk
r�||||<|�|�}|�dt	||��y|�
|�WnYnXYnXdS)Nr�limit)r��
ve_default�getElementsByTagNamer�r�
IndexErrorr~�
createElement�setAttributer@�appendChild)�xml�ve_defaults�ve_cfg�val�default�noder-r-r.�check_def_value�s 
r�cs�fdd�|jD�S)Ncs&g|]}t|tj�r|j�kr|�qSr-)�
isinstancer�ZElementZtagName)r3�_)�tagr-r.�
<listcomp>�sz"xml_filter_tag.<locals>.<listcomp>)Z
childNodes)r�r�r-)r�r.�xml_filter_tag�sr�cCsFx@t||�D]2}|dk	r$|�|�s$q|dk	r<|�|�|kr<q|SWdS)N)r�ZhasAttributer�)r�r��attrZattr_valZ
child_noder-r-r.�xml_filter_first�sr�cCs&t|||d�}|dkrt��|�|�S)N)r�r�r�)r�r�r�)r�r�r�Zfiltered_child_noder-r-r.�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�r@)r�r�r�r�Zfirst_child_noder-r-r.�set_child_tag_atrr	sr�c
Cs�yt�\aaWnPtk
r^}z2tr8ttdt|�g�ntt|��t	�
d�Wdd}~XYnXdat}y\|r�|r�t�
d�at}q�tttdt|��}|r�|�d�daq�t�t�anttd�daWn:tk
r�tr�tdd	d
g�t	�
d�ntd�YnXt�d�at�d
�at�t�atjj�t�t�d�}t|�dk�rN|dant�
d�at�t�t�d�a t�d�a!tt!�dk�r�t"t!dj#j$�a!nda!ia%t&t'dtt%t|d�}|dd�y,t�d�d�(d�}t)|t%dd�t%d<Wnnt*tt+fk
�rVt)|dt%dd�t%d<t�
d�}|�,dt|d��yt�|�WnYnXYnXy t"t�d�d�(d��t%d<Wnbt*tt+fk
�r�|dt%d<t�
d�}	|	�,dt|d��yt�|	�WnYnXYnX|dd�t-dk�r�|dd�n|ddt.id�|dd�|dd�|d d�xbt/t%�D]V\}
}|dk�r4d!�0|
�}t�rftdd|g�nd"|}
t	j1�2d#|
�t	�
d��q4WdS)$z�
    :param bool lvp_defaults: load reseller's default limits instead of global
    :param int lvp_id: lvp id to load customise defaults
    r:r;N�true�defaultsr�rr&�WARNINGzdefault section error in ve.cfgz(warning: default section error in ve.cfgr*�packagez
enter-by-name�binaryro)r�r�r�r�r�rO)r�rNr�)r��other�
maxentryprocsrQrP�rR)r�r�rSrTrUz!ERROR: Incorrect {} default valuez	error: %sz%s
)3rr��ve_lveconfigrr>r?rIr@rArBrC�ubc�LVE_DEFAULTr�r��LVP_DEFAULTr�r!r�Z	cloneNoder�r��ve_lve�
ve_package�ve_lvpr*�mapr1r0r��ve_enter_by_namer��	ve_binary�ve_cfg_versionr�Z
firstChildZ	nodeValuer�rr�r�r�rr~r�rm�MEM_DEFAULT_CL5r�format�stderrrF)�lvp_id�lvp_defaultsr9Zdefault_limitsZdefaults_root_nodeZve_enter_by_name_listZ
check_def_valr�rNrQ�key�valuerK�err_messager-r-r.�get_XML_cfgs�











 







r�c
CsFytt||dd��}|||<|Stttfk
r@t||�SXdS)Nr�)r�r�)r�r�rr�r~)r��elr��
setup_datar�r-r-r.�check_value�sr�cCs&t��}|��|�|�}t|�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
    N)�lveapir#r0Zget_idr�)�resellerr1Zreseller_idr-r-r.�_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�tj
d�|d	���fd
d�}x�tD]�}	||	�r�td|	tt�}
y&ttt|	dd
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_idr�r-r.�<lambda>��z$prepare_setup_data.<locals>.<lambda>z"/usr/local/directadmin/data/users/z
/user.confrXzpackage=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-r.r�rrOrNr�)r�r�)r�rPr�rRrr�r�rQrTrSrU)�copyr�r��get_reseller_packages_mapr<�is_dar��getpwuid�pw_namer��readrsr�r�r�r�r�r�r�r�rr�r~r�)
rr�r�Zres_pkg_dict�is_needed_planZuser_pwd�filenamer��textr�r�rNrQr-)rr�r.�prepare_setup_data�sP






rc	CsZy,tjtd|gtjtjd�}|��|jdkStk
rTtddtd|f�YnXdS)Nz-l)�stdoutr�r���zfailed to run "%s %s %s")�
subprocess�Popen�UMOUNT�PIPEZcommunicate�
returncode�OSErrorrE)�path�pr-r-r.�
umount_dir�s
rc	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�)r3�mr-r-r.r�sz"prepare_mounts.<locals>.<listcomp>zfailed to parse /proc/mounts�
F)ryr�isfile�EXCLUDE_MOUNTS_CONFr�r��appendr�r�r�rE�unshareZCLONE_NEWNS�	readlinesr�r�searchr)Zreg_exp_listZconfr�r�r�ZmountsZATTEMPTSr��errorZmountZreg_exprr-r-r.�prepare_mounts�s<
" 


r!cCsNd}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)�err_msgN)rZcallrrAr!rt�	lve_start)Z	MOUNT_CMDr-r-r.r%sr%cCstj|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)rt�
lve_creater�)r�r&r-r-r.r'(sr'cCstj|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}})r$r&N)rtZlve_lvp_creater�)r�r&r-r-r.�
lvp_create4sr(cCs8t�}x,t�D]"}tj|d�|�d�|�|�qW|S)Nz9lvectl: Can`t destroy lvp with id {}; error code {{code}})r$)�listr/rt�lve_lvp_destroyr�r)Zdestroyed_listr�r-r-r.�destroy_lvp_all@sr+cCs(|dkrt�ntj|d�|�d�dS)N�allz9lvectl: Can`t destroy lvp with id {}; error code {{code}})r$)r+rtr*r�)r�r-r-r.�lvp_destroyIsr-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)r$TZOKZWARNzwarning: {})r�r*r+�boolr+r�r)r,�lve_id_listr�Zcheck_inside_listZresellers_supported�detect_inside_lvpr>r?�
SINGLE_FORMATrIrA)r�Zcant_remove_msgZcant_destroy_msgZ	destroyedZid_r-r-r.r�Rs0




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�|dkr�tj|t	d
�|�d�ntj|t	d�|�d�n0|dk�rtjt	dd�ntj|t	d�|�d�WnBtk
�r`}z"tddt|�dt|��Wdd}~XYnXWdQRXdS)Nr rPrNrOrRrQr�rSrTr`rUrz&Can`t setup default settingsfor LVP {})r$z/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_lvpr(�warnings�catch_warnings�filterwarningsr�r�rwr�r�r�Z	ls_memoryr�rmr�r�r�rt�lve_set_defaultr�Z
lve_lvp_setupr��RuntimeWarningrEr@)r�r�Zrwr-r-r.r�us6

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-usersr�r�)�packages_users�GetControlPanelUsersr��dict�KeyError)r�r�r��old_packages_usersZtemp_packager-r-r.�"get_package_and_reseller_by_lve_id�s(


r=c

Cst}|dkrtd|�t}|atdkr0t|d�d}|dk	rB|}n0|rLt}	nt}	x |	D]}
t|
d�|krV|
}PqVW|dkr�y||d}Wnttfk
r�d}YnXy||d}Wnttfk
r�d}YnX|dk	�r�t	||d	�t
d
|tt�}yt
t|ddd
�|d�td<Wntttfk
�r4YnXtddk�rPtdtd<t
d|tt�tdk�rxt
d|tt�ndtd<ytt|ddd
��}|td<Wntttfk
�r�YnXt
d|tt�t
d|tt�t
d|tt�n t	|||d�tdk�rdtd<|�st||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�useridrX)r�N)r�r�r�)r�rOrNr�)r�r�)r�rPr�rRr�r�rQrTrSrU)r�r�rY)r8r9r�r�r�r�r��	NameErrorr;rr�r�r�r�r�rr�r~r�r�r�)
r�rrJr��out_noder�r<Znew_packages_usersr��	node_listr�r�rQr-r-r.�	lve_apply�sh







rBcCst|�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)rA)�stringr-r-r.�_print_string_safesrDcGs2td�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.
    rXcSsg|]}d|�qS)z%8sr-)r3r�r-r-r.r�.sz_pprint.<locals>.<listcomp>Nrz%30s)rD�join)rpr-r-r.�_pprint)srFcGs<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.
    rXcSsg|]}d|�qS)z%8sr-)r3r�r-r-r.r�6sz_pprint_f.<locals>.<listcomp>N���z%30sr)rDrE)rpr-r-r.�	_pprint_f1srHcGs2td�d|dgdd�|dd�D���dS)ze
    Command to print data with first column 30 symbols wide.
    Useful to print packages data.
    rXz%30srcSsg|]}d|�qS)z%8sr-)r3r�r-r-r.r�>sz_pprint_p.<locals>.<listcomp>r;N)rDrE)rpr-r-r.�	_pprint_p9srIcGs@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.
    rXz%30srcSsg|]}d|�qS)z%8sr-)r3r�r-r-r.r�Fsz_pprint_r.<locals>.<listcomp>r;rN)rDrE)rpr-r-r.�	_pprint_rAsrJcCsft|�}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�TFrXirz*{})r@�typer�
startswithr�r�r�)r��was_changedr-r-r.�_pmem_vmem_to_bytes_valueIs	"
rNcCs�d}d}t|t�r,|�d�r,|dd�}d}yt|�}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"
    rXFr�r;NTrWirz{}{}Mz{}{}K)r�rrLr�rr�)r�rJrM�vr-r-r.�_mb_mem_s rP�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
    rNrVcs��|d�S)NrX)r{)r�)r�r-r.r�rz_formatter.<locals>.<lambda>rXcSstrt|�St|�S)N)r[rNrP)r�r-r-r.r�srOrSrRrQrTrPrU)rcrdrkZNCPUrerfrgrhrirj�PACKAGEN�,c3s"|]}d|��|d�fVqdS)z	"%s":"%s"rXN)r{)r3r�)�
fields_mapr-r.r6�sz_formatter.<locals>.<genexpr>z{%s}csg|]}��|d��qS)rX)r{)r3r�)rTr-r.r��sz_formatter.<locals>.<listcomp>)r�rr�r@rqr>rE)
�printer�
default_idZdefault_package�more_fields�get_dataZ_cpu�convert_mem_limits�resrpr�r-)r�rTr.�
_formatterzs(r[cs��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	rPt|�nt�t||d�t�t��t|||dd��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	�r8t
|��d<n|�d<|dk�r^t	�rTdnd�d<n|�d<d�d
k�r�dtt�d
�d��d��d
<nt�d
�d�d
<�d
�d<tt
�d���d
<g}t	�r�d��fdd��D��}d|g}n��fdd��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
        rXr�r�Nr;)r�T)rr�rJcs4t�|�tt|�kr(dtt|�St�|�S)Nr�)r@r�)r�)r�r-r.r�sz2_user_formatter.<locals>.wrapper.<locals>.<lambda>cSstrt|�St|�S)N)r[rNrP)r�r-r-r.r�sr�rNrOrSrR�vmemrQrPrTrUzN/Ar�rVr�rSc3s"|]}d|�|��fVqdS)z	"%s":"%s"N)�lower)r3r�)r�r-r.r6�sz3_user_formatter.<locals>.wrapper.<locals>.<genexpr>z{%s}csg|]}�|���qSr-)r])r3r�)r�r-r.r��sz4_user_formatter.<locals>.wrapper.<locals>.<listcomp>)r8r�rr�r�rr�rBr@r>�_normalize_strr�r�r�rE)r�r�r�Z
check_changedrYrZr�)rprU)r�r.�wrapper�sR


$z _user_formatter.<locals>.wrapperr-)rprUr_r-)rprUr.�_user_formatter�s	>r`cCs�t�yBtd�|dk	r(|r(td|�n|dk	r>td|d�ntd�WnYnXdg}tt|d�}t�|}t|�}xtD]}|||�7}q�Wtr�tdd�	|�d	�dS)
Nz
list-usersr>zlist-reseller-users)r�rR)rWz	{"data":[rSz]})
r�r9r[rFrqr`r8r>rArE)r>r�rWrJrp�	formatterr�r-r-r.�paneluserslimits�s$

rbcCsBtd�g}x0tt�D]$\}}|�t|�|d|df�qW|S)z@Get list of tuples[lve_id, reseller, package] from control panelz
list-usersr�r�)r9rr8rr�)rJZstr_uidZpayloadr-r-r.�paneluserslists
 rccCs>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
clveconfigrdr9r)r8�keysr�)rd�packagesr-r-r.�panelpackagesdict
s
rgcCspt�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-usersrR�RESELLER)rW)rUz	{"data":[rSz]}N)
r�r9r[rHrqr`r8r>rArE)rJrprar�r-r-r.�all_users_limitss
rics6t��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�)r3r�r-r-r.r�;sz(_filtering_da_admins.<locals>.<listcomp>csi|]\}}|�kr||�qSr-r-)r3r�r�)�uids_da_adminsr-r.�
<dictcomp><sz(_filtering_da_admins.<locals>.<dictcomp>)r<Z	getCPNamerr)�ve_dictr-)rjr.�_filtering_da_admins0s	rmc
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-)r3�krO)�cfg_lvp_id_listr-r.rkHsz&prepare_apply_data.<locals>.<dictcomp>T)r�)r�r�)r�rr�)r9r:r8r*r�r1�id_list�itemsr�r/r�r,r/r�r�rm�guess_reseller_by_package)
r�Zpackages_users_rArprlr�r�r5Zpkg�	resellersr-)ror.�prepare_apply_dataAs>


rtc	Cszt�td�t�}t��r*tt���}nt�}tjj}dtj_xH|�	�D]<}|�
|d�dkrJ||d||d}}t|||d�qJW|tj_td�t���rvtd�}tj�
�}x�|�	�D]�}t|dd�t|||d||d|d�t|d	�td|d	�tj
�|�}tj�|�}	xR|	D]J}
t�|
�j}|�
|d�|k�r`tjj||d
�||�d�t||d��q$Wq�WdS)
Nz
list-usersTrr�r�)r@r�)r�r�)r@r�r�)r�z6Can`t move lve_id={} to lvp_id={}; error code {{code}})r$)r�)r�r9rtr*r+r:Zlve_id_lvp_id_pairsr)r&rer{rBr,r��get_reseller_namerrZcpapirr�r�r��lve_lvp_mover�)rlZlve_lvp_mapZorig_ignore_errorr�r@r�Zkernel_mappingZlvp_id_r7�usersr��lve_id_r-r-r.�
lve_apply_allss<

 

ryc
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)rGr�r�r�r*r,Zmap_lve_id_listrtrvr�rr;r)Z
lve_existsr��remover-�
parentNode�removeChildrr�rB)r�r�rwr�r-r-r.�_remove_reseller�s,





r}cCsHt|�rtj|d�n.tr8tddd|g�t�d�ntd|�dS)z&Disable reseller limits and call hooks)r�r&r�z!no configuration found for LVP %srz*warning: no configuration found for LVP %sN)r}r�throw_eventr>r?rBrCrA)r7r�r-r-r.�disable_reseller_limits�srcCs�td�t�d}xPtD]H}t|d�|krd}t|�t|�|j�|�tt	�t�t
|�qW|s�tr�tddd|g�t
�d�ntd|�dS)	NTF)r�r&r�z no configuration found for VE %srz)warning: no configuration found for VE %s)rGr�r�r�r�r'r{r|rr�rBr>r?rBrCrA)r��Deletedr�r-r-r.�
lve_delete�s"
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)ryr�existsr>r?rArBrCr-r-r-r.�lve_enter_check�s
r�cCsDt�y,||��}tdd�}|�|�|��WnYnXdS)Nz/proc/lve/enter�w)r�r�r�rF�close)Zsignr�r�r�r-r-r.�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)r�r>r�r�rA)rJ�firstr�rr-r-r.�
list_binaries�s



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

r�cCs�td�t�t�d}xBtD]:}|�d�|krd}td|�|j�|�tt	�t�qW|s�t
rxtddd|g�ntd|�t
�d	�dS)
NTFrr�r&r�zno configuration found for %sz&warning: no configuration found for %sr)rGr�r�r�r�r�r{r|rr�r>r?rArBrC)r�Zdeletedr�r-r-r.�
del_binarys 


r�cCsjtd�t�xtD]}|�d�|krdSqWtd|�t�d�}|�d|�t�	|�t
t�t�dS)NTrr�r�)rGr�r�r�r�r�r�r�r�r�r)r�r�Zbin_xmlr-r-r.�
set_binary(s



r�c
sFy4|rt}n|rt}nt}��fdd�|D�d}Wntk
rHdSX|r�xB�dD]6}|dkrrt|dd�}nt||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
�nt
�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-)r3r9)�	is_needed�set_datar-r.r�Hsz#lve_set_default.<locals>.<listcomp>rNzset-defaultrQr�r�r��ve_id)r�r�r�r�r�r{r|�set�LIMITS_LIST_NAMEr�r�r��plan_deleter�)
r��package_flagr�r�r�r�Ztag_�nZto_keepr�r-)r�r�r.r68s8	
r6cCsN|dkrdSt|�}|dkr dSyt|�}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)r5r7ZusernameZreseller_users_listr-r-r.�_check_reseller_user_pairfs	
r�cCs�|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	�WnYnXy4t
|d}t|�}|r�|dnd}t||d	�Wnt
aYnX|ddk�rvd}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�Wtt�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�|�rDt 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�qdW|�s�tj(�|
�tt�|�r<tj�|�}t|dd�t|d|d�t)�|t)�*d��t|d�tj+�,|�}x2tj+j-|d�D] }tj.�/||�t||d	��qW|�rtt0j1|d	�n8t���r^ttj�|d�d�n
t|d�t|d�n�x`tD]X}||k�r||dk�r�t!�2d�d�dt||��nt!�2|�d�dt||���q|Wtt�t|d�t|d|d�dS)Nrr7r�FT)r�r�)r�rX)r�cSst|d�|dkS)N)r�r�)r�)r�r�r-r-r.r�rzlve_set.<locals>.<lambda>zset-default)r�r�r�rOrN)r�rQr�r�r��saver*r�r�z
save-username)3r{r�rGr*r,r2r�r+r0r9r8rrrr�r�r6r�r�r�r�r�r�rr�r~r�r�r�r@r�rrBr!r�r�rrr�r{ZinsertBefore�	lastChildrtZlve_infor�ruZlvp_lve_id_listr)rvrr~r�)r�r�r7r�r�rsr�Zhas_veZis_needed_userr�Zcpu_is_differentZ
setted_cpuZel_listr�r�Zis_differentr�Zel_nameZaddedZel2Zenables_reseller_limitsrxr-r-r.�lve_set�s�




























 $
r�cCsvtd�t�t|d�}t|�dkr,d}nt|�dkr@|d}|dk	rftdkrf||d<t|dd�nt|dd�dS)	zX
    Set package with some heuristic algorithm to simulate old package set behavior
    Tr�rNr;r7)�is_resellerF)rGr�rrr�r��package_set_ext)r�r�Z
reseller_listr�r-r-r.�package_set>sr�cCs�td�t�d}|r dd�}ndd�}d|kr>t|d|d��xtD�]}|||�rFx�tD]�}||kr\yJ|dkr�|�d	�d
�dt||��n|�|�d
�dt||��Wq\tt	t
fk
�r|dkr�t�d	�}|�dt||��nt�|�}|�dt||��|�
|�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�|�}|�dt||��|�
|��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�r�r7)r�)r�r�r-r-r.r]sz!package_set_ext.<locals>.<lambda>cSs|�d�|dko|�d�S)Nr�r�r�)r�)r�r�r-r-r.r`szset-default)r�r�rQr�rr�r�r7rXr�r�r�r�rOrN)r�)r�)rGr�r6r�r�r�r�r@rr�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_packager	r�r�r�r'Zpackage_resellerr�r�r-r-r.r�Wsp


 "











r�)r�r��returncsHt�}|��}y tt��fdd�|��}|dStk
rBdSXdS)z7
    Find the right package id from plesk DB query
    cs|d�dhko|d�kS)Nrr�r;r-)�x)r�r�r-r.r�rz'_plesk_get_package_id.<locals>.<lambda>r�N)r$Zlist_domain_packages_with_id�next�filter�
StopIteration)r�r�Zpanelrf�packr-)r�r�r.r��sr�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)r8rr9)Zpackages_users_copy�reseller_packages_mapr-r-r.r�s
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
    r7r�T)r�FN)rr�)r�r7�package_namer�r-r-r.�reseller_package_set�sr�c
Cs\|d}t��sdSd|��}tj�|�s.dSt|���}|dd�}i}xx|D]p}|�d�r�|���	d�}|d�
dd���}|dd	kr�|d||<|tkr�|�|�|�d
�rPd|krPdSqPWx2dD]*}||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_�=rrXr;�DEFAULTZ_PACKAGE_EXTENSIONSr*)rSrRr\�
i�)r<r=ryrrr�rrLr�r�r�r�rzrrZmemory_to_pager��is_limits_equals�create_cpanel_limitsr�rr%rE)
r�r�Zpackage_pathZcpanel_package_dataZnew_cpanel_package_dataZold_cpanel_datar�Z
line_partsZ
limit_nameZmemory_page_valueZcpanel_dataZlimit_valueZ
limit_liner-r-r.r��s>






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)rer;)Z
old_limitsZ
new_limitsr�r-r-r.r�s
r�cCs�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�rQr�rr�)rRr\rSr�r�)r�r�r@r�r�rrZpage_to_memoryr�rr�r~)Z
package_idZxml_packagesZresult_datar�r�r-r-r.r�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
�ntd��t�dS)NTFcs|�d��ko|�d�S)Nr�r�)r�)r�)rr-r.r,rzplan_delete.<locals>.<lambda>cs|�d��ko|�d��kS)Nr�r�)r�)r�)rr7r-r.r.rrr&r�z"no configuration found for plan %srz+warning: no configuration found for plan %s)rGr�r�r{r|rr�rrr�r�r>r?rBrCrAry)rr7r��is_needed_packager�Zresellers_listr�r-)rr7r.r�'s8

r�cCst||d�dS)N)r7)r�)rr7r-r-r.�reseller_plan_deleteMsr�c	CsTyt|�|�d�d��Sttfk
rN|dkr<t|Sd�t|d�SXdS)Nrr�rNz{}%rV)r@r�r�rr�r�r�)r�r�r-r-r.�
get_xml_limitQsr�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)Zinput_stringZchar_to_searchZordinal�count�idxZchr-r-r.�_get_char_indexbsz'_normalize_str.<locals>._get_char_indexNr@r��r;)r|r��rfind)Zdata_strr�Zjson_strZtrd_idxZlast_idxr-r-r.r^Ys	
r^cs.�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�}|dkr�|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
        r�cs�|�d�ko|�d��kS)Nr�r�)r�)r�)r�r7r-r.r�sz5_package_formatter.<locals>.wrapper.<locals>.<lambda>cs�|�d�ko|�d�S)Nr�r�)r�)r�)r�r-r.r�sr�cSstrt|�St|�S)N)r[rNrP)r�r-r-r.r�srRr\rWrSrOrXrN)r�rVr�r�rr�rQrPrTrUNrSc3s&|]}d|��|��d�fVqdS)z	"%s":"%s"zN/AN)r{r])r3r�)r�r-r.r6�sz6_package_formatter.<locals>.wrapper.<locals>.<genexpr>z{%s}csg|]}��|��d��qS)rX)r{r])r3r�)r�r-r.r��sz7_package_formatter.<locals>.wrapper.<locals>.<listcomp>)r�rr�r>r^rmr�r�r@r�r�r�r�r�rr{r�rE)r�r7r�rYr�r�rZr�)rpr�rU)r�r�r7r.r_�sJ



" z#_package_formatter.<locals>.wrapper)N)rJrI)rpr�rUr_r-)rpr�rUr.�_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)rVF)r�rUr;TZadminz	{"data":[rSz]})r�r9r8rr[rI�DEFAULT_PACKAGEr�rqr�rr<rr>rArE)rf�reseller_packagesrJrar�r7�
packages_list�reseller_packager-r-r.�get_packages_list�s(

r�cCs�t�td�dg}t�|}tttd�}t|dd�}x0tt�D]$\}}x|D]}||||�7}qNWq@Wt	r�t
dd�|�d�dS)	Nzlist-resellers-packagesrh)rVT)r�z	{"data":[rSz]})r�r9rqr[rIr�r�rr8r>rArE)rWrprJrar7r�r�r-r-r.�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-packagesrh)rVrWF)r�rUT)r�z
{"data": [rSz]})
r�r9r8rrqr[rJr�r�rr>rArE)
rfr�rWrprJrar�r7r�r�r-r-r.�get_all_packages_list�s"


r�cCs t�}x|D]}|||<qW|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'}
    )r:)�package_listZpackages_users_dictr�r-r-r.�_convert_packages_list	s
r��list-allc
Cs|dkrdS|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	dSn�|d	kr�||||�ia	dS|d
k�r,tdk�r$|�}t
|�a	t	anta	dS|dk�rTtdk�rL|�a	t	anta	dS|dk�rvtdk�rn|�ata	dS|dk�r�tdk�r�||�}
|
a|
a	nta	dS|dk�r�tdk�r�|	�ata	dSWn@tk
�r�}zt|�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-allr>r�z
list-packageszlist-resellers-packagesz
list-userszlist-reseller-usersF)r>r�rXr)�
list_users�admin_packages�reseller_package_by_uid�get_uids_list_by_package�resellers_packages�get_reseller_users�list_allr>Nr�r�)r�r�Tz
list-packageszlist-resellers-packagesz
list-userszlist-reseller-userszlist-all)�clcommon.cpapir�r�r�r�r�r�r��cached_usersr8r;r�cached_list_packagesr��cached_resellers_packages�cached_reseller_users�cached_defaultrrDr)ZoptionZlve_package_idr�r�r�r�r�r�r�r�r7r�r�Zreseller_users_dictr9r-r-r.r9*	sp$"









r9cCst�tt�S)z2
    Retrieves panel users count
    :return:
    )r9r�r8r-r-r-r.�get_panel_users_count�	sr�cCs>td�td||d�r:x"t|D]}tt|�||d�q WdS)Nz
list-usersr�)r�)r9r8rBr�)rr�r5r-r-r.r��	sr�cCsfx`|D]X}|�dd�}|����}x:|D]2}t|�dkr(yt|�}t|�Wq(Yq(Xq(WqWdS)Nr�rXr)r�r�r�r�r�r�)�
users_listr�rwr�r-r-r.�destroy_many�	s

r�cCs�t�y
t�WnYnXx`|D]X}|�dd�}|����}x:|D]2}t|�dkrFyt|�}t|�WqFYqFXqFWq$WdS)Nr�rXr)r�r9r�r�r�r�r�rB)r�r�rwr�r-r-r.�
apply_many�	s


r�cCs6tjt|�t|�|dt|�dt|�dd�dS)NzCan`t put proccess with pid z in lve z; error code {code})r$)rtZlve_enter_pid_flagsr�r@)ZLVEid�PID�flagsr-r-r.�	limit_pid�	sr�cCstjt|�dt|�d�dS)NzCan`t release process with pid )r$)rtZ
lve_leave_pidr�r@)r�r-r-r.�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-r.�get_globals�	s
r�cCs`g}t��}td�t��}|��ax6t|�D]*\}}x |D]}||kr<|�|g�q<Wq.W|S)Nzlist-resellers-packages)r8rr9r�extend)r�r�Z
pkg_users_oldr�r7r�Zpackage_name_in_keyr-r-r.rr�	s
rrcCs|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_FILEryrrrr�r>r?rBr�rFrC)�argsZendurance_custom_scriptZret_codeZstd_outrKr�r-r-r.�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��mmaprrP)r�r-r-r.�_page_to_memory_or_bytes
sr�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)r)r*r�rsr�r/rurr;rr�r})Zreseller_id_list_for_deleteZcpapi_resellers_listZlve_reseller_idZlve_reseller_nameZreseller_id_for_deleter-r-r.�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�rr;r�r"r)r')r�r�r-r-r.�remove_absent_users>
sr�)F)N)r)r)rrrrrrr)N)NN)rF)NNN)F)F)r)NFNNr)rQr�N)NN)r)r)r)F)F)N)FN)r�rXN)N)�Z
__future__rrr�builtinsrZ
past.builtinsrrZfuture.utilsrr	�typingr
rrr
r�
contextlibr�rr3rHryr�r�r�rBrZxml.dom.minidomZdomZminidomr�r|�	functoolsrrr�rr�rrZclcommon.constrr�rrrrZclcommon.cpapi.cpapiexceptionsrr<ZcleventsrrZcllimits.librZclveconfig.ve_configrrrZclveconfig.ve_lockrrrr r!r"r#Zclcontrollibr$Zsecureior%ZGET_CP_PACKAGE_SCRIPTr�r�r�rlrrrIr1r�rzr{ZIS_DEBUGr�ZLVErtr*r/r2r8rDrGrEr�r�r�r�rmr>r[rwr�r8r�r�r�r�r�r�r�r�r�r�rZr\rqrxr�r�r�r�r�r�r�r�r��contextmanagerr�r�r�r?r�r�r�r�r�r�r�r�rrr!r%r'r(r+r-r�r�r=rBrDrFrHrIrJrNrPr[r`rbrcrgrirmrtryr}rr�r�r�r�r�r�r�r�r6r�r�r�r�r@r�rr�r�r�r�r�r�r�r^r�r�r�r�r�r�r�r�r�r�r9r�r�r�r�r�r�r�rrr�r�r�r�r-r-r-r.�<module>
s�

	


			D
$
		#$&VL
2)
	
.#6H-&'L
k
https://t.me/AnonymousX5 - 2025