Edit File: parser.cpython-37.pyc
B V�d� � @ 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