Your IP : 18.220.236.72


Current Path : /opt/cloudlinux/venv/lib/python3.11/site-packages/cllimits/__pycache__/
Upload File :
Current File : //opt/cloudlinux/venv/lib/python3.11/site-packages/cllimits/__pycache__/lvectl_lib.cpython-311.pyc

�

T�g�C���ddlZddlZddlZddlmZddlmZddlmZddl	m
Z
Gd�de��ZGd�d	��ZdS)
�N)�cpapi)�FormattedException)�exec_utility)�(_convert_memory_value_to_adaptive_formatc��eZdZdS)�LvectlExceptionN)�__name__�
__module__�__qualname__���J/opt/cloudlinux/venv/lib64/python3.11/site-packages/cllimits/lvectl_lib.pyrrs�������Dr
rc���eZdZdZdZdZdZdZdZdZ	dZ
d�Zed���Z
dd	�Zdd
�Zdd�Zd�Zd
�Zdd�Zd�Zdd�Zd�Zd�Zdd�Zd�Zd�Zd�Zd�Zd�Zd�Zdd�ZdS)�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���=��}�-�-���!���#����r
c��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���(�*�*�
��
�*�*�,�,�-�-����������r
Fc���|�||���||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_id8su��	
���.�w��?�?�?��d�(�(�(��-��(:�7�(C�D�D�K��I�&����m�D�$6�q�$9�:�:���	�"��r
c��|�||��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_nameMsA��	
�!�!�"8�-�H�H�H��-��(<�(@�(@��(O�(O�P�P���r
c�@�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_supportedrrr�json�loadsrr/r)	rr0r*�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�"�!�
	"�
	"r
c��|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.cpapirW�str�
IndexError)rr)rW�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 [�]rY)�what�uid�error)�message�contextr)r�param�lve_id�errs    r�__set_errorzLveCtl.__set_error�s<��H�$)�&�47�%?�[�#�[�[�[�[�R�A�A�B�B�	Br
c��|t|��g}d|vsd|vr,|�d|�d����|d<dD]�}|�|��}|��t|�����}|dvr|jdkr�K|dvr|jd	kr�[|d
kr|���r|�d�}|�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
        �meprD�
maxEntryProcs)�speedrIrHrlrErGrFN)rHrGrF�)rF�rm�%z--�=�rzsave-all-parametersz--save-all-parametersz--reseller=T��stderr�Limits)r[r/�stripr�isdigitr�lenrrr�_LveCtl__set_error)r�command�container_id�limitsr0rO�k�vrP�outrhs           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�����|�A�|�|��|�|�,�,�,�,��{���q� � ��1��:�:�+�,�,�	8����6�7�7�7��	>����<�]�<�<�=�=�=�)�$�*<�k�RV�W�W�W���#�s��q�=�=�!�$�"2�"2�8�\�3�"O�"O�P�P�P��=r
c��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-resellerTrsrruN)rrrry)r�reseller_idrOrPrrhs      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��=r
c��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
        rX�set)r0r)�boolr^r�)rrgr|r0s    r�set_lve_limits_by_user_idz LveCtl.set_lve_limits_by_user_id�sN����<�<�	�� @� @�� H� H�	��1��(�(����m�(�\�\�\��qr
c�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��qr
c�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��qr
c���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[�extendrrrry)rrgr0�argsrPrhs      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��qr
c��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[rrrry)rr�r�rPrhs     r�set_lvp_unlimitedzLveCtl.set_lvp_unlimited�sZ����F���]�;��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�;���"L�"L�M�M�M��qr
c���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[�joinrrrry)rrgr|r�rPrhs      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��qr
c���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�)rr0r|r�rPrhs      r�reset_reseller_limitszLveCtl.reset_reseller_limitssq����M� 2� 2�4S����&�AQ�AQ�4S�4S�T��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�9�m�S�"Q�"Q�R�R�R��qr
c�:�t|jddg��\}}|S)zO
        Apply all already configured limits
        :return: ret code
        �applyr;)rr)rrPrhs   r�apply_all_limitszLveCtl.apply_all_limits$s$��
%�T�%7�'�5�9I�J�J�
��#��r
c��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-resellerr5rzDisable reseller limitsr�)rr0r�rPrhs     r�disable_reseller_limitszLveCtl.disable_reseller_limits,s^��"�3�}�#5�#5�x�@��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�3L�m�]`�"a�"a�b�b�b��qr
c��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)r4r5r6r7rr8)�namer|r9�ID�:r;r<r=r>r?r@rA)rCrDrErFrGrH)�idr�r|r�)rrJrKrLrrrrMrNrr�	resellersr/�splitr)rr*r0rOrPrQrR�reseller�id_r��
reseller_infos           rr.zLveCtl._load_resellers_info9s��� "����}�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�	Hr
c�"�|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�paneluserslimitsr5r7rr8r9r%�
VE_DEFAULTr:r;r<rCr>rErBrIr=rDrorArHr@rG�r?rFr��0�0K)rr.rr[rrrrMrNrrrrrU)
rr*r$r�rOrPrQrR�	json_data�pkg_data�
pkg_limits�pkg_name�reseller_defaultss
             rr&zLveCtl._load_infojs�����'��%�%�&<�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	r
r�__doc__�MEGAHERZ�GIGAHERZrr�_packages_limitsrrr�staticmethodr"r,r2rUr^ryr�r�r�r�r�r�r�r�r�r�r�r.r&rr
rrrs����������H��H��K�&�M����M��O�$�$�$�����\������*����!"�!"�!"�!"�F���.B�B�B�
&Q�&Q�&Q�&Q�P
P�
P�
P�����������
�
�
�
�������
�
�
�������/H�/H�/H�bK/�K/�K/�K/�K/�K/r
r)
r'rMrJ�clcommonr�clcommon.clexceptionr�cllimits.libr�cllimits.lib.utilsrrrrr
r�<module>r�s���������������������3�3�3�3�3�3�%�%�%�%�%�%�G�G�G�G�G�G�	�	�	�	�	�(�	�	�	�]/�]/�]/�]/�]/�]/�]/�]/�]/�]/r

?>