Current File : //opt/alt/python37/lib/python3.7/site-packages/cllimits/lib/__pycache__/limits.cpython-37.pyc
B
F�sdD� � @ s� d dl mZ d dl mZ d dl mZ d dlmZ d dlZd dlZd dlm Z d dl
m Z m Z d dl
mZmZ d d lmZ d d
lmZ d d lmZmZmZ d d
lmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z" d dl#m$Z$m%Z%m&Z& d dl'm(Z( d dl)m*Z*m+Z+ d dl,m-Z-m.Z.m/Z/m0Z0 d dl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z; d dl<m=Z= d dl>m?Z? G dd� de=�Z@dd� dd� dd� dd� d d� d d� d d� d d� d d� d!d� d"�
ZAd#d$d%gZBd&d'gZCG d(d)� d)eD�ZEG d*d+� d+eE�ZFdS ),� )�print_function)�absolute_import)�division)�
basestringN)�ClPwd)�
MySQLGovernor�MySQLGovException)� iteritems�
itervalues)�CloudlinuxLicenseLib)� defaultdict)�convert_to_kernel_format�is_active_reseller_limits�lve� )
�CloudlinuxLimitsOptsParser� CloudlinuxLimitsNoLveOptsParser�AVAILABLE_LIMITS�AVAILABLE_LVE_KEYS�AVAILABLE_MYSQL_KEYS�AVAILABLE_QUOTA_KEYS�AVAILABLE_KEYS�AVAILABLE_LVP_KEYS�ADMIN_ONLY_OPTIONS�AVAILABLE_MYSQL_KEYS_ALL)�QuotaDisabledException�ClQuotaException�
ClQuotaLib)�convert_mem_value_to_bytes)�print_dictionary�replace_params)�LveCtl�CageFs�CageFsException�LvectlException)
�cpinfo�reseller_users� is_reseller� getCPName� NotSupported� domain_owner�is_admin�get_main_username_by_uid�Feature�is_panel_feature_supported)�FormattedException)�
EncodingErrorc @ s e Zd ZdS )�DomainExceptionN)�__name__�
__module__� __qualname__� r5 r5 �D/opt/alt/python37/lib/python3.7/site-packages/cllimits/lib/limits.pyr1 % s r1 c C s4 | d � dttt|�� r"d�|�n|��d �i�S )N�cpu�allz{}%�d )�update�str�intr
�isdigit�format)�limits�valuer5 r5 r6 �<lambda>* s rA c C s | d � d|i�S )N�ior8 )r: )r? r@ r5 r5 r6 rA / � c C s | d � d|i�S )Nr7 �mysql)r: )r? r@ r5 r5 r6 rA 0 rC c C s | d � d|i�S )NrB rD )r: )r? r@ r5 r5 r6 rA 1 rC c C s0 | d � |�d�d �� |�d�d �� d��S )N�inodes�,r r )�soft�hard)r: �split�strip)r? r@ r5 r5 r6 rA 2 s c C s | � d|i�S )N�ep)r: )r? r@ r5 r5 r6 rA 6 rC c C s | � d|i�S )N�nproc)r: )r? r@ r5 r5 r6 rA 7 rC c C s | � dt|�i�S )N�pmem)r: r )r? r@ r5 r5 r6 rA 8 s c C s | � dt|�i�S )N�vmem)r: r )r? r@ r5 r5 r6 rA : s c C s | � d|i�S )N�iops)r: )r? r@ r5 r5 r6 rA < rC )
�speedrB z mysql-cpuzmysql-iorE �
maxEntryProcsrL rM rN rO �rootZadmin�-ZabusersZsinglec s6 e Zd ZdZdZdZdZdZdZdZ � fdd�Z
dd� Z dEd d
�Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zed �d d!�Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z d*d+� Z e d,d-� �Z d.d/� Z d0d1� Z!d2d3� Z"d4d5� Z#d6d7� Z$d8d9� Z%d:d;� Z&d<d=� Z'd>d?� Z(d@dA� Z)dFdCdD�Z*� Z+S )G�CloudlinuxLimitsz2
Class for run cloudlinux-limits commands
NFc s& t tj�sttt��t�S t� �| �S )N)r. r- ZLVE�superrT �CloudlinuxLimitsNoLve�__new__)�cls)� __class__r5 r6 rW S s
zCloudlinuxLimits.__new__c C s: t � | _t� | _t� | _t� | _t� | _ t
� | _ i | _ dS )z#
Empty constructor
N)
r �_clpwdr! �_lvectlr" �_cagefsr � _cl_quotar �_mysql_governorr � _opts_parser� _users_list)�selfr5 r5 r6 �__init__X s zCloudlinuxLimits.__init__�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
)�result� timestampN)�
isinstance�dictr: �timer �_is_json)ra �dataZ
force_jsonrd r5 r5 r6 �
print_datag s
zCloudlinuxLimits.print_datac C s� t � }|�� s"d| _| �ddi�S d|k| _| �|�| _| j�d�sP| j�d�rnt�� sn| �dd� t
� �i� | j�d�dk r�t� � s�| �dd i� | � � | �
� | jd
r�| �� S | jd r�| �� S | jd r�| �� S | jd
r�| �� S | �ddi�S dS )z$
Run command action
Trd z Cloudlinux license isn't validz--jsonz--for-resellerz--reseller-namez;--for-reseller/--reseller-name options not supported on {}.NzEReseller limits are not supported. Can not use --for-reseller option.�set�getzdisable-reseller-limitszenable-reseller-limitszUnknown action, exit)r Zget_license_statusri �_error_and_exit� _parse_args�_optsrm r Zis_panel_supportedr> r( Zreseller_limit_supported�_verify_reseller�_check_user_owner�_set_run�_get_run�_disable_reseller_limits�_enable_reseller_limits)ra �argvZlicencer5 r5 r6 �runr s0
zCloudlinuxLimits.runc C s@ | j �|| j�\}}|s&| �t|�� t|d t�r<d|d< |S )z%
Parse CLI arguments
z--lve-idr )r_ �
parse_argsri rn r rf r )ra rw Zstatusrj r5 r5 r6 ro � s z CloudlinuxLimits._parse_argsc C sh | j �d�}|sdS | j �d�dkr(dS | �� \}}|dkr@dS || �|�krd| �d||d�d�� dS ) z�
Checks that specified reseller is an owner of specified user and
terminate program if need
:return:
z--for-resellerNz--lve-idr )NzN/Az2%(username)s is not owned by reseller %(reseller)s)�username�reseller)rd �context)rp rm � _check_user�_get_reseller_namern )ra � for_reseller�_rz r5 r5 r6 rr � s z"CloudlinuxLimits._check_user_ownerc C s4 | j d s$| j d r$| j d | j d< | �� | �� S )z#
Run `get` command
z--reseller-namez--for-reseller)rp �_get_check_user�_get_action_list)ra r5 r5 r6 rt � s zCloudlinuxLimits._get_runc
C sn | j d p| j d }y | j�|� | �i d� W n8 tk
rh } z| �t|j|jd�� W dd}~X Y nX dS )z7
Run `disable-reseller-limits` command
z--reseller-namez
--reseller-idT)rd r| N) rp r[ Zdisable_reseller_limitsrk r$ rn rg �messager| )ra �res_id�er5 r5 r6 ru � s z)CloudlinuxLimits._disable_reseller_limitsc
C sv | j d p| j d pd}y | j�t|�� | �i d� W n8 tk
rp } z| �t|j|j d�� W d d }~X Y nX d S )Nz--reseller-namez
--reseller-idz--allT)rd r| )
rp r[ Z#_set_container_limits_by_id_or_namer; rk r$ rn rg r� r| )ra r� r� r5 r5 r6 rv � s
z(CloudlinuxLimits._enable_reseller_limitsc C sP | � � \}}|dk r$|dk r$||fS | j}|d r�y| �|d �}W n8 tk
r| } z| �t|j|jd�� W dd}~X Y nX y| j� |�}W n: t
j k
r� } z| �t|j|jd�� W dd}~X Y nX | � |�s�t
|�r�| �dd|id�� || j|< �nJ|d }|�r�t|�}xf|D ]^}y| j� |�}W n< t
j k
�rj } z| �t|j|jd�� W dd}~X Y nX || j|< �qW d| jd< n�yt�� | _W n: tk
�r� } z| �t|j|jd�� W dd}~X Y nX d }xl| jD ]b}y| j�|�d }W n< t
j k
�r2 } z| �t|j|jd�� W dd}~X Y nX || j|< �q�W d}||fS )
z|
Determines users according to --lve-id/--username/--domain arguments
Or build all panel users list
Nz--domain)rd r| z %(username)s is not a panel userrz z--reseller-namerR r � )r} rp �get_user_name_by_domainr1 rn rg r� r| rZ �get_uidr �NoSuchUserException�get_package_name_by_user_id�boolr` r& r! Zget_panel_users_uid_list�_panel_uids_listr0 � get_names)ra �uidrz �or� �
reseller_nameZresellers_users�namer5 r5 r6 r� � sP ((
( ( (z CloudlinuxLimits._get_check_userc C s | j �|�S )N)r[ r� )ra r� r5 r5 r6 r� " s z,CloudlinuxLimits.get_package_name_by_user_idc C s| ||dd�}|dkrFd|d< d|d< | j �d�}|dk r<|nd|d< n2| �|d �}|r`||d< | �|�|d< | �|� |S ) a?
Retrives username, domain and reseller
for supplied uid/username, for uid==0 returns '-'
either reseller_name if has --for-reseller
:param uid: uid
:param username: username
:return: dict {"id": uid, "username": username, "domain": domain, "reseller": reseller}
rS )�idrz r{ r �domainrz z--for-resellerNr{ )rp rm �_get_user_main_domainr~ �_fix_single_uid_users)ra r� rz �userr Zuser_main_domainr5 r5 r6 � _get_username_domain_reseller% s
z.CloudlinuxLimits._get_username_domain_resellerc C s$ y
| � |�S tk
r dS X dS )zP
Retrives user's main domain
:param username: User name
N)�get_user_domainr1 )ra rz r5 r5 r6 r� ? s
z&CloudlinuxLimits._get_user_main_domain)r� c C sv |d }| j �|�}t|�dkr$dS xL|D ]D}| �|�}| �|�}|dkr*|dkr*||d< ||d< ||d< dS q*W dS ) a
Check and fix user info if some users with same uid present. See LU-1816 for details
:param user: User info. Example:
{'id': 10001, 'username': 'cltest2', 'reseller': 'admin', 'domain': 'cltest2.com'}
:return: None
r� r NzN/ArS r� r{ rz )rZ r� �lenr� r~ )ra r� r� Z usernamesrz Z main_domainr{ r5 r5 r6 r� I s
z&CloudlinuxLimits._fix_single_uid_usersc s� i }| j }| j�� r d|d< nd|d< |d dkrN| j�� rFd|d< nd|d< | j�� rbd|d< nd|d< t|d t�r�|d � d �nd� t
� fd
d �t D ��}g }|d dk�r&|d
dk�r&|d dk�r&y | j j
|d |d d� W n: tk
�r$ } z| �t
|j|jd�� W dd}~X Y nX y | �� }W nb tk
�r� } zB|d dk�r�d|d< d|k�r�t
� |d< t|�|d d< d}W dd}~X Y nX �x�t| j�D �]|\}}| �||�} |d �rH| j�� �rHy&| j�| d �}
|
�r�dnd| d< W nP tk
�rF } z0d|d< d|k�r&t
� |d< t|�|d d< W dd}~X Y nX | �|�| d< |d dk �s�|d
dk �s�|d dk �r�d| j _| j �||d �} |d dk�r�|d �r�| j�� �s�d|d< d|k�r�t
� |d< d|d d< n�| j� � �s d|d< d|k�rt
� |d< d|d d< n�y"| j� |�\} }
| |
d �| d< W n� t k
�r� } z0d|d< d|k�rpt
� |d< t|�|d d< W dd}~X Y n8 t k
�r� } z|j|d < |j|d < W dd}~X Y nX t |d |d |d! |d" g��r|�ry�|d �r| j�!| d �| d< |d �r.| j�"| d �| d#< |d! �sB|d" �r�| jj#| d d$d%�\}}|d! �r�d&| k�rxt
� | d&< || d& d'< |d" �r�d(| k�r�t
� | d(< || d( d'< W nP tk
�r } z0d|d< d|k�r�t
� |d< t|�|d d< W dd}~X Y nX | �$|| � | | d)< |�%| � �q�W t&|�|d*< ||d+< |d �s�| j j'dk �r�g }x0t(| j j'�D ] }| �$||d) � |�%|� �qdW ||d,< ng |d,< | �)|d$� d-S ).z3
Process output for selected users
�enabled�cageFS�disabledz--for-resellerN�quota�mySqlGovz--limitsrF c 3 s | ]}|� p|� kfV qd S )Nr5 )�.0�k)�lkr5 r6 � <genexpr>{ s z4CloudlinuxLimits._get_action_list.<locals>.<genexpr>z--lve-idz
--usernamez--domainz--human-readable-numbersz--reseller-name)r{ )rd r| �error�errorsF�cagefsrz �packagerE z'LVE Manager/quota package not installedz#Quotas not activated on this system)rG rH Zwarningr| z mysql-govzmysql-restrictz mysql-cpuzmysql-ioZmySqlRestrictedT)�with_package_markr7 rD rB r? �items�users� resellersr )*rp r\ �is_cagefs_presentr] Zis_clquota_presentr^ �is_governor_presentrf r; rI rg r r[ Z
_load_infor$ rn r� r| �can_get_governor_limitsr r r` r� �get_user_statusr# r� �
_package_data�get_limits_by_user_idZis_clquota_activatedZget_inodes_limits_by_uidr r �anyZ get_governor_status_by_usernameZ get_restrict_status_by_username�get_limits_by_user�_filter_limits�appendr� Z_resellers_datar
rk )ra rd r� �keysr� r� Zis_governor_availabler� rz r� �
cagefs_statusr? Z
soft_limitZ
hard_limitZ cpu_limitZio_limitZresellers_datar{ r5 )r� r6 r� ^ s�
* (
"*
"
"
z!CloudlinuxLimits._get_action_listc C sP | j �� r6| j �� tkr dS dddd�d�}t|��ndddid�}t|��d S ) NTz %(util)s mode is not %(mode)sZGovernorzabusers or single)�util�mode)r� r| z %(util)s not present in systemr� )r^ r� Zget_governor_mode�MYSQL_GOV_ENABLED_MODEr )ra Z exc_messager5 r5 r6 r� � s
z(CloudlinuxLimits.can_get_governor_limitsc C s� x*dD ]"}|| s|� |d�r|�|� qW |d sJ|� dd�rJ|�d� |d s�|� dd�r�|d � dd�r�|d �d� |d s�|�d� |d s�|� dd�r�|d � dd�r�|d �d� |d s�|�d� d S )
z�
Remove limits that are stored as key: False in `keys` dict;
Used for --limits key;
:type keys: dict[str, bool]
:type limits: dict[str, str | dict]
)rM rN rL rO FrQ rK rP r7 r8 rB N)rm �pop)ra r� r? r� r5 r5 r6 r� s
$
$z CloudlinuxLimits._filter_limitsc C s | � � \}}| �� | �||�S )z#
Run `set` command
)�_set_check_userrq � _set_action)ra r� rz r5 r5 r6 rs s zCloudlinuxLimits._set_runc s� � � � \}}t� fdd�dD ��}|dks2|dkrD|sD� �ddi� t� fdd�tD ��rxt� fdd�tD ��sx||fS � jd s� jd
dk r�t� jd
�s�t� jd
�s� �d d � jd
id
�� n$� �|�}t|�r� �dd| i� ||fS )z&
Find user for modify
c s g | ]}� j | �qS r5 )rp )r� �x)ra r5 r6 �
<listcomp>/ s z4CloudlinuxLimits._set_check_user.<locals>.<listcomp>)z--mysql-unrestrict-allz--reseller-nameNrd z1You must set lve-id or username for modify limitsc s g | ]}� j | �qS r5 )rp )r� r� )ra r5 r6 r� 7 s c s g | ]}� j d �|� �qS )z--{})rp r> )r� r� )ra r5 r6 r� 8 s z--reseller-namez--for-resellerz.Reseller limits for %(reseller)s are disabled.r{ )rd r| zOReseller limits for %s are enabled. Admin can`t set limits for resellers users.) r} r� rn r r rp r+ r r~ )ra r� rz Zopts_without_userr� r5 )ra r6 r� ) s$
z CloudlinuxLimits._set_check_userc C sX t dg�}t|B t|B t|B t|B d�| }tt | �|@ �dkrPtt | �|@ �S dS dS )z�
Return keys to reset that are only acceptable to this type of limits
:param list defaults: List of keys to reset
:param str t: one of "lve", "cl-quota", "dbgov"
r8 )r �lvpzcl-quota�dbgovr N)rl r r r r r� �list)�defaults�tZall_keyr� r5 r5 r6 �_clean_defaultsI s
z CloudlinuxLimits._clean_defaultsc C s@ | j }|d p|d }|s dS t|�s<| �tdd|id�� dS )z�
If any reseller is specified in cli options ensures that it exists
and really a reseller. Will exit with error if not.
z--reseller-namez--for-resellerNz No such reseller: %(reseller)sr{ )rd r| )rp r' rn rg )ra r� r{ r5 r5 r6 rq \ s z!CloudlinuxLimits._verify_resellerc C s� | j }d}d}|d r:|d dkr:| �||�\}}}d}n@|d dk }d|d< |rf| �||�\}}}n| �|||�\}}}|p�dg}t|�r�tdd � |D �d
�S |dkr�| j�� | �|||||||�S ) zl
Process set limits command
:param uid: User uid
:param username: User name
Fz--for-resellerz--lve-idr Tz--reseller-nameNc s s | ]}|r|V qd S )Nr5 )r� �coder5 r5 r6 r� � s z/CloudlinuxLimits._set_action.<locals>.<genexpr>�* )rp �_set_reseller_action�_set_user_actionr� �nextr[ Zapply_all_limits�_make_set_action_response) ra r� rz r� � action_with_reseller_container� set_default_resellers_limits�is_lvectl_error�ret_listrd r5 r5 r6 r� k s(
z CloudlinuxLimits._set_actionc s� | j std� dS |�t�� dd�� |d s6|d rV|sVd | j_d | j_| �� �n�d}d} t t
��|d d k s~|d d k r�| j j |d d
�\|d< |d< xZt
|�D ]N\}
} |
d d � }
| r�|
tkr�d } | r�|
tkr�d }t| �} t|
} | �| � q�W | �r�|�r�d | j_|�r"| j�|d �� nF|�r\| j�|d
�}
|
d }|d k�rR|
d }|
d � n | j�|�� dd� �� ��fdd�t
��D ��| �|�}|�r�||�d�g}|�d|i� n,||dk�r�dn||�d�g}|�d|i� | �|d � dS )Nrc r )re rd z --defaultz --unlimitedFz --mysql-cpuz
--mysql-ioT)r� � z--for-resellerz--reseller-namer� r� r? c S s t | t�r| �|� n|} | S )N)rf rg r: )� old_valueZ new_valuer5 r5 r6 � update_func� s
z?CloudlinuxLimits._make_set_action_response.<locals>.update_funcc s4 i | ],\}}|� kr&��| � | �n�| |�qS r5 r5 )r� �keyr@ )�
getted_limits�
setted_limitsr� r5 r6 �
<dictcomp>� s z>CloudlinuxLimits._make_set_action_response.<locals>.<dictcomp>)r� r� r? r� rR rS )r� rz r{ r? r� )ri �printr: rh r] Z_inode_user_limitsr[ r� r� r rg r^ r� r r r r; �
OUT_LIMITSZ#get_default_limits_by_reseller_nameZget_limits_by_reseller_namer� r~ rk )ra r� r� r� rd r� r� rz Z
is_lve_limitsZ is_limitsr� r@ Z
value_funcZgetted_reseller_datar� r� r� r5 )r� r� r� r6 r� � sj
z*CloudlinuxLimits._make_set_action_responsec
s d}g }t � }�d }t�fdd�tD ��r�ddi� t � �fdd�tD ��}��d�|d < y | j�|||�} |�| � W n8 tk
r� }
z| �t |
j |
j
d
�� W d d }
~
X Y nX �d �r�y| j� ||� d } W n: tfk
�r }
zd }t |
j |
j
d
�} W d d }
~
X Y nX |�r2d
|d< t � |d< | |d d< y"|d k�rJ| j �
|� d} d }
W n: tfk
�r� }
zd } t |
j |
j
d
�}
W d d }
~
X Y nX | �r�d
|d< d|k�r�t � |d< |
|d d< d} |�| � �d d k �r�y4| ��d d�}|d k �r| j�||� d}d } W n: tfk
�rH }
zd }t |
j |
j
d
�} W d d }
~
X Y nX |�rnd
|d< t � |d< | |d d< �d d k�r�y.| ��d d�d k �r�| j �|� d} d }
W n: tfk
�r� }
zd } t |
j |
j
d
�}
W d d }
~
X Y nX | �rd
|d< d|k�rt � |d< |
|d d< y8| ��d d�}| j�� �rJ|d k �rJ| j�||� W nR tfk
�r� }
z0d|k�rvt � |d< t |
j |
j
d
�|d d< W d d }
~
X Y nX d} |�| � �d �r,y8�d dk�r�d }nd}| j�||� d} |�| � W n: tk
�r* }
z| �t |
j |
j
d
�� W d d }
~
X Y nX t�fdd�tD ���r��y�| �� �rB�d �r��d dk}y | j�||�} |�| � W n: tk
�r� }
z| �t |
j |
j
d
�� W d d }
~
X Y nX �d �r.�d d k}y | j� ||�} |�| � W n: tk
�r, }
z| �t |
j |
j
d
�� W d d }
~
X Y nX �d �r�y| j� � } |�| � W n: tk
�r� }
z| �t |
j |
j
d
�� W d d }
~
X Y nX t�fd d�t D ���rB�d �d! }}|d"k�r�| j� d#��s�d$}d$�d!< y,| j� |d%k�r�d&n|||�} |�| � W n: tk
�r@ }
z| �t |
j |
j
d
�� W d d }
~
X Y nX W n: tk
�r~ }
z| �t |
j |
j
d
�� W d d }
~
X Y nX �d' �r�y0| j j!|�d' ��dd�d(� d} |�| � W n: tk
�r� }
z| �t |
j |
j
d
�� W d d }
~
X Y nX |||fS ))NFz--for-resellerc 3 s | ]}� d | V qdS )z--%sNr5 )r� �s)r� r5 r6 r� � s z4CloudlinuxLimits._set_user_action.<locals>.<genexpr>rQ �mepc 3 s6 | ].}�d | dk r� � ||��d | fV qdS )z--%sN)rm )r� r� )�
limits_mapr� r5 r6 r� � s z--save-all-parameterszsave-all-parameters)rd r| z --unlimitedTr� �lvectlr� r� r z --defaultr zcl-quotar� r� z--cagefsr� c 3 s | ]}� d | V qdS )z--%sNr5 )r� r� )r� r5 r6 r� _ s z --mysql-govZwatchedz--mysql-restrictZ
restrictedz--mysql-unrestrict-allc 3 s | ]}� d | V qdS )z--%sNr5 )r� r� )r� r5 r6 r� z s z --mysql-cpuz
--mysql-ior z1.2-17r� rR �defaultz--inodes)�force)"rg r� r rm r[ Zset_lve_limits_by_user_idr� r$ rn r� r| Zset_lve_unlimitedr] Z set_user_inodes_limits_unlimitedr r� Zreset_lve_limitsZreset_user_inodes_limitsr^ r� Z reset_user_limits_to_defaultsr r\ Zset_user_statusr# r r� Z set_governor_status_for_userZ set_restricted_status_for_userZ%set_unrestricted_status_for_all_usersr Z_is_governor_newer_thenZset_limits_for_userZset_user_inodes_limits)ra r� r� rz r� r� rd r� r? �ret_coder� �lvectl_error_dictZis_quota_errorZquota_error_dict� lve_defaultsZ
gov_limitsZnew_user_status�str7 rB r5 )r� r� r6 r� � s
(
"
"
"
"
*
(
(
(
(,(
(z!CloudlinuxLimits._set_user_actionc
s d}t � }g }t�fdd�tD ��r�ddi� t � �fdd�tD ��}yF�jd d k rj�j��jd |�}n�j��jd |�}|�|� W n8 tk
r� } z�� t |j
|j d
�� W d d }~X Y nX |d d k �rtyB�� |d d �} | d k �r
�j�
|d | �}|�|� d}d }
W n: tfk
�rN } zd
}t |j
|j d
�}
W d d }~X Y nX |�rtd|d< t � |d< |
|d d< |d �ry�j�|d � d }
W n: tfk
�r� } zd
}t |j
|j d
�}
W d d }~X Y nX |�r�d|d< t � |d< |
|d d< d}|�|� |||fS )NFc 3 s | ]}� j d | V qdS )z--%sN)rp )r� r� )ra r5 r6 r� � s z8CloudlinuxLimits._set_reseller_action.<locals>.<genexpr>rQ r� c 3 s: | ]2}�j d | dk r� �||��j d | fV qdS )z--%sN)rp rm )r� r� )r� ra r5 r6 r� � s z--lve-idz--for-resellerz--reseller-name)rd r| z --defaultr� Tr� r� r� z --unlimitedr )rg r� r rp r[ Z set_lvp_defaults_by_reseller_idZ set_lvp_limits_by_reseller_idr� r$ rn r� r| r� Zreset_reseller_limitsZset_lvp_unlimited) ra r� r� r� rd r� r? r� r� r� r� r5 )r� ra r6 r� � sV
(
"
"
z%CloudlinuxLimits._set_reseller_actionc C s6 t dd�}tdd� |D ��| _tdd� |D ��| _d S )N)Zcploginr{ Zdns)Zkeylsc s s | ]}|d |d fV qdS )r r Nr5 )r� �ir5 r5 r6 r� � s z/CloudlinuxLimits._init_cache.<locals>.<genexpr>c s s | ]}|d |d fV qdS )r r� Nr5 )r� r� r5 r5 r6 r� � s )r% rg �_user_to_reseller�_userdomain_cache)ra �infor5 r5 r6 � _init_cache� s
z CloudlinuxLimits._init_cachec C sF | j dkr4y | �� W n tk
r2 t� | _ Y nX | j �|d�}|S )zi
Try to get reseller for given username;
:type username: str
:rtype: str
NrS )r� r� r) rg rm )ra rz r� r5 r5 r6 r~ � s
z#CloudlinuxLimits._get_reseller_namec C sB | j dkr4y | �� W n tk
r2 t� | _ Y nX | j �|d�S )z~
Retrive main domain for supplied user
:param username: User name
:return: Domain name or N/A
NzN/A)r� r� r) rg rm )ra rz r5 r5 r6 r� � s
z CloudlinuxLimits.get_user_domainc
C s^ yt |�}|r|S W n2 ttfk
rF } ztt|���W dd}~X Y nX tdd|id���dS )zn
Retrive username by domain
:param domain_name: Domain name
:return: Username
Nz There is no domain '%(domain)s'r� )r� r| )r* �IOErrorr) r1 r; )ra Z domain_namerz r� r5 r5 r6 r� � s z(CloudlinuxLimits.get_user_name_by_domainc
C s | j }d\}}dd� }|d s*|d dkrr|d }| �|�}t|�}|sft|�rfdd� }| �||�� || j|< n�|d r�|d }y| j�|�}W n: tj k
r� } z| �t
|j |j d�� W d d }~X Y nX | �|�}|s�t|�r�| �||�� || j|< ||fS )
a
User identification from CLI options. Common for all commands.
May return (None, None) when user wasn't specified in any way and this
is a *valid* case when we want to list all users with command like:
"... get --json --for-reseller=reseller1"
)NNc S s dd| id�S )Nz %(username)s is not a panel userrz )rd r| r5 )rz r5 r5 r6 rA s z.CloudlinuxLimits._check_user.<locals>.<lambda>z--lve-idr c S s dd| id�S )Nz%(uid)s is not a panel userr� )rd r| r5 )r� r5 r5 r6 rA s z
--username)rd r| N)
rp r� r, r� rn r` rZ r� r r� rg r� r| )ra r� r� rz Z_username_messageZ user_packageZ _uid_messager� r5 r5 r6 r} s,
(
z CloudlinuxLimits._check_userr 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 message dict: Dictionary with keys "result" as string and optional "context" as dict
re Trd r| zError: %s [%s]N)
ri r: rh r r� r; rm �KeyError�sys�exit)ra r� Z
error_coder� r5 r5 r6 rn + s &z CloudlinuxLimits._error_and_exit)Frc )r ),r2 r3 r4 �__doc__rp ri r� r` r� r� rW rb rk rx ro rr rt ru rv r� r� r� r� rg r� r� r� r� rs r� � staticmethodr� rq r� r� r� r� r� r~ r� r� r} rn �
__classcell__r5 r5 )rY r6 rT C sP
% :
(Z ,9
+rT c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )rV c C s" t � | _t� | _t� | _i | _d S )N)r rZ r" r\ r r_ r` )ra r5 r5 r6 rb > s z CloudlinuxLimitsNoLve.__init__c C s> ddl m} yt||�d �}W n tk
r8 d}Y nX |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
r )�reseller_package_by_uidr r� )�clcommon.cpapir� r; �
IndexError)ra Zuser_idr� Z package_namer5 r5 r6 r� D s
z1CloudlinuxLimitsNoLve.get_package_name_by_user_idc
sd i }| j }| j�� r d|d< nd|d< t|d t�rB|d �d�nd� t� fdd�tD ��}g }x�t| j �D ]�\}}| �
||�}|d �r
| j�� �r
y$| j� |d
�}|r�dnd|d< W nN t k
�r } z.d |d< d |kr�t� |d < t| �|d d< W dd} ~ X Y nX | �
|�|d
< i |d< |�|� qlW t|�|d< ||d< |d �sTg |d< | �|d� dS )z3
Process output for selected users
r� r� r� z--limitsrF Nc 3 s | ]}|� p|� kfV qd S )Nr5 )r� r� )r� r5 r6 r� d s z9CloudlinuxLimitsNoLve._get_action_list.<locals>.<genexpr>r� rz r� r� r� r? r� r� z--for-resellerr� Tr )rp r\ r� rf r; rI rg r r r` r� r� r# r� r� r� rk )
ra rd r� r� r� r� rz r� r� r� r5 )r� r6 r� T s8
"
z&CloudlinuxLimitsNoLve._get_action_listN)r2 r3 r4 rb r� r� r5 r5 r5 r6 rV < s rV )GZ
__future__r r r Z
past.builtinsr r� rh Zclcommonr Z clcommon.libr r Z future.utilsr r
Z cllicenser � collectionsr Z lvectllibr
r r Z arg_parsersr r r r r r r r r r Zcllimits.clquota_libr r r Zutilsr Z cli_utilsr r Zcllimitsr! r" r# r$ r� r% r& r' r( r) r* r+ r, r- r. Zclcommon.clexceptionr/ Z clcommon.cpapi.cpapiexceptionsr0 r1 r� ZADMIN_RESELLER_NAMESr� �objectrT rV r5 r5 r5 r6 �<module> sT 0 0
Mini Shell