????

Your IP : 3.14.141.163


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

�

�o�eD���ddlmZddlmZddlZddlZddlZddlmZddl	m
Z
ddlmZddl
mZGd�d	e
��ZGd
�de��ZdS)�)�absolute_import)�unicodeN)�exec_utility)�FormattedException)�cpapi)�(_convert_memory_value_to_adaptive_formatc��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
�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���=��}�-�-���!���#����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_id:su��	
���.�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_nameOsA��	
�!�!�"8�-�H�H�H��-��(<�(@�(@��(O�(O�P�P���rc
��tj���sdSd|ddg}|s|�d��t	|j|��\}}t
j|��}|dkrt|d���|�	dg��}|r�td	t|dd
���|ddt|dd
���|dd|ddt|dd|��t|dd|�����}|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�SPEED��all�EP�IO�IOPS�NPROC�PMEM�VMEM)r'�cpu�ep�io�iops�nproc�pmem�vmem)�	lvectllib�lve�reseller_limit_supportedr!rr�json�loadsr
r1�dictr)	rr2r,�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]s[���}�5�5�7�7�	��F�&�
�x��O��%�	*����y�)�)�)�(��);�[�I�I���'��j��)�)���q�=�=�!�+�e�"4�5�5�5�"���v�r�2�2���
	"�!�!��\�!�_�W�5�6�6�6���?�4�(��L��O�D�1�2�2�2�!�!�_�V�,�"�1�o�g�.�=�l�1�o�f�>U�>T�V�V�=�l�1�o�f�>U�>T�V�V����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.cpapir[�str�
IndexError)rr+r[�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|zndd�d�S)Nz+%(what)s set error for uid=%(uid)s%(error)sz [%s]r])�what�uid�error)�message�contextr)r�param�lve_id�errs    r�__set_errorzLveCtl.__set_error�s8��H�$)�&�69�%A�W�s�]�]�r�C�C�D�D�	Drc��|t|��g}d|vsd|vr,|�d|�d����|d<dD]�}|�|��}|��t|�����}|dvr|jdkr�K|dvr|jd	kr�[|d
kr|���rd|z}|�d|�d
|������t
|��dkrdS|�d��r|�d��|r|�d|z��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
        �meprG�
maxEntryProcs)�speedrLrKrorHrJrIN)rKrJrI�)rI�rpz%s%%z--�=�rzsave-all-parametersz--save-all-parametersz
--reseller=%sT��stderr�Limits)r_r1�stripr�isdigitr!�lenrrr
�_LveCtl__set_error)r�command�container_id�limitsr2rS�k�vrT�outrks           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�	3�	3�A��
�
�1�
�
�A��y���A�������A��-�-�-�$�2B�a�2G�2G���h���4�#3�q�#8�#8���G�|�|��	�	���|��Q�J������A�A�A�q�q�1�2�2�2�2��{���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-resellerTrurrwN�rrr
r{)r�reseller_idrSrTr�rks      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
        r\�set)r2r)�boolrbr�)rrjr~r2s    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���dd|zdg}|�|�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��%s�--unlimitedNz
--resellerr�	Unlimited)�extendrrr
r{)rrjr2�argsrTrks      r�set_lve_unlimitedzLveCtl.set_lve_unlimited�su���t�f�}�m�4���$��K�K��}�5�6�6�6�$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�;���"L�"L�M�M�M��qrc��dd|zdg}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�r�rr�r�)rr�r�rTrks     r�set_lvp_unlimitedzLveCtl.set_lvp_unlimitedsX����v�
�}�=��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�;���"L�"L�M�M�M��qrc���dd|zdd�|��zg}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�r��--default=%s�,r�Default��joinrrr
r{)rrjr~r�rTrks      r�reset_lve_limitszLveCtl.reset_lve_limits
si���t�f�}�n�s�x�x��7G�7G�&G�H��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�9�f�c�"J�"J�K�K�K��qrc���dd|zdd�|��zg}t|j|��\}}|dkr$t|�d||�����dS)zk
        Reset LVP limits for reseller_name.
        :param: reseller_name str:
        :return: 0
        r�r�r�r�rr�r�)rr2r~r�rTrks      r�reset_reseller_limitszLveCtl.reset_reseller_limitssk����}� 4�n�s�x�x�PV�GW�GW�6W�X��$�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)rrTrks   r�apply_all_limitszLveCtl.apply_all_limits&s$��
%�T�%7�'�5�9I�J�J�
��#��rc��dd|zdg}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-resellerr�r7rzDisable reseller limitsr�)rr2r�rTrks     r�disable_reseller_limitszLveCtl.disable_reseller_limits.s[��"�4�-�#7��B��$�T�%7��>�>�
��#��q�=�=�!�$�"2�"2�3L�m�]`�"a�"a�b�b�b��qrc�(�t��|_tj���sdSgd�}|s|�d��t
|j|��\}}tj	|��}|dkrt|d���tj��D]%}|r||kr�t|��id�|j|<�&|�dg��D]�}|d�d	��\}}	|r||	kr�)t||	tt|d
���|dt|d
���|d|dt!|d|��������}
|
|j|
d<��dS)z�
        Load information about resellers;
        :type human_readable_numbers: bool
        :type reseller_name: str | None
        :return:
        N)r6r7r8r9rr:)�namer~r;�ID�:r=r>r@rArBrCrD)rFrGrHrIrJrK)�idr�r~r�)rRrrMrNrOr!rrrPrQr
r�	resellersrr1�splitr)rr,r2rSrTrUrV�reseller�id_r��
reseller_infos           rr0zLveCtl._load_resellers_info;s��� $�v�v����}�5�5�7�7�	��F�@�@�@��%�	*����y�)�)�)�(��);�[�I�I���'��j��)�)���q�=�=�!�+�e�"4�5�5�5���)�)�	W�	W�H��
��(�!:�!:��6=�h�6G�6G�SU�-V�-V�D� ��*�*�$�����3�3�	H�	H�H� ���,�,�S�1�1�I�C���
��$�!6�!6�� ������'�!2�3�3�3���~�����/�/�/�!�&�)�"�7�+�A�(�6�BR�BX�Z�Z�
������M�;H�D� ��v�!6�7�7�#	H�	Hrc�j�|j�|�||��|j�dSt��|_|�dt	|��g}n|�dt	|��g}ndg}|�d��|s|�d��t
|j|��\}}tj	|��}|dkrt|d���|d}|D]�}	t��}
|	d	}|d
krd}||
d	<d|	d
i|
d<d|	di|
d<t|	d|��|
d<|	d|
d<|jdkr$t|	d|��|
d<|	d|
d<|jdkr|	d|
d<|
|jt|	d��<��|�$|�||��}|r||jd<dSd|jvr\t��}
d|
d	<ddi|
d<ddi|
d<d|
d<d|
d<|jdkr
d|
d<d|
d<|jdkrd|
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�paneluserslimitsr7r9rr:r;r'�
VE_DEFAULTr<r?r=rFrArHrErLr@rGrrrDrKrCrJ�rBrIr��0�0K)rr0rrRr_r!rrrPrQr
rrrrY)
rr,r&r�rSrTrUrV�	json_data�pkg_data�
pkg_limits�pkg_name�reseller_defaultss
             rr(zLveCtl._load_infols�����'��%�%�&<�h�G�G�G���)��F�"�V�V�����,�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�� %�t�}�J�t��!&�J�v��#�J�t����1�$�$�%*�
�6�"�&*�
�7�#���1�$�$�%)�
�6�"�$.�D��q�!�!�!�!'�&r)F)N)NN) rrr
�__doc__�MEGAHERZ�GIGAHERZrr�_packages_limitsrrr�staticmethodr$r.r4rYrbr{r�r�r�r�r�r�r�r�r�r�r�r0r(rrrrrs����������H��H��K�&�M����M��O�$�$�$�����\������*����!"�!"�!"�!"�F���.D�D�D�
&Q�&Q�&Q�&Q�P
P�
P�
P�����������
�
�
�
�������
�
�
�������/H�/H�/H�bK/�K/�K/�K/�K/�K/rr)�
__future__r�
past.builtinsrrMrPr)�cllimits.libr�clcommon.clexceptionr�clcommonr�cllimits.lib.utilsrr
�objectrrrr�<module>r�s���'�&�&�&�&�&�!�!�!�!�!�!�������������%�%�%�%�%�%�3�3�3�3�3�3�������G�G�G�G�G�G�	�	�	�	�	�(�	�	�	�]/�]/�]/�]/�]/�V�]/�]/�]/�]/�]/r