QJ'i^ d dl mZ d dlmZ d dlmZmZmZmZm Z m
Z
d dl m Z d dl
mZmZ d dlmZ ddlmZmZmZmZmZ dd lmZ G d
d Zd S )
) defaultdict)chain)AnyDictListOptionalTupleUnion)LveCtl)DEFAULTSLimitsDataStorage)DEFAULT_PROVIDER )ENTITY_DEFAULTSENTITY_PACKAGEENTITY_RESELLER ENTITY_USER
BaseValidator)EpNprocValidatorc e Zd ZdddddZddiZegZdd iZdd Ze d Z
d
Z e fdZ
d Zd Zd Zd Zd Zd Zd Zd Zd Ze defd Zd
S )LimitsValidatorzUser(s)z
Package(s)z Reseller(s)zDefaults of resellers)userpackageresellerdefaultsr zHoster defaults limitszinherited from Globalzinherited from HosterNc |p
t | _ |pt | j g| _ d d d d d| _ d S )N)
common_msg specific_msgaffected_entity entity_id)r
limits_providerr _list_validators message_dict)self_limits_data_storage
validatorss )py/cllimits_validator/limits_validator.py__init__zLimitsValidator.__init__2 sY 3 J7H7J7J * !
T 1 2 2/
#
c | d }| d }| d }| d }|M|K|t k rd}n?|t k rd| d}n-|t k rd | d}n|t k rd
| d}n d | d}nd }|d }| 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 "z" are affected.z Limits of user with LVE ID "zLimits of Reseller "zLimits of package " )r r r r )msg_dictr r r r msg_about_affect
result_msgs r' +_format_message_string_from_single_msg_dictz;LimitsValidator._format_message_string_from_single_msg_dict@ s l+
/ "#45 [) &9+@H$$#G O33#]9#]#]#] K//#\)#\#\#\ O33#T)#T#T#T #S#S#S#S ! L"FF%5FF FF
r) c g }d}| j D ]?}|r |j |i |\ }}n |j |i |\ }}||z }| | @||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)r" validate_input_limitsvalidate_existing_limitsappend) r$ only_input_limitsargskwargsmsg_list total_result validatorresultr- s r' #_call_validators_and_process_resultz3LimitsValidator._call_validators_and_process_resulta s . & &I
W#B9#BD#SF#S#S #E9#Et#Vv#V#V F "L OOH % % % %X%%r) c : d}|t k r( j t | }n|t k r6 j t
| t
| }nx|t k r( j t
| }nE|t k r8|t k r( j t
| }n|}d}nd } d||| \ }} fd|D }||fS )NFTc : g | ]} | S )r0 ).0r- r$ s r'
z9LimitsValidator.validate_input_limits.. s( hhhS[DDDXNNhhhr) ) r r! get_user_by_uidintr get_package_by_name_and_resellerstrr get_reseller_by_namer r r< ) r$ entity_typer input_limitsr check_reseller_defaultsvalidated_entityr9 r8 s ` r' r2 z%LimitsValidator.validate_input_limitsw s6 #( + % % # 3 C CC NN S S
N
*
* # 3 T TUXYbUcUcehiqerer s s
O
+
+ # 3 H HY X X
O
+
+H$$#'#7#L#LSQZ^^#\#\ #, &* # # # !%!I!I # "
"
h ihhh_ghhhX%%r) c ` j j } fd|D }|S )Nc < g | ]} d | S Fr< )r@ r r$ s r' rA z3LimitsValidator._validate_users.. s) ZZZD$::5$GGZZZr) )r! usersvalues)r$ rO r; s ` r' _validate_usersz LimitsValidator._validate_users s; $*1133ZZZZTYZZZ
r) c ` j j } fd|D }|S )Nc < g | ]} d | S rM rN )r@ r r$ s r' rA z6LimitsValidator._validate_packages.. s) cccw$::5'JJcccr) )r! packagesrP )r$ rT r; s ` r' _validate_packagesz"LimitsValidator._validate_packages s; '07799ccccZbccc
r) c ` j j } fd|D }|S )Nc J g | ] }|j d| S NF)r r< )r@ r r$ s r' rA z7LimitsValidator._validate_resellers.. s< 4 4 4PX &2 ::5(KK222r) )r! resellersrP )r$ rY r; s ` r' _validate_resellersz#LimitsValidator._validate_resellers sK (299;; 4 4 4 4\e 4 4 4
r) c g }| j j D ]c}| j j |t
}|t
k r|j :| | d| d|S rX )r! r keysrY getr r4 r< )r$ r; default_namedefault_entitys r' _validate_defaultsz"LimitsValidator._validate_defaults s 0 9 > > @ @ [ [L ! 1 ; ? ? h W WN ))n.E.M MM$BB5.YY Z Z Z Z
r) c x | j D ] \ }}| n| || } |S )z[
Replacing some specific messages for improve readability of error message
)"dict_of_replaced_specific_messagesitemsreplace)r$ r old_spec_msg new_spec_msgs r' _replace_specific_msgsz&LimitsValidator._replace_specific_msgs sQ
+/*Q*W*W*Y*Y L L &L,#'// lKKLLr) c 2 || j v r
| j | S |S )zY
Replacing some ids of entities for improve readability of error message
) dict_of_replaced_ids_of_entities)r$ r s r' _replace_ids_of_entitiesz(LimitsValidator._replace_ids_of_entities s&
= = =8C Cr) c 2 || j v r
| j | S |S )z[
Replacing some names of entities for improve readability of error message
)"dict_of_replaced_names_of_entities)r$ entity_names r' _replace_names_of_entitiesz*LimitsValidator._replace_names_of_entities s&
$A A A:;G Gr) c t d }d}|D ]}|d }| |d }| |d }|| j v rJ|| | || |d d | D ]\ }} |dn| | d }| D ]\ }
} g } | D ]<}
|
d |
d }n|
d d
|
d d }| | =d d
t | D } | |
d| d }|S )zb
Format message string from result of execution of validation the existing limits
c * t t S N)r listr? r) r' zLLimitsValidator._format_message_string_for_existing_limits.. s D0A0A r) Nr r r r )r r r+
z ()z, c 3 4 K | ]}t | V d S rq )rE )r@ items r' zMLimitsValidator._format_message_string_for_existing_limits.. s( -[-[Dc$ii-[-[-[-[-[-[r) z: )
r rn rj list_of_deleted_ids_of_entitiesr4 rg rc joinset
capitalize)r$ total_result_listtotal_result_dict total_msgr-
name_of_checktype_of_affected_entityr r
entities_dictrG list_entity_dictsaffected_entities entity_dictr s r' *_format_message_string_for_existing_limitsz:LimitsValidator._format_message_string_for_existing_limits s ((A(A B B *
H $\ 2M&*&E&EhO`Fa&b&b # 55h{6KLLID @@@ m ,-D E L L & $ ; ;H^ >K">2:*5k*B-8-E*g*gUcId*g*g*g%,,_====$(II-[-[CHYDZDZ-[-[-[$[$[! ( [+*@*@*B*B [ [FW [ [ [
\ r) c d t | | | | D }| | }|S )Nc & g | ]\ }}|D ]}||S r? r? )r@ r;
msg_dict_listr- s r' rA z. sJ
.
,
6
r) )r rQ rU rZ r` r )r$ r} r s r' r3 z(LimitsValidator.validate_existing_limits s
27$$&&''))((**'')) 3 3
CCDUVV r) returnc H t } | }| d |D ]g}|dk r | | } t |d dd }|dk r |dk r dS X# t $ r Y dw xY wdS )zL
Check is Low PMEM limit present
:return True/False
Fr pmem*r+ i T)r get_panel_users_uid_list
_load_infoget_limits_by_user_idrC rd
ValueError)_lvectl_panel_uids_listuidlimitspmem_limit_bytess r' is_low_pmem_limit_presentz)LimitsValidator.is_low_pmem_limit_present s ((";;==5!!!# Caxx 22377F
#&vf~'='=c2'F'F#G#G #q((-=@Q-Q-Q 44
us 5B
B B )NN)__name__
__module__ __qualname__rl ri r ry rb r( staticmethodr0 r< r r2 rQ rU rZ r` rg rj rn r r3 boolr r? r) r' r r se !+ * *& ,($ '# !8*&
\ @& & &, Td & & & &:
) ) )V
t \ r) r N) collectionsr itertoolsr typingr r r r r r
cllimitsr cllimitslib_v2r r
clveconfig.ve_config_readerr baser r r r r ep_nproc_validatorr r r? r) r' r s $ # # # # # : : : : : : : : : : : : : : : : 6 6 6 6 6 6 6 6 8 8 8 8 8 8 ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ 0 0 0 0 0 0Y Y Y Y Y Y Y Y Y Yr)