Current File : //opt/alt/python37/lib/python3.7/site-packages/wmt/common/__pycache__/notification.cpython-37.pyc
B
��wa}% � @ s d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZ d dlmZmZmZ d d lmZ d d
lmZ d d lmZ d d
dddgddgddgd�ZG dd� d�Z eG dd� d��Z eG dd� d��Z eG dd� d��Z eG d d � d ��Z dS ) � N)� dataclass)� timedelta�datetime)�Header)�
MIMEMultipart)�MIMEText)� PrettyTable�ALL)�WMT_TEMPLATES_DIR�&LICENSE_EXPIRED_FAREWELL_LETTER_MARKER� JWT_TOKEN)�WmtEmailException)�clemail)�
MailHelperZAllZ
SuccessfulZFailedZUndonezAverage time (ms)ZURLz Error codes)�summary_report� error_report�duration_reportc @ s e Zd ZdZdZdZdS )�SupportedNotificationTypesZalert�reportZfarewellN)�__name__�
__module__� __qualname__�ALERT�REPORT�FAREWELL� r r �H/opt/alt/python37/lib/python3.7/site-packages/wmt/common/notification.pyr # s r c @ s& e Zd ZU eed< eed< eed< dS )�Email�subject� text_sample� html_sampleN)r r r �str�__annotations__r r r r r ) s
r c @ s e Zd ZU eed< eed< dS )� EmailReport� text_option� html_optionN)r r r r! r" r r r r r# 0 s
r# c @ sv e Zd ZU eed< eed< dZeed< dZeed< dZeed< dZ eed< d Z
eed
< e d d � �Z d
d� Z
dd� ZdS )�
EmailTemplate� target_mail� from_mailNr r r Z
Administrator� target_nameZen_US�localec C s< t j�td�} t| d��}t�|�� ��d�}W d Q R X |S )Nzlogo.png�rbzutf-8) �os�path�joinr
�open�base64Z b64encode�read�decode)Z logo_pathZlogoZlogo_img_encodedr r r � _get_logoB s zEmailTemplate._get_logoc C sT | j |d�}|tjtjgkr.|�d| jji� |tjkrP|�| jj| jjd�� |S )zI
Convert to txt template keys
see wmt_notify.txt
)�TONAME�DATEZ ERROR_REPORT)ZSUMMARY_REPORTZDURATION_REPORT) r) r r r �updater r$ r r )�self�date� notify_type�templater r r �to_text_templateI s
z EmailTemplate.to_text_templatec C sZ | j || �� d�}|tjtjgkr4|�d| jji� |tjkrV|�| jj| j jd�� |S )zK
Convert to html template keys
see wmt_notify.html
)r4 r5 ZLOGOZERROR_HTML_REPORT)ZSUMMARY_HTML_REPORTZDURATION_HTML_REPORT)
r) r3 r r r r6 r r% r r )r7 r8 r9 r: r r r �to_html_template] s
z EmailTemplate.to_html_template)r r r r! r" r r# r r r) r* � staticmethodr3 r; r< r r r r r& 6 s
r& c @ s� e Zd ZU eed< eed< eed< eed< eed�dd��Zedd � �Z d
d � Z
d d
� Z edd� �Z e
d�dd�Zdd� Zed�dd�Zed�dd�Zed�dd�Zd S ) �Notifier� target_email�
from_emailr �notification_type)�returnc C s� | j tjkr&t�t�� tdd� d�S | j tjkrBt�t�� d�S | j tjkr�t j
� t �s^dS t
�
� t j
�t � }t|d �}|dkr�t|�� d�S t|d �}t|�� d
�S ntd | j � ���d S ) N� )�daysz%Y-%m-%dz%Y-%m-%d %H:%M:%S� i�Q r z day(s) agoi z hour(s) agozunexpected notifier type: )rA r r r ZstrftimeZnowr r r r, r- �existsr �time�getmtime�intr! �NotImplementedError)r7 ZsecondsrD Zhoursr r r �periodz s zNotifier.periodc C s t td��� d S )N�w)r/ r �closer r r r � create_farewell_letter_marker� s z&Notifier.create_farewell_letter_markerc C sD t � }| �� }| �|�}|�|j|jg|� | jtjkr@| � � d S )N)
r �_generate_template_data�_generate_final_messageZsendmailr( r' rA r r rN )r7 Z mail_managerr: �messager r r �notify� s
zNotifier.notifyc C s� | j tjkr(tj�td�tj�td�fS | j tjkrPtj�td�tj�td�fS | j tjkrxtj�td�tj�td�fS t d| j � ���d S )Nzwmt_notify_solo.txtzwmt_notify_solo.htmlz
wmt_alert.txtzwmt_alert.htmlzwmt_farewell_solo.txtzwmt_farewell_solo.htmlzunexpected notifier type: )
rA r r r, r- r. r
r r rJ )r7 r r r �_get_templates� s zNotifier._get_templatesc C s t | S )N)�HEADERS_REPORT_MAPPING)�sectionr r r �_get_table_headers� s zNotifier._get_table_headersc C sh d| j � �}| �d|�}| �d|�}|r0|�� n|}|rJ|jddttd�n|}t|t�|�dd��d�S ) NzNo domains for �text�htmlT)�formatZborderZhrulesZvrulesztext-align: centerztext-align: left)r$ r% ) rK �
_get_tableZ
get_stringZget_html_stringr r# rX Zunescape�replace)r7 rU Zno_domains_messageZ
text_tableZ
html_tableZtext_final_strZhtml_final_strr r r �_generate_tables� s zNotifier._generate_tablesc C sX | j | }|sd S t| �|��}d|_t|t�s6|g}x |D ]}|�|�|�� q<W |S )N�l)r r rV Zalign�
isinstance�listZadd_rowZ to_template)r7 � alternativerU Z
table_data�table�rowr r r rZ � s
zNotifier._get_tablec C sf | j | jd�}| jtjtjgkr4|�d| �d�i� | jtjkr\|�| �d�| �d�d�� tf |�S )N)r' r( r r r )r r ) r? r@ rA r r r r6 r\ r& )r7 r: r r r rO � s
z Notifier._generate_template_data)r c
C s | � � \}}tj�|�r$tj�|�s:td|� d|� d���y�| j}d| j� d|� �}| jtjkrvd| j� � � d|� �}| jtj
kr�d| j� �}t j j
||�|| j�|d �\}}t j j
||�|| j�|d �\}}W n8 t jjjk
�r } ztd
| � ���W d d } ~ X Y nX t|||d �S ) Nz unable to find email templates: z, z. Try to reinstall packagezWeb Monitoring Tool z for �[z] z1[EXPIRED CloudLinux LICENSE] Web Monitoring Tool )Z
templ_datar z cannot generate email, reason: )r r r )rS r, r- rF r
rK rA r r �upperr r ZClEmailZgenerate_mail_jinja2r; r< Zjinja2�
exceptionsZ
TemplateErrorr )
r7 r Z
text_templateZ
html_templaterK r � text_body�_� html_body�er r r �_generate_email� s& zNotifier._generate_email)�
template_datac C s� | � |�}|j�dd��d�}|j�dd��d�}td�}|�t|dd�� |�t|dd�� t|j d��� |d< |j
|d< |j |d< |S ) Nzutf-8�xmlcharrefreplacer` ZplainrX ZSubjectZFromZTo) rj r �encoder2 r r Zattachr r r r( r' )r7 rk Zemailrf rh rQ r r r rP � s
z Notifier._generate_final_messageN)r r r r! r" �dict�propertyrK r= rN rR rS rV r# r\ rZ r& rO rj rP r r r r r> s s
r> )!rX r, r0 rG Z dataclassesr r r Z email.headerr Zemail.mime.multipartr Zemail.mime.textr Z prettytabler r Zwmt.common.constr
r r Zwmt.common.exceptionsr
Zclcommonr Zclcommon.mail_helperr rT r r r# r& r> r r r r �<module> s6
=
Mini Shell