HEX
Server: Apache
System: Linux www3.pit.tblive.com 5.14.0-687.38.1.el9_8.x86_64 #1 SMP PREEMPT_DYNAMIC Wed Aug 12 17:19:12 EDT 2026 x86_64
User: awaldron (1020)
PHP: 8.1.34
Disabled: exec,passthru,shell_exec,system
Upload Files
File: //opt/cloudlinux/venv/lib/python3.11/site-packages/cllimits/__pycache__/lvectl_lib.cpython-311.pyc
�

}�C
hfU���ddlZddlZddlZddlmZddlmZddlmZddl	m
Z
mZGd�de��ZGd�d	��Z
dS)
�N)�cpapi)�FormattedException)�exec_utility)�(_convert_memory_value_to_adaptive_format�"check_pmem_lower_bound_from_stringc��eZdZdS)�LvectlExceptionN)�__name__�
__module__�__qualname__���py/cllimits/lvectl_lib.pyr	r	s�������Drr	c���eZdZdZdZdZdZdZdZdZ	dZ
d�Zed���Z
dd	�Zdd
�Zedd���Zdd�Zd
�Zd�Zd�Zdd�Zd�Zdd�Zd�Zd�Zdd�Zd�Zd�Zd�Zd�Zd�Zd�Z dd�Z!dS) �LveCtlz*
    Library for work with LVE limits
    i@Biʚ;Nz/usr/sbin/lvectlc�|�t|jdg��\}}t|��|_d|_d|_dS)z5
        Object constructor. Get LVE version
        zlve-versionN)r�
_UTILITY_PATH�int�lve_version�
_package_data�_resellers_data)�self�_�
s_lve_versions   r�__init__zLveCtl.__init__%sC��(��(:�]�O�L�L���=��}�-�-���!���#����rc��tj��}t|�����}|�d��|S)zH
        Get panel users uid list
        :return: List of uids
        r)r�
list_users�list�keys�append)�
user_packages�uid_lists  r�get_panel_users_uid_listzLveCtl.get_panel_users_uid_list.sB���(�*�*�
��
�*�*�,�,�-�-����������rFc���|�||���||jvr$tj|j|��}|d=|Stj|jd��}|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
        )�userid�PACKAGEr)�
_load_infor�copy�deepcopy)r�user_id�human_readable_numbers�user_limitss    r�get_limits_by_user_idzLveCtl.get_limits_by_user_id9su��	
���.�w��?�?�?��d�(�(�(��-��(:�7�(C�D�D�K��I�&����m�D�$6�q�$9�:�:���	�"��rc��|�||��tj|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_nameNsA��	
�!�!�"8�-�H�H�H��-��(<�(@�(@��(O�(O�P�P���rc��t|��}|�d��r|r|dkstdd|id����dSdS)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)s�value��message�contextN)�str�
startswithr	)�
identifier�	allow_allr6s   r�_reject_option_like_identifierz%LveCtl._reject_option_like_identifier\sj���J�������C� � �	�)�	���8H�8H�!�D�RY�[`�Qa�b�b���
�	�	�8H�8Hrc�j�tj���sdS|�|��d|ddg}|s|�d��t|j|��\}}tj|��}|dkrt|d���|�
dg��}|r�d	d
|ddi|ddd
|dd
i|dd|ddt|dd|��t|dd|��d�}|SdS)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�msg�data�DEFAULT�all�SPEED�EP�IO�IOPS�NPROC�PMEM�VMEM)r&�cpu�ep�io�iops�nproc�pmem�vmem)�	lvectllib�lve�reseller_limit_supportedr>r rr�json�loadsr	r0r)	rr1r+�lvectl_args�ret_code�std_out�loaded_json�default_info�default_limitss	         r�#get_default_limits_by_reseller_namez*LveCtl.get_default_limits_by_reseller_nameis]���}�5�5�7�7�	��F��+�+�M�:�:�:�&�
�x��O��%�	*����y�)�)�)�(��);�[�I�I���'��j��)�)���q�=�=�!�+�e�"4�5�5�5�"���v�r�2�2���
	"�$��|�A��w�7�8�"�1�o�d�+��l�1�o�d�3�4�$�Q���/�%�a���1�@��a��QW�AX�AW�Y�Y�@��a��QW�AX�AW�Y�Y���N�"�!�
	"�
	"rc��|j�8ddlm}	t||��d��}n\#t$rd}YnNwxYw||jvr!t|j|d��}n t|jdd��}|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.cpapircr:�
IndexError)rr*rc�package_names    r�get_package_name_by_user_idz"LveCtl.get_package_name_by_user_id�s�����%�>�>�>�>�>�>�
"�"�#:�#:�7�#C�#C�A�#F�G�G�����
"�
"�
"�!����
"�����$�,�,�,�"�4�#5�g�#>�y�#I�J�J��� #�4�#5�a�#8��#C�D�D���s�.�=�=c�$�d|||rd|�d�ndd�d�S)Nz+%(what)s set error for uid=%(uid)s%(error)sz [�]re)�what�uid�errorr7r
)r�param�lve_id�errs    r�__set_errorzLveCtl.__set_error�s<��H�$)�&�47�%?�[�#�[�[�[�[�R�A�A�B�B�	Brc�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)rr	)r�
pmem_valuerns   r�_check_pmem_lower_boundzLveCtl._check_pmem_lower_bound�s1��3�:�>�>���	)�!�%�(�(�(�	)�	)rc�2�|t|��g}d|vsd|vr,|�d|�d����|d<dD]�}|�|��}|��t|�����}|dvr|jdkr�K|dvr|jd	kr�[|d
kr|���r|�d�}|dkr|�|��|�d
|�d|������t|��dkrdS|�d��r|�d��|r|�d|����t|j	|d���\}}	}
|dkr$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
        �meprP�
maxEntryProcs)�speedrUrTrxrQrSrRN)rTrSrR�)rR�ry�%rTz--�=�rzsave-all-parametersz--save-all-parametersz--reseller=T��stderr�Limits)r:r0�stripr�isdigitrur �lenrrr	�_LveCtl__set_error)r�command�container_id�limitsr1r[�k�vr\�outrqs           r�_set_container_limits_by_idz"LveCtl._set_container_limits_by_id�s�����L� 1� 1�2���F�?�?�d�f�n�n�&,�j�j���
�
�4�8H�8H�&I�&I�F�?�#�R�
	-�
	-�A��
�
�1�
�
�A��y���A�������A��-�-�-�$�2B�a�2G�2G���H�}�}��!1�Q�!6�!6���G�|�|��	�	���|��G�G�G���F�{�{��,�,�Q�/�/�/����|�A�|�|��|�|�,�,�,�,��{���q� � ��1��:�:�+�,�,�	8����6�7�7�7��	>����<�]�<�<�=�=�=�)�$�*<�k�RV�W�W�W���#�s��q�=�=�!�$�"2�"2�8�\�3�"O�"O�P�P�P��=rc���|�|d���d|g}t|j|d���\}}}|dkr$t|�d||�����dS)z�
        Set limits for given container id
        :param reseller_id: LVP id or reseller's name or '--all'
        :type reseller_id: int | str
        T)r=�set-resellerrrr�N)r>rrr	r�)r�reseller_idr[r\r�rqs      r�#_set_container_limits_by_id_or_namez*LveCtl._set_container_limits_by_id_or_name�sv��	
�+�+�K�4�+�H�H�H�%�{�3��)�$�*<�k�RV�W�W�W���#�s��q�=�=�!�$�"2�"2�8�[�#�"N�"N�O�O�O��=rc��t|��r|�|��sdS|�d|||���dS)z�
        Wrapper for _set_container_limits_by_id, set limits for lve_id;
        :param int lve_id: user's container id
        :param limits: dict with limits to set
        :param reseller_name: Reseller name
        :return: int
        rd�set)r1r)�boolrir�)rrpr�r1s    r�set_lve_limits_by_user_idz LveCtl.set_lve_limits_by_user_id�sN����<�<�	�� @� @�� H� H�	��1��(�(����m�(�\�\�\��qrc�4�|�d||��dS)��
        Wrapper for _set_container_limits_by_id, set limits for lvp_id;
        :type lvp_id: int
        :type limits: dict
        :return: int
        r�r�r��r�lvp_idr�s   r�set_lvp_limits_by_reseller_idz$LveCtl.set_lvp_limits_by_reseller_id�s!��	
�(�(����H�H�H��qrc�4�|�d||��dS)r�zset-reseller-defaultrr�r�s   r�set_lvp_defaults_by_reseller_idz&LveCtl.set_lvp_defaults_by_reseller_ids"��	
�(�(�)?���P�P�P��qrc���dt|��dg}|�|�d|g��t|j|��\}}|dkr$t	|�d||�����dS)z�
        Set unlimited LVE for lve_id
        :param: lve_id `int`: LVE id
        :param reseller_name: Reseller name
        :return: 0
        r��--unlimitedNz
--resellerr�	Unlimited)r:�extendrrr	r�)rrpr1�argsr\rqs      r�set_lve_unlimitedzLveCtl.set_lve_unlimitedsw���s�6�{�{�M�2���$��K�K��}�5�6�6�6�$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�;���"L�"L�M�M�M��qrc���|�|��dt|��dg}t|j|��\}}|dkr$t	|�d||�����dS)z�
        Set unlimited LVP for reseller;
        Accepts name or id;
        :type lvp_id: str | int
        :return: 0
        r�r�rr��r>r:rrr	r�)rr�r�r\rqs     r�set_lvp_unlimitedzLveCtl.set_lvp_unlimitedsn��	
�+�+�F�3�3�3���F���]�;��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�;���"L�"L�M�M�M��qrc���dt|��dd�|����g}t|j|��\}}|dkr$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�Default)r:�joinrrr	r�)rrpr�r�r\rqs      r�reset_lve_limitszLveCtl.reset_lve_limits(so���s�6�{�{�$C����&�1A�1A�$C�$C�D��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�9�f�c�"J�"J�K�K�K��qrc��|�|��dt|��dd�|����g}t|j|��\}}|dkr$t|�d||�����dS)zk
        Reset LVP limits for reseller_name.
        :param: reseller_name str:
        :return: 0
        r�r�r�rr�)r>r:r�rrr	r�)rr1r�r�r\rqs      r�reset_reseller_limitszLveCtl.reset_reseller_limits5s���	
�+�+�M�:�:�:���M� 2� 2�4S����&�AQ�AQ�4S�4S�T��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�9�m�S�"Q�"Q�R�R�R��qrc�:�t|jddg��\}}|S)zO
        Apply all already configured limits
        :return: ret code
        �applyrG)rr)rr\rqs   r�apply_all_limitszLveCtl.apply_all_limitsBs$��
%�T�%7�'�5�9I�J�J�
��#��rc���|�|��dt|��dg}t|j|��\}}|dkr$t	|�d||�����dS)z�
        Disable reseller limits for given name;
        Equivalent to lvectl remove-reseller <str>
        :type reseller_name: str
        :return: 0
        zremove-resellerrArzDisable reseller limitsr�)rr1r�r\rqs     r�disable_reseller_limitszLveCtl.disable_reseller_limitsJsr��	
�+�+�M�:�:�:�!�3�}�#5�#5�x�@��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�3L�m�]`�"a�"a�b�b�b��qrc��i|_tj���sdSgd�}|s|�d��t|j|��\}}tj|��}|dkrt|d���tj��D]}|r||kr�|id�|j|<�|�dg��D]|}|d�
d	��\}}	|r||	kr�)||	d
|di|dd
|d
i|d|dt|d|��d�d�}
|
|j|
d<�}dS)z�
        Load information about resellers;
        :type human_readable_numbers: bool
        :type reseller_name: str | None
        :return:
        N)r@rArBrCrrD)�namer�rE�ID�:rGrHrIrJrKrLrM)rOrPrQrRrSrT)�idr�r�r�)rrVrWrXr rrrYrZr	r�	resellersr0�splitr)rr+r1r[r\r]r^�reseller�id_r��
reseller_infos           rr/zLveCtl._load_resellers_infoXs��� "����}�5�5�7�7�	��F�@�@�@��%�	*����y�)�)�)�(��);�[�I�I���'��j��)�)���q�=�=�!�+�e�"4�5�5�5���)�)�	N�	N�H��
��(�!:�!:��6>�"�-M�-M�D� ��*�*�$�����3�3�	H�	H�H� ���,�,�S�1�1�I�C���
��$�!6�!6����!�8�G�#4�5�"�4�.� �(�4�.�1�$�V�,�%�g�.�D�X�f�EU�E[�]�]�
����M�;H�D� ��v�!6�7�7�#	H�	Hrc�"�|j�|�||��|j�dSi|_|�dt|��g}n|�dt|��g}ndg}|�d��|s|�d��t|j|��\}}tj|��}|dkrt|d���|d}|D]�}	i}
|	d	}|d
krd}||
d	<d|	d
i|
d<d|	di|
d<t|	d|��|
d<|	d|
d<|jdkr$t|	d|��|
d<|	d|
d<|jdkr|	d|
d<|
|jt|	d��<��|�$|�
||��}|r||jd<dSd|jvrPi}
d|
d	<ddi|
d<ddi|
d<d|
d<d|
d<|jdkr
d|
d<d|
d<|jdkrd|
d<|
|jd<dSdS) 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
        N�paneluserlimits�paneluserslimitsrArCrrDrEr&�
VE_DEFAULTrFrGrHrOrJrQrNrUrIrPr{rMrTrLrS�rKrRr��0�0K)rr/rr:r rrrYrZr	rrrra)
rr+r%r�r[r\r]r^�	json_data�pkg_data�
pkg_limits�pkg_name�reseller_defaultss
             rr'zLveCtl._load_info�s�����'��%�%�&<�h�G�G�G���)��F� �����,�c�&�k�k�:�K�K�
�
!�-�s�8�}�}�=�K�K�-�.�K����8�$�$�$�%�	*����y�)�)�)�(��);�[�I�I���'��j��)�)���q�=�=�!�+�e�"4�5�5�5���'�	�!�	A�	A�H��J��	�*�H��<�'�'�$��$,�J�y�!�!&���(9� :�J�u�� %�x��~�6�J�t��!I�(�SY�JZ�J`�"b�"b�J�v��'��~�J�t����1�$�$�%M�h�W]�N^�Nd�&f�&f�
�6�"�&.�w�&7�
�7�#���1�$�$�%-�f�%5�
�6�"�6@�D��s�8�D�>�2�2�3�3��� $� H� H��Si� j� j�� �
�(9��"�1�%����D�&�&�&��J�$-�J�y�!�!&���J�u�� %�s�|�J�t��!%�J�v��"�J�t����1�$�$�%)�
�6�"�&)�
�7�#���1�$�$�%(�
�6�"�$.�D��q�!�!�!�!'�&r)F)N)NN)"r
rr�__doc__�MEGAHERZ�GIGAHERZrr�_packages_limitsrrr�staticmethodr#r-r3r>rarir�rur�r�r�r�r�r�r�r�r�r�r�r/r'r
rrrrs����������H��H��K�&�M����M��O�$�$�$�����\������*�����
�
�
��\�
�""�""�""�""�H���.B�B�B�
)�)�)�(Q�(Q�(Q�(Q�TP�P�P�����������
�
�
�
����������������/H�/H�/H�bK/�K/�K/�K/�K/�K/rr)r(rYrV�clcommonr�clcommon.clexceptionr�cllimits.libr�cllimits.lib.utilsrrr	rr
rr�<module>r�s���������������������3�3�3�3�3�3�%�%�%�%�%�%�D�D�D�D�D�D�D�D�	�	�	�	�	�(�	�	�	�{/�{/�{/�{/�{/�{/�{/�{/�{/�{/r