@jK d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z d dl
m Z m Z G d de Z
G d d ZdS )
N)cpapi)FormattedException) exec_utilityisolated_child_env)(_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
Ze d d Zd d Zd
Zd Zd Zd dZd Zd dZd Zd Zd dZd Zd Z d Z d Z d Z d Z d d Z!dS ) LveCtlz*
Library for work with LVE limits
i@B i ʚ;Nz/usr/sbin/lvectlc t | j dgt \ }}t | | _ d| _ d| _ dS )z5
Object constructor. Get LVE version
z lve-versionenvN)r
_UTILITY_PATHr int lve_version
_package_data_resellers_data)self_
s_lve_versions r __init__zLveCtl.__init__% sO ((:]OQcQeQefff= }-- ! # 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 | }| d r |r|dk st dd|id dS dS )z
Reject reseller identifiers that lvectl would parse as an option or the
special '--all' target. Only the documented '--all' enable-all sentinel
is permitted, and only where the caller explicitly allows it.
-z--allz&Invalid reseller identifier: %(value)svaluemessagecontextN)str
startswithr
)
identifier allow_allr9 s r _reject_option_like_identifierz%LveCtl._reject_option_like_identifier\ sj J C ) 8H8H! DRY[`Qabb
8H8Hr c t j sdS | | d|ddg}|s| d t | j |t \ }}t j | }|dk rt |d | d g }|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 r msgdataDEFAULTallSPEEDEPIOIOPSNPROCPMEMVMEM)r) cpuepioiopsnprocpmemvmem)
lvectlliblvereseller_limit_supportedrA r# r r r jsonloadsr
r3 r ) r r4 r. lvectl_argsret_codestd_out loaded_json default_infodefault_limitss r #get_default_limits_by_reseller_namez*LveCtl.get_default_limits_by_reseller_namei si }5577 F ++M:::&
x O % * y ) ) ) ();[N`NbNb c c c'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.cpapirf r=
IndexError)r r- rf 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 []rh )whatuiderrorr: r )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_valuerq s r _check_pmem_lower_boundz LveCtl._check_pmem_lower_bound s1 3:>> )!%(( ( ) )r c L |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 |dt \ }} }
|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
meprS
maxEntryProcs)speedrX rW r{ rT rV rU N)rW rV rU )rU r| %rW z--= r zsave-all-parametersz--save-all-parametersz --reseller=Tstderrr Limits)
r= r3 stripr isdigitrx r# lenr r r r
_LveCtl__set_error) r command container_idlimitsr4 r^ kvr_ outrt 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 > <] < < = = = )$*
:type reseller_name: str
:return: 0
zremove-resellerrD r r zDisable reseller limitsr )r r4 r r_ rt s r disable_reseller_limitsz LveCtl.disable_reseller_limitsJ s~
++M:::!3}#5#5x@$T%7CUCWCWXXX
# q==!$"2"23Lm]`"a"abb bqr c i | _ t j sdS g d}|s| d t | j |t \ }}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)rC rD rE rF r r rG )namer rH ID:rJ rK rL rM rN rO rP )rR rS rT rU rV rW )idr r r )r rY rZ r[ r# r r r r\ r] r
r resellersr3 splitr ) r r. r4 r^ r_ r` ra resellerid_r
reseller_infos r r2 zLveCtl._load_resellers_infoX s " }5577 F@@@ % * y ) ) ) ();[N`NbNb c c c'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 \ }}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
NpaneluserlimitspaneluserslimitsrD rF r r rG rH r)
VE_DEFAULTrI rJ rK rR rM rT rQ rX rL rS r~ rP rW rO rV rN rU r 00K)r r2 r r= r# r r r r\ r] r
r r r rd )
r r. r( r r^ r_ r` ra json_datapkg_data
pkg_limitspkg_namereseller_defaultss
r r* zLveCtl._load_info 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& r0 r6 rA rd rl r rx r r r r r r r r r r r r2 r* r r r r r s HHK&MMO$ $ $ \ *
\
"" "" "" ""H .B B B
) ) )(Q (Q (Q (QT P P P
/H /H /HbK/ K/ K/ K/ K/ K/r r )r+ r\ rY clcommonr clcommon.clexceptionr cllimits.libr r cllimits.lib.utilsr r r
r r r r r s 3 3 3 3 3 3 9 9 9 9 9 9 9 9D D D D D D D D ( {/ {/ {/ {/ {/ {/ {/ {/ {/ {/r