Current File : //opt/alt/python37/lib/python3.7/site-packages/cllimits/lib/__pycache__/limits.cpython-37.pyc
B
�2Lb�� � @ 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! 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.m/Z/ d dl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 d dl9m: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"�
Z>d#d$d%gZ?d&d'gZ@G d(d)� d)eA�ZBdS )*� )�print_function)�absolute_import)�division)�
basestringN)�ClPwd)�
MySQLGovernor�MySQLGovException)� iteritems�
itervalues)�CloudlinuxLicenseLib)� defaultdict)�convert_to_kernel_format�is_active_reseller_limits�lve� ) � parse_cloudlinux_limits_opts�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)�FormattedException)�
EncodingErrorc @ s e Zd ZdS )�DomainExceptionN)�__name__�
__module__� __qualname__� r2 r2 �D/opt/alt/python37/lib/python3.7/site-packages/cllimits/lib/limits.pyr. % s r. c C s4 | d � dttt|�� r"d�|�n|��d �i�S )N�cpu�allz{}%�d )�update�str�intr
�isdigit�format)�limits�valuer2 r2 r3 �<lambda>* s r> c C s | d � d|i�S )N�ior5 )r7 )r< r= r2 r2 r3 r> / � c C s | d � d|i�S )Nr4 �mysql)r7 )r< r= r2 r2 r3 r> 0 r@ c C s | d � dtt|�d �i�S )Nr? rA i )r7 r8 r9 )r< r= r2 r2 r3 r> 2 r@ c C s0 | d � |�d�d �� |�d�d �� d��S )N�inodes�,r r )�soft�hard)r7 �split�strip)r< r= r2 r2 r3 r> 3 s c C s | � d|i�S )N�ep)r7 )r< r= r2 r2 r3 r> 7 r@ c C s | � d|i�S )N�nproc)r7 )r< r= r2 r2 r3 r> 8 r@ c C s | � dt|�i�S )N�pmem)r7 r )r< r= r2 r2 r3 r> 9 s c C s | � dt|�i�S )N�vmem)r7 r )r< r= r2 r2 r3 r> ; s c C s | � d|i�S )N�iops)r7 )r< r= r2 r2 r3 r> = r@ )
�speedr? z mysql-cpuzmysql-iorB �
maxEntryProcsrI rJ rK rL �rootZadmin�-ZabusersZsinglec @ s e Zd ZdZdZdZdZdZdZdZ dd� Z
dAdd�Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zed�d d �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z!d4d5� Z"d6d7� Z#d8d9� Z$d:d;� Z%d<d=� Z&dBd?d@�Z'dS )C�CloudlinuxLimitsz2
Class for run cloudlinux-limits commands
NFc C s2 t � | _t� | _t� | _t� | _t� | _ i | _
dS )z#
Empty constructor
N) r �_clpwdr �_lvectlr! �_cagefsr � _cl_quotar �_mysql_governor� _users_list)�selfr2 r2 r3 �__init__T 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�dictr7 �timer �_is_json)rX �dataZ
force_jsonr[ r2 r2 r3 �
print_datab 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
Tr[ 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_statusr` �_error_and_exit� _parse_args�_optsrd r Zis_panel_supportedr; r' Zreseller_limit_supported�_verify_reseller�_check_user_owner�_set_run�_get_run�_disable_reseller_limits�_enable_reseller_limits)rX �argvZlicencer2 r2 r3 �runm s0
zCloudlinuxLimits.runc C s< t || j�\}}|s"| �t|�� t|d t�r8d|d< |S )z%
Parse CLI arguments
z--lve-idr )r r` re r r] r )rX rn Zstatusra r2 r2 r3 rf � 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)r[ �context)rg rd � _check_user�_get_reseller_namere )rX � for_reseller�_rp r2 r2 r3 ri � 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)rg �_get_check_user�_get_action_list)rX r2 r2 r3 rk � 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)r[ rr N) rg rS Zdisable_reseller_limitsrb r# re r^ �messagerr )rX �res_id�er2 r2 r3 rl � 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)r[ rr )
rg rS Z#_set_container_limits_by_id_or_namer8 rb r# re r^ ry rr )rX rz r{ r2 r2 r3 rm � s
z(CloudlinuxLimits._enable_reseller_limitsc C sV | � � \}}|dk r$|dk r$||fS | j}|d �ry| �|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 | j �
|�s�t|�r�| �dd|id�� || j|< �nL|d }|�r�t|�}xf|D ]^}y| j� |�}W n< t
j k
�rn } z| �t|j|jd�� W dd}~X Y nX || j|< �q W d| jd< n�y| j �� | _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
�r8 } 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)r[ rr z %(username)s is not a panel userrp z--reseller-namerO r � )rs rg �get_user_name_by_domainr. re r^ ry rr rR �get_uidr �NoSuchUserExceptionrS �get_package_name_by_user_id�boolrW r% Zget_panel_users_uid_list�_panel_uids_listr- � get_names)rX �uidrp �or{ �
reseller_nameZresellers_users�namer2 r2 r3 rw � sP
((
( ( (z CloudlinuxLimits._get_check_userc 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}
rP )�idrp rq r �domainrp z--for-resellerNrq )rg rd �_get_user_main_domainrt �_fix_single_uid_users)rX r� rp �userru Zuser_main_domainr2 r2 r3 � _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_domainr. )rX rp r2 r2 r3 r� 8 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/ArP r� rq rp )rR r� �lenr� rt )rX r� r� Z usernamesrp Z main_domainrq r2 r2 r3 r� B 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 �]z\}}| �||�} |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 | j �|�| 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�rrt
� |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 |�r y�|d �r| j�!| d �| d< |d �r0| j�"| d �| d#< |d! �sD|d" �r�| j�#| d �\}}|d! �r�d$| k�rvt
� | 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' � |�%|� �qbW ||d*< ng |d*< | �)|d+� d,S )-z3
Process output for selected users
�enabledZcageFSZdisabledz--for-resellerN�quotaZmySqlGovz--limitsrC c 3 s | ]}|� p|� kfV qd S )Nr2 )�.0�k)�lkr2 r3 � <genexpr>t s z4CloudlinuxLimits._get_action_list.<locals>.<genexpr>z--lve-idz
--usernamez--domainz--human-readable-numbersz--reseller-name)rq )r[ rr �error�errorsFZcagefsrp �packagerB z'LVE Manager/quota package not installedz#Quotas not activated on this system)rD rE Zwarningrr z mysql-govzmysql-restrictz mysql-cpuzmysql-ioZmySqlRestrictedr4 rA r? r< �items�users� resellersTr )*rg rT Zis_cagefs_presentrU Zis_clquota_presentrV �is_governor_presentr] r8 rF r^ r rS Z
_load_infor# re ry rr �can_get_governor_limitsr r rW r� Zget_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_usernameZget_limits_by_user�_filter_limits�appendr� Z_resellers_datar
rb )rX r[ r� �keysr� r{ Zis_governor_availabler� rp r� Z
cagefs_statusr< Z
soft_limitZ
hard_limitZ cpu_limitZio_limitZresellers_datarq r2 )r� r3 rx W 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)ry rr z %(util)s not present in systemr� )rV r� Zget_governor_mode�MYSQL_GOV_ENABLED_MODEr )rX Z exc_messager2 r2 r3 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]
)rJ rK rI rL FrN rH rM r4 r5 r? N)rd �pop)rX r� r< r� r2 r2 r3 r� s
$
$z CloudlinuxLimits._filter_limitsc C s | � � \}}| �� | �||�S )z#
Run `set` command
)�_set_check_userrh � _set_action)rX r� rp r2 r2 r3 rj 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 r2 )rg )r� �x)rX r2 r3 �
<listcomp>* s z4CloudlinuxLimits._set_check_user.<locals>.<listcomp>)z--mysql-unrestrict-allz--reseller-nameNr[ z1You must set lve-id or username for modify limitsc s g | ]}� j | �qS r2 )rg )r� r� )rX r2 r3 r� 2 s c s g | ]}� j d �|� �qS )z--{})rg r; )r� r� )rX r2 r3 r� 3 s z--reseller-namez--for-resellerz.Reseller limits for %(reseller)s are disabled.rq )r[ rr zOReseller limits for %s are enabled. Admin can`t set limits for resellers users.) rs r� re r r rg r* r rt )rX r� rp Zopts_without_userr� r2 )rX r3 r� $ s$
z CloudlinuxLimits._set_check_userc C sX t dg�}t|B t|B t|B t g �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"
r5 )r �lvpzcl-quotaZdbgovr N)rc r r r r� �list)rX �defaults�tZall_keyr� r2 r2 r3 �_clean_defaultsD 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)srq )r[ rr )rg r& re r^ )rX r� rq r2 r2 r3 rh V 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 )Nr2 )r� �coder2 r2 r3 r� � s z/CloudlinuxLimits._set_action.<locals>.<genexpr>�* )rg �_set_reseller_action�_set_user_actionr� �nextrS Zapply_all_limits�_make_set_action_response) rX r� rp r� � action_with_reseller_container� set_default_resellers_limits�is_lvectl_error�ret_listr[ r2 r2 r3 r� e s(
z CloudlinuxLimits._set_actionc s� | j std� dS |�t�� dd�� |d s6|d rV|sVd | j_d | j_| �� �ndd}d} t t
��xZt |�D ]N\}
} |
dd � }
| rp|
t krpd} | r�|
t
kr�d}t| �} t|
} | �| � qpW | �r�|�rTd | j_|r�| j�|d �� nF|�r&| j�|d
�}
|
d }|d k�r |
d }|
d
� n | j�|�� dd� �� ��fdd�t ��D ��| �|�}|�r�||�d�g}|�d|i� n,||dk�r�dn||�d�g}|�d|i� | �|d� dS )NrZ r )r\ r[ z --defaultz --unlimitedF� Tz--for-resellerz--reseller-namer� r� r< c S s t | t�r| �|� n|} | S )N)r] r^ r7 )� old_valueZ new_valuer2 r2 r3 � update_func� s
z?CloudlinuxLimits._make_set_action_response.<locals>.update_funcc s4 i | ],\}}|� kr&��| � | �n�| |�qS r2 r2 )r� �keyr= )�
getted_limits�
setted_limitsr� r2 r3 �
<dictcomp>� s z>CloudlinuxLimits._make_set_action_response.<locals>.<dictcomp>)r� r� r< r� rO rP )r� rp rq r< r� )r` �printr7 r_ rU Z_inode_user_limitsrS r� rx r r^ r r r r8 �
OUT_LIMITSZ#get_default_limits_by_reseller_nameZget_limits_by_reseller_namer� rt rb )rX r� r� r� r[ r� r� rp Z
is_lve_limitsZ is_limitsr� r= Z
value_funcZgetted_reseller_datar� r� r� r2 )r� r� r� r3 r� � sj
z*CloudlinuxLimits._make_set_action_responsec
st 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�ry.| ��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< d} |�| � �d �r�y8�d dk�rBd }nd}| j�||� d} |�| � W n: tk
�r� }
z| �t |
j |
j
d
�� W d d }
~
X Y nX t�fdd�tD ���r�y�| �� �r��d �r4�d dk}y | j�||�} |�| � W n: tk
�r2 }
z| �t |
j |
j
d
�� W d d }
~
X Y nX �d �r��d dk}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 ���r��d �d }}|d k�rN| j� d!��sNd"}d"�d < y,| j� |d#k�rdd$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% �rjy0| j j |�d% ��dd�d&� d} |�| � W n: tk
�rh }
z| �t |
j |
j
d
�� W d d }
~
X Y nX |||fS )'NFz--for-resellerc 3 s | ]}� d | V qdS )z--%sNr2 )r� �s)r� r2 r3 r� � s z4CloudlinuxLimits._set_user_action.<locals>.<genexpr>rN �mepc 3 s6 | ].}�d | dk r� � ||��d | fV qdS )z--%sN)rd )r� r� )�
limits_mapr� r2 r3 r� � s z--save-all-parameterszsave-all-parameters)r[ rr z --unlimitedTr� �lvectlr� r� r z --defaultr zcl-quotaz--cagefsr� c 3 s | ]}� d | V qdS )z--%sNr2 )r� r� )r� r2 r3 r� L s z --mysql-govZwatchedz--mysql-restrictZ
restrictedz--mysql-unrestrict-allc 3 s | ]}� d | V qdS )z--%sNr2 )r� r� )r� r2 r3 r� g s z --mysql-cpuz
--mysql-ior z1.2-17r� rO �defaultz--inodes)�force) r^ r� r rd rS Zset_lve_limits_by_user_idr� r# re ry rr Zset_lve_unlimitedrU Z set_user_inodes_limits_unlimitedr r� Zreset_lve_limitsZreset_user_inodes_limitsrT Zset_user_statusr" r r� rV Z set_governor_status_for_userr Z set_restricted_status_for_userZ%set_unrestricted_status_for_all_usersr Z_is_governor_newer_thenZset_limits_for_userZset_user_inodes_limits)rX r� r� rp r� r� r[ r� r< �ret_coder{ �lvectl_error_dictZis_quota_errorZquota_error_dict� lve_defaultsZnew_user_status�str4 r? r2 )r� r� r3 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)rg )r� r� )rX r2 r3 r� � s z8CloudlinuxLimits._set_reseller_action.<locals>.<genexpr>rN r� c 3 s: | ]2}�j d | dk r� �||��j d | fV qdS )z--%sN)rg rd )r� r� )r� rX r2 r3 r� � s z--lve-idz--for-resellerz--reseller-name)r[ rr z --defaultr� Tr� r� r� z --unlimitedr )r^ r� r rg rS Z set_lvp_defaults_by_reseller_idZ set_lvp_limits_by_reseller_idr� r# re ry rr r� Zreset_reseller_limitsZset_lvp_unlimited) rX r� r� r� r[ r� r< r� r{ r� r� r2 )r� rX r3 r� � sV
(
"
"
z%CloudlinuxLimits._set_reseller_actionc C s6 t dd�}tdd� |D ��| _tdd� |D ��| _d S )N)Zcploginrq Zdns)Zkeylsc s s | ]}|d |d fV qdS )r r Nr2 )r� �ir2 r2 r3 r� � s z/CloudlinuxLimits._init_cache.<locals>.<genexpr>c s s | ]}|d |d fV qdS )r r� Nr2 )r� r� r2 r2 r3 r� � s )r$ r^ �_user_to_reseller�_userdomain_cache)rX �infor2 r2 r3 � _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
NrP )r� r� r( r^ rd )rX rp r� r2 r2 r3 rt � 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( r^ rd )rX rp r2 r2 r3 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� )ry rr )r) �IOErrorr( r. r8 )rX Z domain_namerp r{ r2 r2 r3 r} � s z(CloudlinuxLimits.get_user_name_by_domainc
C s
| j }d\}}dd� }|d s*|d dkrt|d }| j�|�}t|�}|sht|�rhdd� }| �||�� || j|< n�|d �r|d }y| j�|�}W n: t j
k
r� } z| �t |j |j
d�� W d d }~X Y nX | j�|�}|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 userrp )r[ rr r2 )rp r2 r2 r3 r> � s z.CloudlinuxLimits._check_user.<locals>.<lambda>z--lve-idr c S s dd| id�S )Nz%(uid)s is not a panel userr� )r[ rr r2 )r� r2 r2 r3 r> s z
--username)r[ rr N)rg rS r� r+ r� re rW rR r~ r r r^ ry rr )rX r� r� rp Z_username_messageZ user_packageZ _uid_messager{ r2 r2 r3 rs � 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
r\ Tr[ rr zError: %s [%s]N)
r` r7 r_ r r� r8 rd �KeyError�sys�exit)rX ry Z
error_coder{ r2 r2 r3 re s &z CloudlinuxLimits._error_and_exit)FrZ )r )(r/ r0 r1 �__doc__rg r` r� rW r� r� rY rb ro rf ri rk rl rm rw r� r� r^ r� rx r� r� rj r� r� rh r� r� r� r� r� rt r� r} rs re r2 r2 r2 r3 rQ D sL
& :
(X "9
+rQ )CZ
__future__r r r Z
past.builtinsr r� r_ 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 Zcllimits.clquota_libr r r Zutilsr Z cli_utilsr r Zcllimitsr r! r" r# Zclcommon.cpapir$ r% r& r' r( r) r* r+ Zclcommon.clexceptionr, Z clcommon.cpapi.cpapiexceptionsr- r. r� ZADMIN_RESELLER_NAMESr� �objectrQ r2 r2 r2 r3 �<module> sB , (
Mini Shell