File: //opt/alt/python37/lib/python3.7/site-packages/clpackages/__pycache__/packages.cpython-37.opt-1.pyc
B
P$������ @ 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