Edit File: lvectl_packages_lib.cpython-37.opt-1.pyc
B 5/9e� � @ sd d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl m Z d dl m Z G dd � d e �ZdS ) � )�absolute_importN)�NameMap)�lvp_list)� exec_utility)�(_convert_memory_value_to_adaptive_format)�LvectlExceptionc @ s8 e Zd Zdd� Zd dd�Zddd�Zdd � Zd d � Zd S )� LveCtlPackagec C s( d | _ d| _d | _t� | _| j�� d S )Nz/usr/sbin/lvectl)Z _lve_version� _UTILITY_PATH�_packages_data_listr � _name_mapZ link_xml_node)�self� r �$py/clpackages/lvectl_packages_lib.py�__init__ s zLveCtlPackage.__init__Fc st � j dk rdS ddg}|s$|�d� t� j|�\}}t�|�}|dkrRt|d ��|d }t� � _ t� fdd �t � D ��}x�|D ]�}|d } | d kr�d } |d } t � } t |d �| d< t |d |�| d< d|kr�t |d |�| d< t |d �| d< d|k�r t |d �| d< dt |d �i| d< d|k�r:t |d �| d < d d k�rJd } n| |k} | | | | d �} � j �| � q�W dS )!z� Loads all package info from lvectl :param human_readable_numbers: PMEM and VMEM limits in KBytes, MBytes or GBytes :return: None Nzall-package-listz--jsonz--bytesr �msg�datac 3 s | ]}� j �|�V qd S )N)r Zget_name)�.0Zuid)r r r � <genexpr>9 s z+LveCtlPackage._load_info.<locals>.<genexpr>ZRESELLER)zN/A�rootZIDZSPEED�speedZVMEMZvmemZPMEMZpmemZEPZepZNPROCZnproc�allZIO�ioZIOPSZiops�provider)r � F)�packager Zlimits� enabledLimits) r �appendr r �json�loadsr �list�setr �dict�intr )r �human_readable_numbers� lvectl_cmd�ret_codeZlvectl_data_strZ loaded_jsonZ json_dataZactive_resellers� package_data� reseller_name� package_nameZpackage_limitsr Zpackage_result_dictr )r r � _load_info# sP zLveCtlPackage._load_infoc C sb | j |d� |s| jS t� }x@| jD ]6}|d |kr$|rP|d |krZ|�|� q$|�|� q$W |S )a� Get limits for supplied reseller and package :param reseller_name_arg: Reseller name. If None - all resellers :param package_name_arg: Package name. Only if reseller name is provided. If None - all packages :param human_readable_numbers: PMEM and VMEM limits in KBytes, MBytes or GBytes :return: Data list List item format: { "package": "Basic_16", "provider": "Reseller_Peter", "limits": { "vmem": 768, "pmem": 768, "ep": 25, "nproc": 50, "iops": 100, "speed": { "all": 50 }, "io": { "all": 5 }, "inodes": { "soft": 100000, "hard": 200000 } } )r# r r )r) r r r )r Zreseller_name_argZpackage_name_argr# Z out_data_listr& r r r �get_reseller_package_limits^ s z)LveCtlPackage.get_reseller_package_limitsc C s d|||rd| ndd�d�S )Nz3%(what)s set error for package=%(package)s%(error)sz [%s]r )Zwhatr �error)�message�contextr )r Zparamr( �errr r r Z __set_error� s zLveCtlPackage.__set_errorc C s� |sdS d|g}|dkr&|� d| � x>|�� D ]2\}}|dkrP|�� rPd| }|� d||f � q0W t| j|d�\}}} |d ks�| r�t| �d || ���dS ) a� Sets limits for supplied package :param reseller_name: Reseller name :param package_name: Package name :param limits_to_set: new LVE limits. Available keys: speed, vmem, pmem, maxEntryProcs, io, nproc, iops All other keys are ignoring. If some parameter absent on current LVE version (for example pmem on LVE4), it will be ignored too. :return: None Nzpackage-set-extr z --reseller=%sr z%s%%z--%s=%sTr ZLimits)r �items�isdigitr r r �_LveCtlPackage__set_error) r r' r( Z limits_to_setr$ Z limit_nameZ limit_valuer% Z lvectl_stdoutZ lvectl_stderrr r r �set_reseller_package_limits� s z)LveCtlPackage.set_reseller_package_limitsN)F)F)�__name__� __module__� __qualname__r r) r* r1 r2 r r r r r s ; $r )Z __future__r r Zlveapir Z lvectllibr Z cllimits.libr Zcllimits.lib.utilsr Zcllimitsr �objectr r r r r �<module> s