j d Z ddlZddlZddlZddlmZmZ ddlmZm Z ddl
m Z dZ ej
d Z G d d
e ZdS ) a
This module contains SSA specific config parser:
- has defaults
- adds default values for ones missing in config file
- allows duplicates, the latest value is loaded
- performs value normalization during loading items:
- to int
- to float
- to bool
- 'correlation' is case-insensitive and allowed to be
on, off, 1, 0, yes, no, true, false
- loads default configuration in case of missing config file
N) ConfigParserNoSectionError)CallableList ) config_filessa_confssa_configuration_parserc
e Zd ZdZ fdZedefd Zedefd Zedefd Z de
de d e de fd
Z e
d d fde d
ededee fdZd ZddZdedd fdZddZd Zd Z xZS )SSAConfigParserz2
SSA specific configparser implementation
c
dddddddddd }d d
d d d
ddddd | _ | | _ t |d d S )N
g @ Tg? ) domains_number urls_numberrequest_numberrequests_durationtime correlationcorrelation_coefficient ignore_list summary_notification_enabledz!Size of TOP list for slow domainsz Size of TOP list for slow urlsz[The threshold value of slow requests number in the period of time to mark URL as a slow onez2The threshold value of request duration in secondsz:Period of time in hours required to analyze these requestsz.Flag to enable or disable correlation (On/Off)z.The threshold value of correlation coefficientzEList of URLs or domains that should not be included in a daily reportzBFlag to enable or disable sending daily reports by e-mail (On/Off)F)strict) _commentscopy ssa_defaultssuper__init__)selfdefaults __class__s O/opt/cloudlinux/venv/lib64/python3.11/site-packages/ssa/configuration/parser.pyr zSSAConfigParser.__init__( s !$ '* ,0
B; {!UPK'Wb,p
%MMOO
%00000 returnc dS )N)r r r r r! s r$ int_optionszSSAConfigParser.int_optionsB s HHr% c dS )N)r r r( r) s r$
float_optionsz SSAConfigParser.float_optionsF s ==r% c dS )N)r r r( r) s r$ bool_optionsz SSAConfigParser.bool_optionsJ s <.value_getterj s ***00 '179 9 94+++00gvNNN4---00 179 9 9 y r% c * g | ]}| | fS r( r( ).0r1 r? s r$
z)SSAConfigParser.items..v s( GGG6 f--.GGGr% ) _defaultsr update _sectionsKeyErrordefault_sectionloggerinforestore_default_conflistkeys)r! r0 r8 r3 orig_keysr> r? s `` @@r$ itemszSSAConfigParser.itemsZ s
N ! ! (
HHT^G, - - - - ( ( ($... 7 #% % %
% % ' ' ' ' ' ( NN
!
!
!
!
!
!
! HGGGYGGGGs A ABBc t t 5 ddd n # 1 swxY w Y | t dS # t $ r2 | t
| i Y dS w xY w)z`
Try to read configuration from file,
load defaults if file not present
N)openr readFileNotFoundError read_dictrG r" r) s r$
read_ssa_confz SSAConfigParser.read_ssa_confx s
?k""
IIk " " " " " ? ? ? NNOT]]__= > > > > > > ?s, A $ A (A ( A 8BBc L d} t j dt j t
d }| |d t j |j t
|S| t j
|j r t j |j dS dS dS # |R| t j
|j rt j |j w w w xY w)z=
Try to write current configuration to file,
NwF)modedirdelete)space_around_delimiters) tempfileNamedTemporaryFileospathdirnamer writerenamenamecloseexistsremove)r! temp_files r$ write_ssa_confz SSAConfigParser.write_ssa_conf s
. 3"'//R]B^B^glmmmI JJy%J @ @ @ Iink 2 2 2$!!!7>>).11 .Iin----- %$. . $!!!7>>).11 .Iin---- %.s
A/C AD#argsc t | v rt n| d }| D ])\ }}| ||t | *dS )zQ
Load options from the dict into the main section of the config.
r N)rG sectionsrN setstr)r! rh r0 r1 values r$ overridezSSAConfigParser.override sq &5
%G%G//T]]__]^M_ !ZZ\\ 2 2MFE HHWfc%jj 1 1 1 1 2 2r% Tc |r!d | j d }n
| j d }| j D ]7}| ||| j | | 8dS )z
Parent method is overrided in order to:
- do not print default section
- call its own write_section method instead of parent one.
z {} r N)format _delimitersrE
write_sectionrN )r! fprZ r> r0 s r$ r` zSSAConfigParser.write s # $
d .q 122AA #A~ D DG r7 $w 7 = = ? ?
D
D
D
D D Dr% c
| j r
| j d nd}| d | |D ]\ }}| j | ||| }|| j s'|t
| dd z }nd}| j |d }|| | | d | | d | d| | d | d dS )
z
The difference between this one and the parent _write_section method
is in writing the comments to every option also
r #z[{}]
N
z
r z
)
_comment_prefixesr` rp _interpolation before_write_allow_no_valuerl replacer get) r! rs section_name
section_items delimitercomment_prefixkeyrm comments r$ rr z SSAConfigParser.write_section s-
7;6LU/22RU
..///' J JJC'44T<5:< u > > > X> =e = = = X=
C
%(
-0
$3 H HS H H H$(K H H H H<
?
?
?. . . ."2T 2d 2 2 2 2 D D D D *
:
:
:
:
:
:
:r% r )r loggingr] r[ configparserr r typingr r internal.constantsr rG getLoggerrH r r( r% r$ r s 5 5 5 5 5 5 5 5 ! ! ! ! ! ! ! ! - , , , , , 5 6 6j: j: j: j: j:l j: j: j: j: j:r%