Edit File: clquota_lib.cpython-37.pyc
B 5/9e�1 � @ 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