Current File : //opt/alt/python37/lib/python3.7/site-packages/ssa/configuration/__pycache__/parser.cpython-37.pyc
B
r�b� � @ sT d Z ddlmZmZ ddlmZmZ ddlmZ ddl m
Z
dZ G dd � d e�Z d
S ) 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
� )� ConfigParser�NoSectionError)�Callable�List)�is_cl_solo_edition� )� config_fileZssa_confc s� e Zd ZdZ� fdd�Zeed�dd��Zeed�dd��Zeed�d d
��Z e
e e e d �d d
�Z e
ddfe eeee d�dd�Zdd� Zdd�dd�Zedd�dd�Zd dd �Zd d � Z� ZS ) �SSAConfigParserz2
SSA specific configparser implementation
c
sp t dd�}|rdndd|r dndd|r*dnd|r4dndd d
dd � }d d
dddddddd � | _t� j|dd� d S )NT)�skip_jwt_checkr �
� � g @Fg�������?� ) �domains_number� urls_number�request_number�requests_duration�time� correlation�correlation_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))�strict)r � _comments�super�__init__)�selfZcl_solo�defaults)� __class__� �I/opt/alt/python37/lib/python3.7/site-packages/ssa/configuration/parser.pyr % s(
zSSAConfigParser.__init__)�returnc C s dS )N)r r r r r )r r r r � int_options? s zSSAConfigParser.int_optionsc C s dS )N)r r r )r r r r �
float_optionsC s z SSAConfigParser.float_optionsc C s dS )N)r r r )r r r r � bool_optionsG s z SSAConfigParser.bool_options)�meth�section�optionr! c C s: y|||�}W n& t k
r4 |||| �� d�}Y nX |S )z`
If the value is missing in config file, e.g. time=
set the default one
)�vars)�
ValueErrorr )r r% r& r' �valr r r �empty_value_fallbackK s
z$SSAConfigParser.empty_value_fallbackFN)r& �rawr( r! c st �j �� � y� ��j� � W n& tk
rD ��jkr@t���Y nX t� �� �}� ��fdd���fdd�|D �S )znReturn a list of (name, value) tuples for each option in a section.
Adds special conversions.
c sZ | �j kr���j�| �S | �jkr4���j�| �S | �jkrN���j�| �S � | S d S )N)r$ r+ �
getbooleanr" �getintr# �getfloat)r' )�dr& r r r � value_getterd s
z+SSAConfigParser.items.<locals>.value_getterc s g | ]}|� |�f�qS r r )�.0r' )r1 r r �
<listcomp>p s z)SSAConfigParser.items.<locals>.<listcomp>) � _defaults�copy�update� _sections�KeyError�default_sectionr �list�keys)r r& r, r( Z orig_keysr )r0 r& r r1 r �itemsW s
zSSAConfigParser.itemsc C sN y"t t�� W dQ R X | �t� W n& tk
rH | �t| �� i� Y nX dS )z`
Try to read configuration from file,
load defaults if file not present
N)�openr �read�FileNotFoundError� read_dictr9 r )r r r r �
read_ssa_confr s
z SSAConfigParser.read_ssa_confc C s( t td��}| j|dd� W dQ R X dS )z=
Try to write current configuration to file,
�wF)�space_around_delimitersN)r= r �write)r �
configfiler r r �write_ssa_conf~ s z SSAConfigParser.write_ssa_conf)�argsr! c C sH t | �� krt n
| �� d }x&|�� D ]\}}| �||t|�� q&W dS )zQ
Load options from the dict into the main section of the config.
r N)r9 �sectionsr< �set�str)r rG r& r' �valuer r r �override� s zSSAConfigParser.overrideTc C sN |rd� | jd �}n
| jd }x(| jD ] }| �||| j| �� |� q(W dS )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� _delimitersr7 �
write_sectionr< )r �fprC r0 r& r r r rD � s
zSSAConfigParser.writec
C s� | j r| j d nd}|�d�|�� x�|D ]�\}}| j�| |||�}|dk sR| jsh|t|��dd� }nd}| j� |d�}|dkr�|�|� |� d�� q*|�|� d |� d|� |� d�� q*W |�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_prefixesrD rM �_interpolation� before_write�_allow_no_valuerJ �replacer �get) r rP � section_name�
section_items� delimiterZcomment_prefix�keyrK �commentr r r rO � s $z SSAConfigParser.write_section)T)�__name__�
__module__� __qualname__�__doc__r �property�tupler" r# r$ r rJ r+ r9 �bool�dictr r< rA rF rL rD rO �
__classcell__r r )r r r s"
r N)
rb � configparserr r �typingr r �clcommon.lib.cleditionr �internal.constantsr r9 r r r r r �<module> s
Mini Shell