Current File : //opt/alt/python37/lib/python3.7/site-packages/clpackages/__pycache__/packages.cpython-37.pyc
B
�2Lbv% � @ 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 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_adminc @ 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 s8 d| _ t� | _t� | _t� | _dddddddd g| _d S )
NF�speed�nproc�pmem�vmem�
maxEntryProcs�io�iops�inodes)�_is_json�dict�_optsr
�_lvectl_pack_libr �_clquota_pack_lib� _LIMITS_KEYS)�self� r# �D/opt/alt/python37/lib/python3.7/site-packages/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" r- Zstatus�datar# r# r$ r) > 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 x�|D ]�} | d } |d �s"| d= |d �s2| d= |d �sLd| k�rL| d= |d �s\| d= |d �svd| k�rv| d= |d �s�| d = |d �sd | k�r| d = �qW t|�|d < ||d< | �|� d S ) z#
Run 'get' command
z--limits�,Nc 3 s | ]}|� p|� kfV qd S )Nr# )�.0�k)�lkr# r$ � <genexpr>O 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 systemZwarningr5 �package�default�limitsr r r r Zepr r r �itemsr )r �
isinstance�str�splitr r! �encode�decoder Zget_reseller_package_limitsr r( �messager5 r Zis_clquota_presentZis_clquota_activatedr r
r �len� _print_data) r" �oZ limits_keysr& �
reseller_name� package_nameZ
packages_data�eZ quotas_dictZ package_dataZpackage_name_quotaZ limits_dictr# )r3 r$ r, H sx
(
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 | �|� dS )z#
Run 'set' command
z
--providerz--for-resellerNz5error: reseller limits for %(reseller)s are disabled.Zreseller)r& r5 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# )r1 �xr# r# r$ �
<listcomp>� s z/CloudlinuxPackages._set_run.<locals>.<listcomp>z--inodesr )r r* r r r( r? r@ r �any�%_check_hoster_permissions_for_packager r Zset_reseller_package_limitsr rA r5 r r
rC )r" rE � for_resellerrF r& rD r: rG r# r# r$ r+ � s8
(
(
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)sr8 )r& r5 )r r( )r" rL rF rE rA r# r# r$ rK � 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" r/ Z
force_jsonr& r# r# r$ rC � 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
rN Tr& r5 zError: %s [%s]N)
r rO rP r �printr= r* �KeyError�sys�exit)r" rA Z
error_coderG r# r# r$ r( � s &z"CloudlinuxPackages._error_and_exitN)FrM )r ) �__name__�
__module__� __qualname__�__doc__r% r. r) r, r+ rK rC r( r# r# r# r$ r s
R*
r ) Z
__future__r r rS rP 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 �objectr r# r# r# r$ �<module>
s
Mini Shell