File: //opt/alt/python37/lib/python3.7/site-packages/cllimits/__pycache__/clquota_lib.cpython-37.pyc
B
�a}��(�� @ s� d dl mZ d dlZd dlmZmZmZ d dlmZ d dl m
Z
d dl m Z m
Z
d dlmZmZ d dlZd dlZG dd � d e
�ZG d
d � d e�ZG d d
� d
e�ZG dd� d�ZdS )� )�absolute_importN)�Tuple�Dict�Optional)� exec_utility)�FormattedException)� run_command�ExternalProgramFailed)�is_quota_supported�is_quota_activec @ s e Zd ZdS )�ClQuotaExceptionN)�__name__�
__module__� __qualname__� r r �py/cllimits/clquota_lib.pyr s r c s" e Zd ZdZd� fdd� Z� ZS )�UnexpectedClQuotaErrorzU
Raised when we get output from cl-quota
that cannot be parsed properly.
Nc s$ dd|d�d�}t t| ��|� d S )Nz�Unexpected response got from %(util)s. Raw response from cl-quota: '%(raw_response)s'.Please, contact CloudLinux support for help in resolving this issue.�Quota)�util� raw_response)�message�context)�superr �__init__)�selfr Zbroken_exc_message)� __class__r r r s z UnexpectedClQuotaError.__init__)N)r
r r �__doc__r �
__classcell__r r )r r r s r c s e Zd ZdZ� fdd�Z� ZS )�QuotaDisabledExceptionz=
Raised when quotas are disabled in system globally.
c s" dddid�}t t| ��|� d S )Nz%(util)s is disabledr r )r r )r r r )r �disabled_exc_message)r r r r , s z QuotaDisabledException.__init__)r
r r r r r r r )r r r ( s r c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd#d d
�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )$�
ClQuotaLibz/usr/bin/cl-quotaz/usr/sbin/repquotac C s d | _ d | _d | _d | _d S )N)�_inode_user_limits�_inode_package_dict�_is_clquota_present�_is_clquota_activated)r r r r r 6 s zClQuotaLib.__init__c C s | j dkrt| j| j�| _ | j S )zb
Get quota presence flag
:return: True/False - quotas present/not present
N)r# r
�_CL_QUOTA_PATH�_REPQUOTA_PATH)r r r r �is_clquota_present@ s
z ClQuotaLib.is_clquota_presentc C s | j dkrt| j| j�| _ | j S )zg
Get quota activated flag
:return: True/False - quotas activeted/not activeted
N)r$ r r% r& )r r r r �is_clquota_activatedI s
z ClQuotaLib.is_clquota_activatedc C s. | � � t|�}|| jkr$| j| S | jd S )z�
Retrive inodes limits by uid
:param user_id: Supplied uid
:return: cortege (soft_limit, hard_limit). (None, None) if governor not present or error
�0)�_load_users_info�strr! )r Zuser_id�uidr r r �get_inodes_limits_by_uidR s
z#ClQuotaLib.get_inodes_limits_by_uidFc
C sp | � � t|t�r|dk r0dd|id�}t|��|dkrDdgd }n�|dkrXdgd }n�t|�ttgkrjn4t|�d��dkr�|�d�}nd d
|id�}t|��x\|D ]T}y"|dkr�t|�}|dk r�t � �W q� t k
r� d |dd �d�}t|��Y q�X q�W | j
d
| d|d d|d g}|�r.|� d� y t |� W n0 t
k
�rj } ztt|���W dd}~X Y nX dS )z�
Set inodes limits for user uid
:param: int uid: user id
:param: list limits: new inodes limits
:param: bool force: save limits if even they are equal to defaults
:return: None
r z*User id '%(uid)s' isn't a positive integerr, )r r � unlimited� �default�,z$Limits %(limits)s aren't acceptable.�limitszFLimit value '%(limit)s' isn't a positive integer or string %(default)s)�limitr0 z --user-id=%dz--soft-limit=%sz--hard-limit=%s� z--forceN)r* �
isinstance�intr �type�tuple�list�len�split�
ValueErrorr% �appendr r r+ )r r, r2 Zforce� exc_message�l�cmd�er r r �set_user_inodes_limits_ sH
z!ClQuotaLib.set_user_inodes_limitsc C s | � |d� dS )zq
Set unlimited inodes limits for user uid
:param: int uid: user id
:return: None
r. N)rB )r r, r r r � set_user_inodes_limits_unlimited� s z+ClQuotaLib.set_user_inodes_limits_unlimitedc C s | � |d� dS )zo
Set default inodes limits for user uid
:param: int uid: user id
:return: None
r0 N)rB )r r, r r r �reset_user_inodes_limits� s z#ClQuotaLib.reset_user_inodes_limitsc C s� | j dk rdS | �� r | �� s$t� �y| �dg�}W n tk
rP d| _ � Y nX t� | _ x:|�� D ].\}}t|d d �t|d d �f| j |< qdW d| j kr�d| _ ddd| j d �d
�}t|��dS ) zF
Loads users info from cl-quota
:return: None
Nz--jsonr4 r/ r) z'There is no %(what)s found in %(where)szdefault settingsz %s output)�what�where)r r )
r! r' r( r � _get_quotasr �dict�itemsr6 r% )r �datar, r2 r> r r r r* � s"
*
zClQuotaLib._load_users_infoc C s� t | j|�\}}yt�|�}W n" ttfk
r@ t|d��Y n X |sLt� �|d dkrzt|d |�di �|�d�d���|d S )N)r �resultZsuccessr �details)r r rL rI ) r r% �json�loads� TypeErrorr< r r �get)r Zcommand�_Zs_quota_limitsrJ r r r rG � s
zClQuotaLib._get_quotasc C s& | � � r| �� sdS t| jdg� dS )zV
Reset inodes limits for all users to package limits
:return:
Nz--sync)r' r( r r% )r r r r �reset_inodes_limits� s z ClQuotaLib.reset_inodes_limitsc C s� | j dk rdS | �� r | �� s$t� �y| �ddg�}W n tk
rR d| _ � Y nX t� | _ xh|�� D ]\\}}|d d }|dkr�d}nt|�}|d d }|dkr�d}nt|�}||d�| j |< qfW dS ) zI
Loads packages info from cl-quota
:return: None
Nz--jsonz--all-package-limitsr4 �-r r/ )ZsoftZhard) r" r' r( r rG r rH rI r6 )r Zpackages�packager2 Z
soft_limitZ
hard_limitr r r �_load_packages_info� s(
z ClQuotaLib._load_packages_infoc C s. | � � |s| jS || jkr*|| j| iS i S )aY
Get inodes limits for supplied reseller and package
:param package_name_arg: Package name. Only if reseller name is provided. If None - all packages
:return: Packages limits dictionary: {package_name: { "soft": 100000, "hard": 200000 } }
If package with supplied name not found, dictionary will be empty
)rU r" )r Zpackage_name_argr r r �get_reseller_package_limits� s
z&ClQuotaLib.get_reseller_package_limitsc C s d|||rd| ndd�d�S )Nz3%(what)s set error for package=%(package)s%(error)sz [%s]� )rE rT �error)r r r )r Zparam� package_name�errr r r Z __set_error s zClQuotaLib.__set_errorc C s� dddid�}| � � r | �� s&t|��|�d�}t|�dkrNtdddid���d | d
|d d |d
g}t| j|�\}}|d ks�|r�t| �d||���dS )z�
Set inodes limits for package
Called from cloudlinux-packages set
:param package_name: Package name
:param limits_to_set: Limits to set: soft_limit,hard_limit
:return: None
z%(util)s is disabledr r )r r r1 r/ z%(util)s argument errorzcl-quotaz --package=%sz--soft-limit=%sr z--hard-limit=%sr4 z
Inodes limitsN)r' r( r r; r: r r% �_ClQuotaLib__set_error)r rY Z
limits_to_setr r2 � cl_quota_cmd�ret_code�stdoutr r r �set_reseller_package_limits s
z&ClQuotaLib.set_reseller_package_limitsc C s� ddi}| � � r| �� s t|��t|�dkr8tddi��t|�ddhkspt|�ddh }tddd �|�id
���d |� �g}d|kr�|�d |d � �� d|kr�|�d
|d � �� t| j|�\}}|dks�|r�t| � d||���dS )z�
Set inodes limits for package
Called from cl-syncpkgs
:param package_name: Package name
:param limits: Limits to set: {"inodes_soft":1000, "inodes_hard":2000}
:return: None
r zQuota is disabledr z Unspecified limits for cl-quotaZ inodes_softZ inodes_hardz-Incorrect limits for cl-quota: %(bad_limits)s�
bad_limitsz, )r r z
--package=z
--soft-limit=z
--hard-limit=z
Inodes limitsN)
r' r( r r: �set�joinr= r r% r[ )r rY r2 r r` r\ r] r^ r r r � set_package_limits_independently# s" z+ClQuotaLib.set_package_limits_independentlyN)F)r
r r r% r& r r' r( r- rB rC rD r* rG rR rU rV r[ r_ rc r r r r r 2 s"
0 r )Z
__future__r rM �typingr r r Z cllimits.libr Zclcommon.clexceptionr Zclcommon.utilsr r Zcllimits.lib.utilsr
r Z cldetectlib�osr r r r r r r r �<module>
s