Edit File: ep_nproc_validator.cpython-37.pyc
B 5/9e1 � @ s� d dl mZ d dlmZmZ d dlmZmZmZm Z m Z m Z m Z d dl mZmZmZ ddlmZmZmZmZmZ G dd� de�Zd S ) � )�absolute_import)�LimitsDataStorage�DEFAULTS)�Package�User�Reseller� OWN_SOURCE�RESELLER_SOURCE�PACKAGE_SOURCE� GLOBAL_SOURCE)�Dict�Tuple�Union� )� BaseValidator� ENTITY_USER�ENTITY_PACKAGE�ENTITY_RESELLER�ENTITY_DEFAULTSc sp e Zd Zd� fdd� Zedd� �Zdd� Zdd � Zd d � Zd d � Z dd� Z ddd�Z ddd�Z ddd�Z � ZS )�EpNprocValidatorNc s t t| �j|d� d| _d S )N)�_limits_data_storage� )�superr �__init__�diff_nproc_and_ep)�selfr )� __class__� �+py/cllimits_validator/ep_nproc_validator.pyr s zEpNprocValidator.__init__c C s" d� | j�}t| j�}||d< |S )Nz�NPROC limit must be greater than EP + {} limit, because number of processes and threads within LVE includes also Apache processes/threads, SSH sessions and etc, which enter into LVE.� common_msg)�formatr �dictZ _message_dict)r r � message_dictr r r r" s z EpNprocValidator.message_dictc C s( |dks|dkrd}n||| j k}|S )Nr T)r )r �ep�nproc�resultr r r �_ep_and_nproc_is_valid+ s z'EpNprocValidator._ep_and_nproc_is_validc C s� | j }| j�|�}t|p |jj�}t|p.|jj�}| �||�}|jj}|jj} |rht |d< |j |d< �n| t ksx|t kr�t |d< |j |d< |t kr�d} n| t kr�d} nd} | dkr�dnd� | � � t|| �j�� �|d< n�| t k�rB|t k�rB| |k�rB||d< |tk�r|jj|d< n*|tk�r.|jj|d< n|tk�r�t|d< n@| t k�r�|t k�r�| |k�r�t |d< |j |d< d� || �|d< ||fS ) zY :param has_input_limits: User has input limits, which will be set to it �affected_entity� entity_idr# r$ Nz{} is inherited from {}� specific_msgzOBoth validated limits (EP, NPROC) inheritance from different sources: {} and {})r" �limits_providerZget_user_limits�intr# �valuer$ r& �sourcer �idr r �upper�getattr� capitalizer �package�namer �resellerr r ) r �userr# r$ �has_input_limitsr" Z user_limitsr% � ep_source� nproc_source�inheritance_limitr r r � _validate_nproc_and_ep_for_user3 sL z0EpNprocValidator._validate_nproc_and_ep_for_userc C sr | j }| j�|�}t|p |jj�}t|p.|jj�}| �||�}|jj}|jj} |rht |d< |j |d< �n| t ksx|t kr�t |d< |j |d< |t kr�d} n| t kr�d} nd} | dkr�dnd� | � � t|| �j�� �|d< n�| t k�r*|t k�r*| |k�r*||d< |tk�r|jj |d< n|tk�rjt|d< n@| t k�rj|t k�rj| |k�rjt |d< |j |d< d� || �|d< ||fS ) zY :param has_input_limits: User has input limits, which will be set to it r' r( r# r$ Nz{} is inherited from {}r) zOBoth validated limits (EP, NPROC) inheritance from different sources: {} and {})r" r* Zget_package_limitsr+ r# r, r$ r& r- r r3 r r r/ r0 r1 r Zproviderr r ) r r2 r# r$ r6 r" Zpackage_limitsr% r7 r8 r9 r r r �"_validate_nproc_and_ep_for_packagei sH z3EpNprocValidator._validate_nproc_and_ep_for_packagec C s� | j }| j�|�}t|p |jj�}t|p.|jj�}| �||�}|jj}|jj} |rft |d< |j |d< nn| t ksv|t kr�t |d< |j |d< |t kr�d} n| t kr�d} nd} | dkr�dnd� | � � t|| �j�� �|d< ||fS )zY :param has_input_limits: User has input limits, which will be set to it r' r( r# r$ Nz{} is inherited from {}r) )r" r* Zget_reseller_limitsr+ r# r, r$ r& r- r r3 r r r/ r0 r1 ) r r4 r# r$ r6 r" Zreseller_limitsr% r7 r8 r9 r r r �#_validate_nproc_and_ep_for_reseller� s, z4EpNprocValidator._validate_nproc_and_ep_for_resellerc C s� | j }t|t�r|jnt}| j�|�}t|p2|jj �}t|pB|j j �}| � ||�}|jj } |j j } |rxt |d< ||d< nl| tks�| tkr�t |d< ||d< | tkr�d} n| tkr�d} nd} | dkr�dnd�| �� t|| �j �� �|d< ||fS )zY :param has_input_limits: User has input limits, which will be set to it r' r( r# r$ Nz{} is inherited from {}r) )r" � isinstancer r3 r r* Zget_defaults_limitsr+ r# r, r$ r& r- r r r r/ r0 r1 ) r r4 r# r$ r6 r" Z id_entityZdefault_limitsr% r7 r8 r9 r r r �#_validate_nproc_and_ep_for_defaults� s. z4EpNprocValidator._validate_nproc_and_ep_for_defaultsFc C s� | j }d}t|t�r*| �||||�\}}nnt|t�rJ| �||||�\}}nNt|t�rn|sn| �||||�\}}n*t|t�r||s�|tkr�| � ||||�\}}||fS )NT) r" r= r r: r r; r r< r r> )r �validated_entityr# r$ r6 �check_reseller_defaultsr" r% r r r � _validate� s6 zEpNprocValidator._validatec C s | � |ddd|�\}}||fS )zF Validate limits which already are recorded in ve.cfg NF)rA )r r? r@ r% r" r r r �validate_existing_limits s z)EpNprocValidator.validate_existing_limitsc C sf | j }|�dd�}|�dd�}|dk s.|dk r2dnd}|dksB|sHd}n| �|||||�\}}||fS )z6 Validate limits which we want to set r$ Nr# TFr )r" �getrA ) r r? Z input_limitsr@ r" r$ r# r6 r% r r r �validate_input_limits s z&EpNprocValidator.validate_input_limits)N)F)F)F)�__name__� __module__� __qualname__r �propertyr" r& r: r; r< r> rA rB rD � __classcell__r r )r r r s 64%' ' r N)Z __future__r Zcllimitslib_v2r r Zcllimitslib_v2.lve_storager r r r r r r �typingr r r �baser r r r r r r r r r �<module> s $