/0?: d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
m Z G d de Z G d d Z
dS )
N)cpapi)FormattedException) exec_utility)(_convert_memory_value_to_adaptive_format"check_pmem_lower_bound_from_stringc e Zd ZdS )LvectlExceptionN)__name__
__module__ __qualname__ py/cllimits/lvectl_lib.pyr r s Dr r c e Zd ZdZdZdZdZdZdZdZ dZ
d Z e d Z
d d Zd d
Zd d Zd Zd
Zd Zd dZd Zd dZd Zd Zd dZd Zd Zd Z d Z d Z d Z d dZ dS ) LveCtlz*
Library for work with LVE limits
i@B i ʚ;Nz/usr/sbin/lvectlc | t | j dg \ }}t | | _ d| _ d| _ dS )z5
Object constructor. Get LVE version
z lve-versionN)r
_UTILITY_PATHint lve_version
_package_data_resellers_data)self_
s_lve_versions r __init__zLveCtl.__init__% sC ((:]OLL= }-- ! # r c t j } t | }| d |S )zH
Get panel users uid list
:return: List of uids
r )r
list_userslistkeysappend)
user_packagesuid_lists r get_panel_users_uid_listz LveCtl.get_panel_users_uid_list. sB (**
**,,--r Fc | || || j v r$t j | j | }|d= |S t j | j d }|d= |S )z
Reads limits by user id
:param user_id: user/lve id
:param human_readable_numbers: if True PMEM and VMEM limits in KBytes, MBytes or GBytes
False - in bytes
:return: dictionary
)useridPACKAGEr )
_load_infor copydeepcopy)r user_idhuman_readable_numbers user_limitss r get_limits_by_user_idz LveCtl.get_limits_by_user_id9 su
.w??? d( ( ( -(:7(CDDKI& mD$6q$9:: "r c | || t j | j | }|S )a
Reads limits by reseller name
:param reseller_name:
:param: id - user/lve id
:param human_readable_numbers: if True PMEM and VMEM limits in KBytes, MBytes or GBytes
False - in bytes
:rtype: dict
)_load_resellers_infor( r) r get)r
reseller_namer+ reseller_limitss r get_limits_by_reseller_namez"LveCtl.get_limits_by_reseller_nameN sA
!!"8-HHH -(<(@(@ (O(OPP r c @ t j sdS d|ddg}|s| d t | j | \ }}t
j | }|dk rt |d | dg }|rd d
|d d i|d d d
|d d
i|d d |d d t |d d | t |d d | d}|S dS )a
Reads limits by reseller name
:param reseller_name: reseller's name
:param human_readable_numbers: if True PMEM and VMEM limits in KBytes, MBytes or GBytes
False - in bytes
:rtype: dict[str, str | dict]
N
list-reseller--json --with-name--bytesr msgdataDEFAULTallSPEEDEPIOIOPSNPROCPMEMVMEM)r& cpuepioiopsnprocpmemvmem) lvectlliblvereseller_limit_supportedr r r jsonloadsr r0 r ) r r1 r+ lvectl_argsret_codestd_out loaded_json default_infodefault_limitss r #get_default_limits_by_reseller_namez*LveCtl.get_default_limits_by_reseller_name\ sI }5577 F&
x O % * y ) ) ) ();[ I I'j)) q==!+e"455 5"vr22
"$ |Aw 78"1od+ l1od 34$Q/%a 1@ a QWAXAWY Y@ a QWAXAWY Y N " !
"
"r c | j 8ddlm} t || d }n\# t $ r d}Y nNw xY w|| j v r!t | j | d }n t | j d d }|S )z
Get package name by user id.
None package returns as str(None) for user checker compatibility:
'not str(None)' is True
:param: id - user/lve id
:return: Package name
Nr )reseller_package_by_uid r& )r clcommon.cpapirX str
IndexError)r r* rX package_names r get_package_name_by_user_idz"LveCtl.get_package_name_by_user_id s % > > > > > >
" "#:#:7#C#CA#F G G
"
"
" !
" $,,, "4#5g#>y#I J J #4#5a#8#C D D s . ==c $ d|||rd| dndddS )Nz+%(what)s set error for uid=%(uid)s%(error)sz []rZ )whatuiderror)messagecontextr
)r paramlve_iderrs r __set_errorzLveCtl.__set_error s< H$)&47%?[#[[[[R A AB B Br c F t | }|rt | dS )a
Check if PMEM value is below configured lower bound and raise exception if needed
:param pmem_value: PMEM value as string (e.g., "5m", "512M", etc.)
:raises LvectlException: If pmem is below configured lower bound or config is corrupted
N)r r )r
pmem_valuerd s r _check_pmem_lower_boundz LveCtl._check_pmem_lower_bound s1 3:>> )!%(( ( ) )r c 2 |t | g}d|v sd|v r,| d| d |d< dD ]}| | }|t | }|dv r | j dk rK|dv r | j d k r[|d
k r| r| d }|d k r| | | d
| d| t | dk rdS | d r| d |r| d| t | j |d \ }} }
|dk r$t | d||
dS )a-
Set limits for given container id
:param: str command: 'set' | 'set-reseller', based on container type
:param: int | str container_id: LVE | LVP id for set limits
:param: dict limits: new LVE limits. Available keys: speed, vmem, pmem, mep, io, nproc, iops
and 'save-all-parameters'.
All other keys are ignoring. If some parameter absent on current LVE version
(for example pmem on LVE4), it will be ignored too.
:param reseller_name: Reseller name
meprE
maxEntryProcs)speedrJ rI rp rF rH rG N)rI rH rG )rG rq %rI z--= r zsave-all-parametersz--save-all-parametersz --reseller=TstderrLimits) r\ r0 stripr isdigitrm r lenr r r _LveCtl__set_error) r command container_idlimitsr1 rP kvrQ outri s r _set_container_limits_by_idz"LveCtl._set_container_limits_by_id s L 1 12 F??dfnn&,jj
48H8H&I&IF? #R
-
-A
1
AyA A---$2Ba2G2GH}}!1Q!6!6G|| |GGGF{{,,Q/// |A|||| , , , , { q 1 ::+ , , 8 6 7 7 7 > <] < < = = = )$*>
# q==!$"2"2;"L"LMM Mqr c dt | dg}t | j | \ }}|dk r$t | d|| dS )z
Set unlimited LVP for reseller;
Accepts name or id;
:type lvp_id: str | int
:return: 0
r r r r r\ r r r r} )r r r rQ ri s r set_lvp_unlimitedzLveCtl.set_lvp_unlimited sZ F ];$T%7>>
# q==!$"2"2;"L"LMM Mqr c dt | dd | g}t | j | \ }}|dk r$t | d|| dS )z
Reset LVE limits for lve_id. Set default limits for LVE package or
system default LVE
:param: lve_id `int: LVE id
:return: 0
r
--default=,r Defaultr\ joinr r r r} )r rh r r rQ ri s r reset_lve_limitszLveCtl.reset_lve_limits so s6{{$C&1A1A$C$CD$T%7>>
# q==!$"2"29fc"J"JKK Kqr c dt | dd | g}t | j | \ }}|dk r$t | d|| dS )zk
Reset LVP limits for reseller_name.
:param: reseller_name str:
:return: 0
r r r r r r )r r1 r r rQ ri s r reset_reseller_limitsz LveCtl.reset_reseller_limits% sq M 2 24S&AQAQ4S4ST$T%7>>
# q==!$"2"29mS"Q"QRR Rqr c : t | j ddg \ }}|S )zO
Apply all already configured limits
:return: ret code
applyr< )r r )r rQ ri s r apply_all_limitszLveCtl.apply_all_limits1 s$
%T%7'59IJJ
#r c dt | dg}t | j | \ }}|dk r$t | d|| dS )z
Disable reseller limits for given name;
Equivalent to lvectl remove-reseller
:type reseller_name: str
:return: 0
zremove-resellerr6 r zDisable reseller limitsr )r r1 r rQ ri s r disable_reseller_limitsz LveCtl.disable_reseller_limits9 s^ "3}#5#5x@$T%7>>
# q==!$"2"23Lm]`"a"abb bqr c i | _ t j sdS g d}|s| d t | j | \ }}t j | }|dk rt |d t j D ]}|r||k r |i d| j |< | dg D ]|}|d
d \ }} |r|| k r)|| d
|d i|d d
|d
i|d |d t |d | dd}
|
| j |
d < }dS )z
Load information about resellers;
:type human_readable_numbers: bool
:type reseller_name: str | None
:return:
N)r5 r6 r7 r8 r r9 )namer r: ID:r< r= r> r? r@ rA rB )rD rE rF rG rH rI )idr r r )r rK rL rM r r r rN rO r r resellersr0 splitr ) r r+ r1 rP rQ rR rS resellerid_r
reseller_infos r r/ zLveCtl._load_resellers_infoF s " }5577 F@@@ % * y ) ) ) ();[ I I'j)) q==!+e"455 5 )) N NH
(!:!:6>"-M-MD * * $33 H HH ,,S11IC
$!6!6 !8G#45"4. (4.1 $V , %g . DXfEUE[ ] ]
M ;HD v!6 7 7# H Hr c " | j | || | j dS i | _ |dt | g}n|dt | g}ndg}| d |s| d t | j | \ }}t j | }|dk rt |d |d }|D ]} i }
| d } | d
k rd } | |
d < d | d
i|
d< d | d i|
d< t | d | |
d< | d |
d< | j dk r$t | d | |
d< | d |
d< | j dk r | d |
d < |
| j t | d < |$|
|| } | r | | j d< dS d| j vrPi }
d |
d < d d i|
d< d d i|
d< d |
d< d |
d< | j dk r
d |
d< d |
d< | j dk rd |
d < |
| j d< dS dS ) z
Loads all package info from lvectl
:param human_readable_numbers: if True PMEM and VMEM limits in KBytes, MBytes or GBytes
False - in bytes
:return: None
Npaneluserlimitspaneluserslimitsr6 r8 r r9 r: r&
VE_DEFAULTr; r< r= rD r? rF rC rJ r> rE rs rB rI rA rH r@ rG r 00K)r r/ r r\ r r r rN rO r r r r rV )
r r+ r% r rP rQ rR rS json_datapkg_data
pkg_limitspkg_namereseller_defaultss
r r' zLveCtl._load_infow s ' % %& 2 2 3 3 $ H HSi j j
(9"1% D& & &J$-Jy !!& Ju %s|Jt !%Jv "Jt 1$$%)
6"&)
7# 1$$%(
6"$.D q ! ! !! ' &r )F)N)NN)!r
r r __doc__MEGAHERZGIGAHERZr r _packages_limitsr r r staticmethodr# r- r3 rV r_ r} rm r r r r r r r r r r r r/ r' r
r r r r s HHK&MMO$ $ $ \ * !" !" !" !"F .B B B
) ) )(Q (Q (Q (QT
P
P
P
/H /H /HbK/ K/ K/ K/ K/ K/r r )r( rN rK clcommonr clcommon.clexceptionr cllimits.libr cllimits.lib.utilsr r r r r
r r r s 3 3 3 3 3 3 % % % % % %D D D D D D D D ( i/ i/ i/ i/ i/ i/ i/ i/ i/ i/r