Edit File: lvectl_lib.cpython-37.pyc
B 5/9e D � @ s� d dl mZ d dlmZ d dlZd dlZd dlZd dlmZ d dl m Z d dl m Z d dl mZ G dd � d e �ZG d d � d e�ZdS ) � )�absolute_import)�unicodeN)� exec_utility)�FormattedException)�cpapi)�(_convert_memory_value_to_adaptive_formatc @ s e Zd ZdS )�LvectlExceptionN)�__name__� __module__� __qualname__� r r �py/cllimits/lvectl_lib.pyr s r c @ s� e Zd ZdZdZdZdZdZdZdZ dZ dd� Z e dd � �Z d/d d �Zd0d d�Zd1dd�Zdd� Zdd� Zd2dd�Zdd� Zd3dd�Zdd � Zd d � Zd4d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+d,� Z d5d-d.�Z dS )6�LveCtlz* Library for work with LVE limits i@B i ʚ;Nz/usr/sbin/lvectlc C s, t | jdg�\}}t|�| _d| _d| _dS )z5 Object constructor. Get LVE version z lve-versionN)r � _UTILITY_PATH�int� lve_version� _package_data�_resellers_data)�self�_Z s_lve_versionr r r �__init__& s zLveCtl.__init__c C s" t �� } t| �� �}|�d� |S )zH Get panel users uid list :return: List of uids r )r Z list_users�list�keys�append)Z user_packagesZuid_listr r r �get_panel_users_uid_list/ s z LveCtl.get_panel_users_uid_listFc C sL | j ||d� || jkr2t�| j| �}|d= |S t�| 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_numbersZ user_limitsr r r �get_limits_by_user_id: s z LveCtl.get_limits_by_user_idc C s" | � ||� t�| 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! Zreseller_limitsr r r �get_limits_by_reseller_nameO s z"LveCtl.get_limits_by_reseller_namec C s� t j�� sdS d|ddg}|s(|�d� t| j|�\}}t�|�}|dkrVt|d ��|� dg �}|r�t d t |d d d �|d d t |d d d �|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] Nz list-resellerz--jsonz --with-namez--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 r r �json�loadsr r$ �dictr ) r r% r! � lvectl_args�ret_code�std_out� loaded_jsonZ default_infoZdefault_limitsr r r �#get_default_limits_by_reseller_name] s. z*LveCtl.get_default_limits_by_reseller_namec C sz | j dkrFddlm} yt||�d �}W qv tk rB d}Y qvX n0|| j krdt| j | d �}nt| 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 Zclcommon.cpapirD �str� IndexError)r r rD Z package_namer r r �get_package_name_by_user_id� s z"LveCtl.get_package_name_by_user_idc C s d|||rd| ndd�d�S )Nz+%(what)s set error for uid=%(uid)s%(error)sz [%s]rF )ZwhatZuid�error)�message�contextr )r Zparam�lve_id�errr r r Z __set_error� s zLveCtl.__set_errorc C s |t |�g}d|ks d|kr2|�d|�d��|d< x�dD ]x}|�|�}|dkrPq8t |��� }|dkrr| jdkrrq8n|dkr�| jd kr�q8|d kr�|�� r�d | }|�d ||f � q8W t|�d kr�dS |�d�r�|�d� |r�|�d| � t| j|dd�\}} } |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 Zmepr3 � maxEntryProcs)�speedr8 r7 rO r4 r6 r5 N)r7 r6 r5 � )r5 � rP z%s%%z--%s=%s� r zsave-all-parametersz--save-all-parametersz --reseller=%sT)�stderr�Limits) rG r$ �stripr �isdigitr �lenr r r �_LveCtl__set_error) r ZcommandZ container_id�limitsr% r? �k�vr@ �outrN r r r �_set_container_limits_by_id� s0 z"LveCtl._set_container_limits_by_idc C s< d|g}t | j|dd�\}}}|dkr8t| �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 z set-resellerT)rT r rU N)r r r rY )r Z reseller_idr? r@ r] rN r r r �#_set_container_limits_by_id_or_name� s z*LveCtl._set_container_limits_by_id_or_namec C s, t |�r| �|�sdS | jd|||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 rE �set)r% r )�boolrI r^ )r rM rZ r% r r r �set_lve_limits_by_user_id� s z LveCtl.set_lve_limits_by_user_idc C s | � d||� dS )z� Wrapper for _set_container_limits_by_id, set limits for lvp_id; :type lvp_id: int :type limits: dict :return: int z set-resellerr )r^ )r �lvp_idrZ r r r � set_lvp_limits_by_reseller_id� s z$LveCtl.set_lvp_limits_by_reseller_idc C s | � d||� dS )z� Wrapper for _set_container_limits_by_id, set limits for lvp_id; :type lvp_id: int :type limits: dict :return: int zset-reseller-defaultr )r^ )r rc rZ r r r � set_lvp_defaults_by_reseller_id� s z&LveCtl.set_lvp_defaults_by_reseller_idc C sR dd| dg}|dk r$|� d|g� t| j|�\}}|dkrNt| �d||���dS )z� Set unlimited LVE for lve_id :param: lve_id `int`: LVE id :param reseller_name: Reseller name :return: 0 r` z%sz --unlimitedNz --resellerr � Unlimited)�extendr r r rY )r rM r% �argsr@ rN r r r �set_lve_unlimited� s zLveCtl.set_lve_unlimitedc C s< dd| dg}t | j|�\}}|dkr8t| �d||���dS )z� Set unlimited LVP for reseller; Accepts name or id; :type lvp_id: str | int :return: 0 z set-resellerz%sz --unlimitedr rf )r r r rY )r rc rh r@ rN r r r �set_lvp_unlimited s zLveCtl.set_lvp_unlimitedc C sF dd| dd� |� g}t| j|�\}}|dkrBt| �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` z%sz --default=%s�,r �Default)�joinr r r rY )r rM rZ rh r@ rN r r r �reset_lve_limits s zLveCtl.reset_lve_limitsc C sF dd| dd� |� g}t| j|�\}}|dkrBt| �d||���dS )zk Reset LVP limits for reseller_name. :param: reseller_name str: :return: 0 z set-resellerz%sz --default=%srk r rl )rm r r r rY )r r% rZ rh r@ rN r r r �reset_reseller_limits s z LveCtl.reset_reseller_limitsc C s t | jddg�\}}|S )zO Apply all already configured limits :return: ret code Zapplyr+ )r r )r r@ rN r r r �apply_all_limits& s zLveCtl.apply_all_limitsc C s< dd| dg}t | j|�\}}|dkr8t| �d||���dS )z� Disable reseller limits for given name; Equivalent to lvectl remove-reseller <str> :type reseller_name: str :return: 0 zremove-resellerz%sz--jsonr zDisable reseller limits)r r r rY )r r% rh r@ rN r r r �disable_reseller_limits. s z LveCtl.disable_reseller_limitsc C s t � | _tj�� sdS dddg}|s.|�d� t| j|�\}}t� |�}|dkr\t |d ��x2t � � D ]&}|rx||krxqft |�i d�| j|< qfW x�|�d g �D ]z}|d �d �\}} |r�|| kr�q�t || t t |d d �|d t |d d �|d |d t|d |�d�d�} | | j| d < q�W dS )z� Load information about resellers; :type human_readable_numbers: bool :type reseller_name: str | None :return: Nz list-resellerz--jsonz --with-namez--bytesr r' )�namerZ r( �ID�:r* )r+ r, r- r. r/ r0 )r2 r3 r4 r5 r6 r7 )�idrr rZ rr )r> r r9 r: r; r r r r< r= r r Z resellersr r$ �splitr ) r r! r% r? r@ rA rB �resellerZid_rr Z reseller_infor r r r# ; s< zLveCtl._load_resellers_infoc C s | j dkr| �||� | jdk r$dS t� | _|dk rBdt|�g}n |dk rXdt|�g}ndg}|�d� |sv|�d� t| j|�\}}t� |�}|dkr�t |d ��|d }x�|D ]�} t� } | d } | d kr�d } | | d < d | d i| d< d | d i| d< t | d |�| d< | d | d< | j dk�r@t | d |�| d< | d | d< | j dk�rX| d | d < | | jt | d �< q�W |dk �r�| �||�} | �r�| | jd< dS d| jk�rt� } d | d < d d i| d< d d i| d< d | d< d | d< | j dk�r�d | d< d | d< | j dk�r d | d < | | jd< 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 NZpaneluserlimitsZpaneluserslimitsz--jsonz--bytesr r' r( r Z VE_DEFAULTr) r+ r* r2 r- r4 r1 r8 r, r3 rR r0 r7 r/ r6 � r. r5 rs �0Z0K)r r# r r> rG r r r r< r= r r r r rC ) r r! r rw r? r@ rA rB Z json_dataZpkg_dataZ pkg_limitsZpkg_nameZreseller_defaultsr r r r l sl zLveCtl._load_info)F)F)F)N)N)N)NN) r r r �__doc__ZMEGAHERZZGIGAHERZr r Z_packages_limitsr r r � staticmethodr r" r& rC rI rY r^ r_ rb rd re ri rj rn ro rp rq r# r r r r r r s8 # ( 1r )Z __future__r Z past.builtinsr r9 r< r Z cllimits.libr Zclcommon.clexceptionr Zclcommonr Zcllimits.lib.utilsr r �objectr r r r r �<module> s