Current File : //opt/alt/python37/lib/python3.7/site-packages/wmt/common/__pycache__/config.cpython-37.pyc
B
��wa� � @ s� d dl Z d dlZd dlZd dlmZmZmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d dlmZ eG d d
� d
��ZG d d � d �ZdS )
� N)� dataclass�asdict�field)� CONFIG_PATH)�WmtConfigException)�parse)�get_admin_email)� gethostname)�Listc @ sd e Zd ZU dZdZeed< dZeed< dZe ed< e
e d�Z e
e ed < d
Zeed < d
Zeed < dS )
�CfgzC
Default values, in case config has not been specified yet
� �
ping_interval�
� ping_timeoutN� report_email)�default_factory� ignore_listT� summary_notification_enabled�alert_notifications_enabled)�__name__�
__module__� __qualname__�__doc__r
�int�__annotations__r r �strr �listr r
r �boolr � r r �B/opt/alt/python37/lib/python3.7/site-packages/wmt/common/config.pyr s
r c @ s| e Zd Zdd� Zdd� Zdd� Zed�dd �Zed
d � �Z e
d �d
d�Z edd� �Z dd� Z
ed�dd�Zed�dd�ZdS )�
ConfigManagerc C sD t j�� | _dt� � �| _t� | _| �� | _ | �
� | _ | � � | _
d S )Nzweb-monitoring-tool@)r Z__dataclass_fields__�keys�allowed_paramsr Z
from_emailr �default_report_email� _init_cfg�cfg�generate_ignored_domains�_ignored_domains�_get_target_emailZ target_email)�selfr r r �__init__ s
zConfigManager.__init__c C s | j jr| j jS | jS )a
This function checks to see which email address to use for TO: field of smtp.
If report_email has been defined by user then report_email will be used.
By default (in case not defined by user) default_report_email will be used
)r% r r# )r) r r r r( ' s z ConfigManager._get_target_emailc C s
t | j�S )N)r r% )r) r r r �to_dict/ s zConfigManager.to_dict)�returnc C sj | � � st� S | �� }t� }xH|�� D ]<\}}|| jkrVt�d|� dt| j�� �� q&t|||� q&W |S )Nzunsupported parameter "z:", please ensure config contains only allowed parameters: ) �
is_presentr �read�itemsr" �loggingZwarningr �setattr)r) �datar% �key�valuer r r r$ 2 s
zConfigManager._init_cfgc C s t j�t�S )N)�os�path�isfiler r r r r r- E s zConfigManager.is_present)�new_jsonc
C s� yt �|�}W n0 t jk
r> } ztt|���W dd}~X Y nX t|�� ��| j�shtdt | j�� ���| �
� |�}|� d�r�t |� d�t�r�|d �
d�|d< ttd��}t j||dd� W dQ R X tf |�| _| �
� S )z�
Changes configuration of wmt
Returns:
self.to_dict()
Raises:
WmtConfigException
Example:
wmt-api-solo --config-change {'key': 'val'}
Nz@some of passed params are unsupported, only allowed parameters: r �,�w� )�indent)�json�loads�JSONDecodeErrorr r �setr! �issubsetr" r r+ �get�
isinstance�split�openr �dumpr r% )r) r8 Z
new_config�eZconfig�fr r r �modifyI s
zConfigManager.modifyc
C sX y"t t��} t�| �}W d Q R X W n0 tjk
rR } ztt|���W d d }~X Y nX |S )N)rE r r= �loadr? r r )rH r2 rG r r r r. m s
zConfigManager.readc C s | � � | _d S )N)r$ r% )r) r r r �reloadv s zConfigManager.reloadc C s
|| j kS )z5
Check if domain is in ignored list.
)r' )r) Zdomainr r r �is_domain_ignoredy s z ConfigManager.is_domain_ignoredc C sB t � }x6| jjD ]*}t|�}d|kr0|�dd�}|�|� qW |S )al
Generates ignored domains from self.ignore_list and
returns it for using as cache in self.ignored_domains set().
Purpose of this function is to generate formatted url as "http://www.test.com"
from user entered urls in self.ignore_list.
It makes sense to check domains names with this cache before pings or reports
ZhttpsZhttp)r@ r% r r �replace�add)r) Zdomains�dZdomain_from_ignore_listr r r r& s z&ConfigManager.generate_ignored_domainsN)r r r r* r( r+ r r$ � staticmethodr- r rI r. rK r rL r@ r& r r r r r s $ r )r0 r5 r= Z dataclassesr r r Zwmt.common.constr Zwmt.common.exceptionsr Zwmt.common.url_parserr Zclcommon.cpapir Zsocketr �typingr
r r r r r r �<module> s
Mini Shell