HEX
Server: LiteSpeed
System: Linux server902.web-hosting.com 4.18.0-553.54.1.lve.el8.x86_64 #1 SMP Wed Jun 4 13:01:13 UTC 2025 x86_64
User: deshuvsd (2181)
PHP: 8.1.33
Disabled: NONE
Upload Files
File: //opt/cloudlinux/venv/lib/python3.11/site-packages/cllimits/__pycache__/lvectl_lib.cpython-311.pyc
�

�8iCF���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__���J/opt/cloudlinux/venv/lib64/python3.11/site-packages/cllimits/lvectl_lib.pyr	r	s�������Drr	c���eZdZdZdZdZdZdZdZdZ	dZ
d�Zed���Z
dd	�Zdd
�Zdd�Zd�Zd
�Zd�Zdd�Zd�Zdd�Zd�Zd�Zdd�Zd�Zd�Zd�Zd�Zd�Zd�Zdd�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�@�tj���sdSd|ddg}|s|�d��t	|j|��\}}t
j|��}|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 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_name\sI���}�5�5�7�7�	��F�&�
�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.cpapirX�str�
IndexError)rr*rX�package_names    r�get_package_name_by_user_idz"LveCtl.get_package_name_by_user_ids�����%�>�>�>�>�>�>�
"�"�#:�#:�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 [�]rZ)�what�uid�error)�message�contextr
)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_valuerds   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
        �meprE�
maxEntryProcs)�speedrJrIrprFrHrGN)rIrHrG�)rG�rq�%rIz--�=�rzsave-all-parametersz--save-all-parametersz--reseller=T��stderr�Limits)r\r0�stripr�isdigitrmr �lenrrr	�_LveCtl__set_error)r�command�container_id�limitsr1rP�k�vrQ�outris           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|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
        �set-resellerTrwrryN)rrr	r})r�reseller_idrPrQr�ris      r�#_set_container_limits_by_id_or_namez*LveCtl._set_container_limits_by_id_or_name�s]��&�{�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
        rY�set)r1r)�boolr_r�)rrhr�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_id�s"��	
�(�(�)?���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})rrhr1�argsrQris      r�set_lve_unlimitedzLveCtl.set_lve_unlimited�sw���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\rrr	r})rr�r�rQris     r�set_lvp_unlimitedzLveCtl.set_lvp_unlimitedsZ����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})rrhr�r�rQris      r�reset_lve_limitszLveCtl.reset_lve_limitsso���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�)rr1r�r�rQris      r�reset_reseller_limitszLveCtl.reset_reseller_limits%sq����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
        �applyr<)rr)rrQris   r�apply_all_limitszLveCtl.apply_all_limits1s$��
%�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-resellerr6rzDisable reseller limitsr�)rr1r�rQris     r�disable_reseller_limitszLveCtl.disable_reseller_limits9s^��"�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)r5r6r7r8rr9)�namer�r:�ID�:r<r=r>r?r@rArB)rDrErFrGrHrI)�idr�r�r�)rrKrLrMr rrrNrOr	r�	resellersr0�splitr)rr+r1rPrQrRrS�reseller�id_r��
reseller_infos           rr/zLveCtl._load_resellers_infoFs��� "����}�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�paneluserslimitsr6r8rr9r:r&�
VE_DEFAULTr;r<r=rDr?rFrCrJr>rErsrBrIrArH�r@rGr��0�0K)rr/rr\r rrrNrOr	rrrrV)
rr+r%r�rPrQrRrS�	json_data�pkg_data�
pkg_limits�pkg_name�reseller_defaultss
             rr'zLveCtl._load_infows�����'��%�%�&<�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-r3rVr_r}rmr�r�r�r�r�r�r�r�r�r�r�r/r'r
rrrrs����������H��H��K�&�M����M��O�$�$�$�����\������*����!"�!"�!"�!"�F���.B�B�B�
)�)�)�(Q�(Q�(Q�(Q�T
P�
P�
P�����������
�
�
�
�������
�
�
�������/H�/H�/H�bK/�K/�K/�K/�K/�K/rr)r(rNrK�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�	�	�	�	�	�(�	�	�	�i/�i/�i/�i/�i/�i/�i/�i/�i/�i/r