$K r 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 lm Z m Z g d
g dddgdZ G d d Z e G d d Z e G d d Z!e G d d Z"e G d d Z#dS ) N) dataclass) timedeltadatetime)Header)
MIMEMultipart)MIMEText) PrettyTableALL)WMT_TEMPLATES_DIR&LICENSE_EXPIRED_FAREWELL_LETTER_MARKER JWT_TOKEN)WmtEmailException)clemail)
MailHelper) get_hostname get_ip_addr)All
SuccessfulFailedUndoneAverage time (ms))URLErrorsz Error codesr r )summary_report error_reportduration_reportc e Zd ZdZdZdZdS )SupportedNotificationTypesalertreportfarewellN)__name__
__module__ __qualname__ALERTREPORTFAREWELL /builddir/build/BUILDROOT/cl-web-monitoring-tool-3.2.3-1.el9.cloudlinux.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/wmt/common/notification.pyr r % s E
FHHHr) r c . e Zd ZU eed< eed< eed< dS )Emailsubject text_sample html_sampleNr" r# r$ str__annotations__r( r) r* r, r, + s3
LLLr) r, c $ e Zd ZU eed< eed< dS ) EmailReport text_option html_optionNr0 r( r) r* r4 r4 2 s* r) r4 c e Zd ZU eed< eed< dZeed< dZeed< dZeed< dZ eed< d Z
eed
< e d Z e d Z
e d
Zd Zd ZdS )
EmailTemplate target_mail from_mailNr r r
Administrator target_nameen_USlocalec " t pdS NUNKNOWN)r r( r) r*
_get_hostnamezEmailTemplate._get_hostnameD s ~~**r) c , | sdS t | pdS r@ )r )hostnames r* r zEmailTemplate.get_ip_addrH s" 98$$1 1r) c t j t d } t | d 5 }t j | d }d d d n # 1 swxY w Y |S )Nzlogo.pngrbutf-8) ospathjoinr openbase64 b64encodereaddecode) logo_pathlogologo_img_encodeds r* _get_logozEmailTemplate._get_logoN s GLL!2J??
)T
"
" Md % / < < C CG L L M M M M M M M M M M M M M M M s :A<<B B c Z | }| j ||| | d}|t j t j fv r!| d| j j i |t j k r,| | j j | j
j d |S )zI
Convert to txt template keys
see wmt_notify.txt
)TONAMEDATEHOSTNAMEIP_ADDR ERROR_REPORT)SUMMARY_REPORTDURATION_REPORT) rB r< r r r& r% updater r5 r r selfdate notify_typerD templates r* to_text_templatez EmailTemplate.to_text_templateU s
%%''& ''11
5<>X>^_ _ _ OO 1 =
4; ; ; OO"&"5"A#'#7#C
r) c | }| j || || | d}|t j t j fv r!| d| j j i |t j k r,| | j
j | j j d |S )zK
Convert to html template keys
see wmt_notify.html
)rU rV LOGOrW rX ERROR_HTML_REPORT)SUMMARY_HTML_REPORTDURATION_HTML_REPORT) rB r< rS r r r& r% r\ r r6 r r r] s r* to_html_templatez EmailTemplate.to_html_templatel s
%%''&NN$$ ''11
5<>X>^_ _ _ OO#T%6%B
4; ; ; OO'+':'F(,(<(H
r) )r" r# r$ r1 r2 r r4 r r r< r> staticmethodrB r rS rb rh r( r) r* r8 r8 8 s NNN"&NK&&& $L+$$$#'O['''&K&&&FC+ + \+ 2 2 \2
\ . r) r8 c e Zd ZU eed< eed< eed< eed< edefd Zed Z d Z
d Z ed
Z de
fd Zd Zdefd
ZdefdZdefdZdS )Notifier target_email
from_emailr notification_typereturnc | j t j k r7t j t j t
d z
d S | j t j k r&t j t j d S | j t j k rt j
t sdS t j
t j
t z
}t |dz }|dk rt! | dS t |d z }t! | d
S t# d | j ) N )daysz%Y-%m-%dz%Y-%m-%d %H:%M:%S iQ r z day(s) agoi z hour(s) agounexpected notifier type: )rn r r& r strftimenowr r% r' rH rI existsr
timegetmtimeintr1 NotImplementedError)r^ secondsrr hourss r* periodzNotifier.period s& !%?%F F F$X\^^iQ6G6G6G%GTT T
#'A'G
G
G$X\^^5HII I
#'A'J
J
J7>>),,
rikkBG$4$4Y$?$??Gw9-..Daxx d))0000GtO,, e**2222%&[4CY&[&[\\ \r) c T t t d d S )Nw)rK r closer( r) r* create_farewell_letter_markerz&Notifier.create_farewell_letter_marker s%
3S99??AAAAAr) c t }| }| | }| |j |j g| | j t j k r| d S d S N)
r _generate_template_data_generate_final_messagesendmailr: r9 rn r r' r )r^ mail_managerra messages r* notifyzNotifier.notify s !|| //11..x88 h 083G2H'RRR !%?%H H H . . 0 0 0 0 0 I Hr) c j | j t j k rJt j t d t j t d fS | j t j k rJt j t d t j t d fS | j t j k rJt j t d t j t d fS t d| j )Nzwmt_notify.txtzwmt_notify.htmlz
wmt_alert.txtzwmt_alert.htmlzwmt_farewell_solo.txtzwmt_farewell_solo.htmlrt )
rn r r& rH rI rJ r r% r' r{ )r^ s r* _get_templateszNotifier._get_templates s !%?%F F F7<< 13CDD7<< 13DEEF
F
#'A'G
G
G7<< 1?CC7<< 13CDDE
E
#'A'J
J
J7<< 13JKK7<< 13KLLM
M &&[4CY&[&[\\ \r) c t | S r )HEADERS_REPORT_MAPPING)sections r* _get_table_headerszNotifier._get_table_headers s
%g..r) c Z d| j }| d| }| d| }|r| n|}|r#| ddt t n|}t |t
j | dd S ) NzNo domains for texthtmlT)formatborderhrulesvrulesztext-align: centerztext-align: left)r5 r6 ) r~
_get_table
get_stringget_html_stringr
r4 r unescapereplace)r^ r no_domains_message
text_table
html_tabletext_final_strhtml_final_strs r* _generate_tableszNotifier._generate_tables s V..000DV DNf334;?;>;> 4 @ @ @ Tf &
n&<&<=QSe&f&fgg
r) c | j | }|sd S t | | }d|_ t |t
s|g}|D ]*}| | | +|S )Nl)r r r align
isinstancelistadd_row to_template)r^ alternativer
table_datatablerows r* r zNotifier._get_table s [)
4D 3 3G < <== *d++ &$ J 8 8C MM#//+66 7 7 7 7 r) c p | j | j d}| j t j t j fv r*| d| d i | j t j k r>| | d | d d t di |S )N)r9 r: r r r )r r r( ) rl rm rn r r& r% r\ r r8 )r^ ra s r* r z Notifier._generate_template_data s ,
!&@&GIcIi%j j j OO 5 5n E E
!%?%F F F OO"&"7"78H"I"I#'#8#89J#K#K
((x(((r) c J | \ }}t j | r t j | st d| d| d | j }d| j d| dt }| j t j k r d| j
d| }| j t j k r
d | j }t j
|| || j |
\ }}t j
|| || j |
\ }}n3# t j j j $ r} t d | d } ~ ww xY wt) ||| S )
Nz unable to find email templates: z, z. Try to reinstall packagezWeb Monitoring Tool z for z
from server [z] z1[EXPIRED CloudLinux LICENSE] Web Monitoring Tool )
templ_datar- z cannot generate email, reason: )r- r. r/ )r rH rI rw r r~ rn r r r% upperr' r ClEmailgenerate_mail_jinja2rb rh jinja2
exceptions
TemplateErrorr, )
r^ r
text_template
html_templater~ r- text_body_ html_bodyes
r* _generate_emailzNotifier._generate_email s '+':':'<'<$
}w~~m,, @BGNN=4Q4Q @# %?} %? %?Xe %? %? %? @ @
@ K[FoT-Coo&oo_k_m_mooG%)C)IIIId 4 : : < <IIII%)C)LLLfdNdff!( !E!E &*A*A&$J`*a*a "F "! "! GY #??? &*A*A&$J`*a*a @ ! !LAyy ~(6 K K K#$Ia$I$IJJ J K W)SSSSs -C2E F9F F
template_datac | | }|j dd d }|j dd d }t d }| t |dd | t |dd t |j d |d< |j
|d< |j |d< |S ) NrG xmlcharrefreplacer plainr SubjectFromTo) r r. encoderO r/ r attachr r r- r: r9 )r^ r emailr r r s r* r z Notifier._generate_final_message s $$]33%,,W6IJJQQRYZZ %,,W6IJJQQRYZZ
..x 7G<<===x 67;;<<< #EM7 ; ; B B D D '1%1
r) N)r" r# r$ r1 r2 dictpropertyr~ ri r r r r r4 r r r8 r r r r( r) r* rk rk sD OOO LLL
] ] ] ] X]$ B B \B1 1 1 ] ] ] / / \/
;
) ) ) ) ) Tm T T T T4 ] r) rk )$r rH rL rx dataclassesr r r email.headerr email.mime.multipartr email.mime.textr prettytabler r
wmt.common.constr r r
wmt.common.exceptionsr clcommonr clcommon.mail_helperr clcommon.lib.networkr r r r r, r4 r8 rk r( r) r* r s\
! ! ! ! ! ! ( ( ( ( ( ( ( ( . . . . . . $ $ $ $ $ $ ( ( ( ( ( ( ( ( a a a a a a a a a a 3 3 3 3 3 3 + + + + + + : : : : : : : :, , ,
$ $ $ +-
I I I I I I I IX F F F F F F F F F Fr)