Edit File: packages.cpython-37.pyc
B 5/9eF2 � @ s� d dl mZ d dl mZ d dlZd dlZd dlmZ d dlmZ d dl m Z m Z d dl m Z mZmZ dd lmZ d d lmZ d d lmZ d d lmZmZmZ d d lmZ d dl m Z m Z G dd� de �Z dS )� )�print_function)�absolute_importN)� iteritems)�CloudlinuxLicenseLib)�print_dictionary�replace_params)� ClQuotaLib�QuotaDisabledException�ClQuotaException� )� parse_cloudlinux_packages_opts)� LveCtlPackage)�LvectlException)�DEFAULT_PACKAGE�is_active_reseller_limits�lve)�is_admin)� MySQLGovernor�MySQLGovExceptionc @ sT e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z ddd�Z ddd�Z dS )�CloudlinuxPackagesz4 Class for run cloudlinux-packages commands c C sD d| _ t� | _t� | _t� | _dddddddd d d g | _t� | _ d S ) NF�speed�nproc�pmem�vmem� maxEntryProcs�io�iops�inodesz mysql-cpuzmysql-io) �_is_json�dict�_optsr �_lvectl_pack_libr �_clquota_pack_lib� _LIMITS_KEYSr �_mysql_governor)�self� r& �py/clpackages/packages.py�__init__# s zCloudlinuxPackages.__init__c C sx t � }|�� s"d| _| �ddi�S d|k| _| �|�| _| j�d�rZt�� sZ| �ddi� | jd rl| � � S | � � S dS ) z$ Run command action T�resultz Cloudlinux license isn't validz--jsonz--for-resellerzLerror: reseller limits are not supported, can not use --for-reseller option.�setN) r Zget_license_statusr �_error_and_exit� _parse_argsr �getr Zreseller_limit_supported�_set_run�_get_run)r% �argvZlicencer& r& r' �run, s zCloudlinuxPackages.runc C s& t || j�\}}|s"| �t|�� |S )z% Parse CLI arguments )r r r+ r )r% r0 Zstatus�datar& r& r' r, A s z CloudlinuxPackages._parse_argsc s� | j }t|d t�r"|d �d�nd� t� fdd�| jD ��}t� }| j d }d}| j d dk rt| j d �� �d�}y| j� ||| j d �}W n8 t k r� } z| � t|j |j d �� W dd}~X Y nX |d �r | j�� s�d |d < t� |d< d|d d < �n| j�� �s*d |d < t� |d< d|d d < n�y| j� |�}W n� tk �r� } z*i }t� |d< t|�|d d < d |d < W dd}~X Y n< tk �r� } z i }|j |d< |j |d< W dd}~X Y nX x>|D ]6} | d } | tk�r�d} | |k�r�|| | d d < �q�W g } x4|D ],} d| d d i| d d< | �| d � �q W | j�� �rFd|d< y�| j d dk�sp| j d | k�r�| j�| j d �} x^|D ]V} | d }|tk�r�d}|| k�r�| | d | d d d< | | d | d d d< �q�W W n^ tk �rB } z>|d dk�r2d |d< d|k�r2t� |d< t|�|d d< W dd}~X Y nX nd |d< �xR|D �]H} | d } |d �s�| d= | d d= | d i k�r�| d= |d �s�| d = |d �s�d | k�r�| d = |d �s�| d!= |d" �s�d"| k�r�| d"= |d �s| d d= | d i k�r| d = |d# �s(d#| k�r(| d#= |d$ �s^d| d k�r^| d d= | d i k�r^| d= |d% �sVd | k�rVd| d k�rV| d d= | d i k�rV| d = �qVW t|�|d&< ||d< | �|� d'S )(z# Run 'get' command z--limits�,Nc 3 s | ]}|� p|� kfV qd S )Nr& )�.0�k)�lkr& r' � <genexpr>R s z.CloudlinuxPackages._get_run.<locals>.<genexpr>z --providerz --packagezunicode-escapez--human-readable-numbers)r) �contextr �errorZquota�errorsz'LVE Manager/quota package not installedz#Quotas not activated on this systemZwarningr8 �package�default�limits�allr ZcpuZenabledZmySqlGovZmysqlr Zdisabledr r r Zepr r z mysql-cpuzmysql-io�itemsr )r � isinstance�str�splitr r# �encode�decoder! Zget_reseller_package_limitsr r+ �messager8 r" Zis_clquota_presentZis_clquota_activatedr r r �appendr$ �is_governor_presentZget_package_limitsr �len� _print_data)r% �oZ limits_keysr) � reseller_name� package_nameZ packages_data�eZ quotas_dictZ package_dataZpackage_name_quotaZ package_namesZpackage_limits_governorZ limits_dictr& )r6 r' r/ K s� ( " $ " zCloudlinuxPackages._get_runc C s& | j d }| j �d�}|dk rBt|�sBt|�sB| �dd|id�� | j d �� �d�}t� }| j }t|d |d |d |d |d |d |d g��r:|d |d |d |d |d |d |d d�}| � |||� tdd� t |�D ��}y| j � |||� W n: t k �r8 } z| �t|j|jd�� W dd}~X Y nX |d �r�y| j� ||d � W n: tk �r� } z| �t|j|jd�� W dd}~X Y nX |d dk �s�|d dk �r| j�� �ry | j�||d |d � W n< tfk �r } z| �t|j|jd�� W dd}~X Y nX | �|� dS )z# Run 'set' command z --providerz--for-resellerNz5error: reseller limits for %(reseller)s are disabled.Zreseller)r) r8 z --packagezunicode-escapez--speedz--pmemz--vmemz--maxEntryProcsz--nprocz--ioz--iops)r r r r r r r c S s g | ]}|d dk r|�qS )r Nr& )r4 �xr& r& r' � <listcomp>� s z/CloudlinuxPackages._set_run.<locals>.<listcomp>z--inodesz --mysql-cpuz --mysql-ior )r r- r r r+ rC rD r �any�%_check_hoster_permissions_for_packager r! Zset_reseller_package_limitsr rE r8 r" r r$ rG Zset_package_limitsr rI )r% rK � for_resellerrL r) rJ r= rM r& r& r' r. � sB ( (( ( zCloudlinuxPackages._set_runc C s, |d kr(t |�r(d}| �|d|id�� d S )NzMerror: hoster cannot manage package limits for reseller's package %(package)sr; )r) r8 )r r+ )r% rR rL rK rE r& r& r' rQ � s z8CloudlinuxPackages._check_hoster_permissions_for_packageF�successc C s2 t |t�r |�|t�� d�� t|| jp*|� dS )z� Output data wrapper :param: `dict` data - data for output to stdout :param: `bool` force_json - always output json format )r) � timestampN)r@ r �update�timer r )r% r2 Z force_jsonr) r& r& r' rI s z CloudlinuxPackages._print_datar c C s� | j r$|�dt�� i� t|d� nXy tt|d �|�di � � W n6 tk rz } ztdt|�|f � W dd}~X Y nX t� |� dS )z� Print error and exit :param dict message: Dictionary with keys "result" as string and optional "context" as dict rT Tr) r8 zError: %s [%s]N) r rU rV r �printrA r- �KeyError�sys�exit)r% rE Z error_coderM r& r& r' r+ s &z"CloudlinuxPackages._error_and_exitN)FrS )r ) �__name__� __module__� __qualname__�__doc__r( r1 r, r/ r. rQ rI r+ r& r& r& r' r s 0 r )!Z __future__r r rY rV Z future.utilsr Z cllicenser Z cli_utilsr r Zcllimits.clquota_libr r r Zcl_packages_arg_parser Z clpackagesr Zcllimitsr Z lvectllibr r r Zclcommon.cpapir Z clcommon.libr r �objectr r& r& r& r' �<module> s