Edit File: limits_validator.cpython-37.pyc
B 5/9e�2 � @ s� d dl mZ d dlmZ d dlmZmZmZ d dlm Z m Z d dl m Z d dl mZ d dlmZmZmZmZmZmZ dd lmZmZmZmZmZ dd l m Z d d l m Z G d d � d e �Z!dS )� )�absolute_import)�chain)� itervalues�iterkeys� iteritems)�LimitsDataStorage�DEFAULTS)�DEFAULT_PROVIDER)� defaultdict)�Dict�Tuple�List�Optional�Union�Any� )� BaseValidator� ENTITY_USER�ENTITY_PACKAGE�ENTITY_RESELLER�ENTITY_DEFAULTS)�EpNprocValidator)�LveCtlc @ s� e Zd Zddddd�ZddiZegZdd iZd(d d �Ze d d� �Z dd� Z e fdd�Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Ze ed%�d&d'��Zd S ))�LimitsValidatorzUser(s)z Package(s)z Reseller(s)zDefaults of resellers)�user�package�reseller�defaultsr zHoster defaults limitszinherited from Globalzinherited from HosterNc C s2 |pt � | _|pt| j�g| _d d d d d�| _d S )N)� common_msg� specific_msg�affected_entity� entity_id)r �limits_providerr �_list_validatorsZ message_dict)�selfZ_limits_data_storageZ validators� r% �)py/cllimits_validator/limits_validator.py�__init__5 s zLimitsValidator.__init__c C s� | d }| d }| d }| d }|dk r�|dk r�|t kr>d}q�|tkrRd�|�}q�|tkrfd�|�}q�|tkrzd �|�}q�d �|�}nd }|dkr�d }d �|||�}|S ) z� Format message string from single message dict. That function should be used for formatting message while validation input limits r r r r! Nz"Hoster defaults limits is affectedz.Defaults limits of Reseller "{}" are affected.z-Limits of user with LVE ID "{}" are affected.z%Limits of Reseller "{}" are affected.z$Limits of package "{}" are affected.� z{} {} {})r r �formatr r )�msg_dictr r r r! Zmsg_about_affectZ result_msgr% r% r& �+_format_message_string_from_single_msg_dictC s, z;LimitsValidator._format_message_string_from_single_msg_dictc O sZ t � }d}xF| jD ]<}|r,|j||�\}}n|j||�\}}||M }|�|� qW ||fS )a� Call each validator from validators list and processing results args and kwargs are argumets for calling of validator We validate limits which already are recorded in ve.cfg if only_input_limits is False. We validate limits which we want to set if only_input_limits is True We return aggregated result as bool value and list of messages which explain which limits are wrong and who holds those T)�listr# �validate_input_limits�validate_existing_limits�append) r$ Zonly_input_limits�args�kwargs�msg_list� total_resultZ validator�resultr* r% r% r& �#_call_validators_and_process_resulth s z3LimitsValidator._call_validators_and_process_resultc s� d}|t kr � j�t|��}nj|tkr>� j�t|�t|��}nJ|tkrX� j�t|��}n0|t kr�|t krz� j�t|��}n|}d}nd }� � d|||�\}}� fdd�|D �}||fS )NFTc s g | ]}� � |��qS r% )r+ )�.0r* )r$ r% r& � <listcomp>� s z9LimitsValidator.validate_input_limits.<locals>.<listcomp>) r r" Zget_user_by_uid�intr Z get_package_by_name_and_reseller�strr Zget_reseller_by_namer r r5 ) r$ � entity_typer! Z input_limitsr Zcheck_reseller_defaultsZvalidated_entityr3 r2 r% )r$ r& r- ~ s( z%LimitsValidator.validate_input_limitsc s" t � jj�}� fdd�|D �}|S )Nc s g | ]}� � d |��qS )F)r5 )r6 r )r$ r% r& r7 � s z3LimitsValidator._validate_users.<locals>.<listcomp>)r r" �users)r$ r; r4 r% )r$ r& �_validate_users� s z LimitsValidator._validate_usersc s" t � jj�}� fdd�|D �}|S )Nc s g | ]}� � d |��qS )F)r5 )r6 r )r$ r% r& r7 � s z6LimitsValidator._validate_packages.<locals>.<listcomp>)r r" �packages)r$ r= r4 r% )r$ r& �_validate_packages� s z"LimitsValidator._validate_packagesc s" t � jj�}� fdd�|D �}|S )Nc s" g | ]}|j d k r� �d|��qS )NF)r r5 )r6 r )r$ r% r& r7 � s z7LimitsValidator._validate_resellers.<locals>.<listcomp>)r r" � resellers)r$ r? r4 r% )r$ r& �_validate_resellers� s z#LimitsValidator._validate_resellersc C sT g }xJt | jj�D ]:}| jj�|t�}|tkr:|jd kr:q|�| �d|�� qW |S )NF)r r" r r? �getr r/ r5 )r$ r4 Z default_nameZdefault_entityr% r% r& �_validate_defaults� s z"LimitsValidator._validate_defaultsc C s2 x,t | j�D ] \}}|dkr P |�||�}q W |S )z[ Replacing some specific messages for improve readability of error message N)r �"dict_of_replaced_specific_messages�replace)r$ r Z old_spec_msgZ new_spec_msgr% r% r& �_replace_specific_msgs� s z&LimitsValidator._replace_specific_msgsc C s || j kr| j | S |S dS )zY Replacing some ids of entities for improve readability of error message N)� dict_of_replaced_ids_of_entities)r$ r! r% r% r& �_replace_ids_of_entities� s z(LimitsValidator._replace_ids_of_entitiesc C s || j kr| j | S |S dS )z[ Replacing some names of entities for improve readability of error message N)�"dict_of_replaced_names_of_entities)r$ Z entity_namer% r% r& �_replace_names_of_entities� s z*LimitsValidator._replace_names_of_entitiesc C s* t dd� �}d}x^|D ]V}|d }| �|d �}| �|d �}|| jkrJq|| | �|| �|d �d�� qW x�t|�D ]�\}} d �|dkr�d n||�}x�t| �D ]z\} } t� } x@| D ]8} | d dkr�| d }nd �| d | d �}| �|� q�W d � d d� t | �D ��} d�|| � � | �}q�W qzW |S )zb Format message string from result of execution of validation the existing limits c S s t t�S )N)r r, r% r% r% r& �<lambda>� � zLLimitsValidator._format_message_string_for_existing_limits.<locals>.<lambda>Nr r r! r )r! r z{}{} r( z{} ({})z, c s s | ]}t |�V qd S )N)r9 )r6 �itemr% r% r& � <genexpr> s zMLimitsValidator._format_message_string_for_existing_limits.<locals>.<genexpr>z {}{}: {} ) r rI rG � list_of_deleted_ids_of_entitiesr/ rE r r) r, �join�set� capitalize)r$ �total_result_listZtotal_result_dict� total_msgr* Z name_of_checkZtype_of_affected_entityr! r Z entities_dictr: Zlist_entity_dictsZaffected_entitiesZ entity_dictr r% r% r& �*_format_message_string_for_existing_limits� s< z:LimitsValidator._format_message_string_for_existing_limitsc C s6 dd� t | �� | �� | �� | �� �D �}| �|�}|S )Nc S s" g | ]\}}|D ] }|s|�qqS r% r% )r6 r4 Z msg_dict_listr* r% r% r& r7 s z<LimitsValidator.validate_existing_limits.<locals>.<listcomp>)r r< r> r@ rB rT )r$ rR rS r% r% r& r. s z(LimitsValidator.validate_existing_limits)�returnc C s� t � } | �� }| �d� xb|D ]Z}|dkr,q | �|�}y,t|d �dd��}|dkr`|dk r`dS W q tk rv Y q X q W dS )zL Check is Low PMEM limit present :return True/False Fr Zpmem�*r( i T)r Zget_panel_users_uid_listZ _load_infoZget_limits_by_user_idr8 rD � ValueError)Z_lvectlZ_panel_uids_listZuidZlimitsZpmem_limit_bytesr% r% r& �is_low_pmem_limit_present% s z)LimitsValidator.is_low_pmem_limit_present)NN)�__name__� __module__� __qualname__rH rF r rN rC r' � staticmethodr+ r5 r r- r< r> r@ rB rE rG rI rT r. �boolrX r% r% r% r& r s, % 3r N)"Z __future__r � itertoolsr Z future.utilsr r r Zcllimitslib_v2r r Zclveconfig.ve_config_readerr � collectionsr �typingr r r r r r �baser r r r r Zep_nproc_validatorr Zcllimitsr �objectr r% r% r% r& �<module> s