PNG  IHDRxsBIT|d pHYs+tEXtSoftwarewww.inkscape.org<,tEXtComment File Manager

File Manager

Path: /opt/cloudlinux/venv/lib/python3.11/site-packages/wmt/common/

Viewing File: notification.py

#!/opt/cloudlinux/venv/bin/python3

import html
import os
import base64
import time
from dataclasses import dataclass
from datetime import timedelta, datetime

from email.header import Header
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

from prettytable import PrettyTable, ALL

from wmt.common.const import WMT_TEMPLATES_DIR, LICENSE_EXPIRED_FAREWELL_LETTER_MARKER, JWT_TOKEN
from wmt.common.exceptions import WmtEmailException
from clcommon import clemail
from clcommon.mail_helper import MailHelper
from clcommon.lib.network import get_hostname, get_ip_addr


HEADERS_REPORT_MAPPING = {
    'summary_report': ['All',
                       'Successful',
                       'Failed',
                       'Undone',
                       'Average time (ms)'],
    'error_report': ['URL',
                     'Errors',
                     'Error codes'],
    'duration_report': ['URL',
                        'Average time (ms)']
}


class SupportedNotificationTypes:
    ALERT = 'alert'
    REPORT = 'report'
    FAREWELL = 'farewell'


@dataclass
class Email:
    subject: str
    text_sample: str
    html_sample: str


@dataclass
class EmailReport:
    text_option: str
    html_option: str


@dataclass
class EmailTemplate:

    target_mail: str
    from_mail: str

    summary_report: EmailReport = None
    error_report: EmailReport = None
    duration_report: EmailReport = None
    target_name: str = 'Administrator'
    locale: str = 'en_US'

    @staticmethod
    def _get_hostname():
        return get_hostname() or 'UNKNOWN'

    @staticmethod
    def get_ip_addr(hostname):
        if not hostname:
            return 'UNKNOWN'
        return get_ip_addr(hostname) or 'UNKNOWN'

    @staticmethod
    def _get_logo():
        logo_path = os.path.join(WMT_TEMPLATES_DIR, 'logo.png')
        with open(logo_path, 'rb') as logo:
            logo_img_encoded = base64.b64encode(logo.read()).decode('utf-8')
        return logo_img_encoded

    def to_text_template(self, date, notify_type):
        """
        Convert to txt template keys
        see wmt_notify.txt
        """
        hostname = self._get_hostname()
        template = {
            'TONAME': self.target_name,
            'DATE': date,
            'HOSTNAME': hostname,
            'IP_ADDR': self.get_ip_addr(hostname)
        }
        if notify_type in [SupportedNotificationTypes.REPORT, SupportedNotificationTypes.ALERT]:
            template.update({
                'ERROR_REPORT': self.error_report.text_option
            })
        if notify_type == SupportedNotificationTypes.REPORT:
            template.update({
                'SUMMARY_REPORT': self.summary_report.text_option,
                'DURATION_REPORT': self.duration_report.text_option,
            })
        return template

    def to_html_template(self, date, notify_type):
        """
        Convert to html template keys
        see wmt_notify.html
        """
        hostname = self._get_hostname()
        template = {
            'TONAME': self.target_name,
            'DATE': date,
            'LOGO': self._get_logo(),
            'HOSTNAME': hostname,
            'IP_ADDR': self.get_ip_addr(hostname)
        }
        if notify_type in [SupportedNotificationTypes.REPORT, SupportedNotificationTypes.ALERT]:
            template.update({
                'ERROR_HTML_REPORT': self.error_report.html_option
            })
        if notify_type == SupportedNotificationTypes.REPORT:
            template.update({
                'SUMMARY_HTML_REPORT': self.summary_report.html_option,
                'DURATION_HTML_REPORT': self.duration_report.html_option
            })
        return template


@dataclass
class Notifier:
    target_email: str
    from_email: str
    report: dict
    notification_type: str

    @property
    def period(self) -> str:
        if self.notification_type == SupportedNotificationTypes.REPORT:
            return datetime.strftime(datetime.now() - timedelta(days=1), '%Y-%m-%d')
        elif self.notification_type == SupportedNotificationTypes.ALERT:
            return datetime.strftime(datetime.now(), '%Y-%m-%d %H:%M:%S')
        elif self.notification_type == SupportedNotificationTypes.FAREWELL:
            if not os.path.exists(JWT_TOKEN):
                return ''
            seconds = time.time() - os.path.getmtime(JWT_TOKEN)
            days = int(seconds // (3600 * 24))
            if days > 0:
                return f'{str(days)} day(s) ago'
            else:
                hours = int(seconds // 3600)
                return f'{str(hours)} hour(s) ago'
        else:
            raise NotImplementedError(f'unexpected notifier type: {self.notification_type}')

    @staticmethod
    def create_farewell_letter_marker():
        open(LICENSE_EXPIRED_FAREWELL_LETTER_MARKER, 'w').close()

    def notify(self):
        mail_manager = MailHelper()
        template = self._generate_template_data()
        message = self._generate_final_message(template)
        mail_manager.sendmail(template.from_mail, [template.target_mail], message)
        if self.notification_type == SupportedNotificationTypes.FAREWELL:
            self.create_farewell_letter_marker()

    def _get_templates(self):
        if self.notification_type == SupportedNotificationTypes.REPORT:
            return os.path.join(WMT_TEMPLATES_DIR, 'wmt_notify.txt'), \
                   os.path.join(WMT_TEMPLATES_DIR, 'wmt_notify.html')
        elif self.notification_type == SupportedNotificationTypes.ALERT:
            return os.path.join(WMT_TEMPLATES_DIR, 'wmt_alert.txt'), \
                   os.path.join(WMT_TEMPLATES_DIR, 'wmt_alert.html')
        elif self.notification_type == SupportedNotificationTypes.FAREWELL:
            return os.path.join(WMT_TEMPLATES_DIR, 'wmt_farewell_solo.txt'), \
                   os.path.join(WMT_TEMPLATES_DIR, 'wmt_farewell_solo.html')
        else:
            raise NotImplementedError(f'unexpected notifier type: {self.notification_type}')

    @staticmethod
    def _get_table_headers(section):
        return HEADERS_REPORT_MAPPING[section]

    def _generate_tables(self, section) -> EmailReport:
        no_domains_message = f'No domains for {self.period}'
        text_table = self._get_table('text', section)
        html_table = self._get_table('html', section)

        text_final_str = text_table.get_string() if text_table else no_domains_message
        html_final_str = html_table.get_html_string(format=True,
                                                    border=True,
                                                    hrules=ALL,
                                                    vrules=ALL) if html_table else no_domains_message
        return EmailReport(
            text_option=text_final_str,
            html_option=html.unescape(html_final_str.replace('text-align: center', 'text-align: left'))
        )

    def _get_table(self, alternative, section):
        table_data = self.report[section]
        if not table_data:
            return None
        table = PrettyTable(self._get_table_headers(section))
        table.align = 'l'
        if not isinstance(table_data, list):
            table_data = [table_data]
        for row in table_data:
            table.add_row(row.to_template(alternative))
        return table

    def _generate_template_data(self) -> EmailTemplate:
        template = {
            'target_mail': self.target_email,
            'from_mail': self.from_email
        }
        if self.notification_type in [SupportedNotificationTypes.REPORT, SupportedNotificationTypes.ALERT]:
            template.update({
                'error_report': self._generate_tables('error_report'),
            })
        if self.notification_type == SupportedNotificationTypes.REPORT:
            template.update({
                'summary_report': self._generate_tables('summary_report'),
                'duration_report': self._generate_tables('duration_report')
            })
        return EmailTemplate(**template)

    def _generate_email(self, report: EmailTemplate):
        text_template, html_template = self._get_templates()
        if not os.path.exists(text_template) or not os.path.exists(html_template):
            raise WmtEmailException(f'unable to find email templates: {text_template}, {html_template}. '
                                    'Try to reinstall package')

        try:
            period = self.period
            subject = f'Web Monitoring Tool {self.notification_type} for {period} from server {get_hostname()}'
            if self.notification_type == SupportedNotificationTypes.ALERT:
                subject = f'[{self.notification_type.upper()}] {subject}'
            if self.notification_type == SupportedNotificationTypes.FAREWELL:
                subject = f'[EXPIRED CloudLinux LICENSE] Web Monitoring Tool {self.notification_type}'
            subject, text_body = clemail.ClEmail.generate_mail_jinja2(
                text_template, templ_data=report.to_text_template(period, self.notification_type),
                subject=subject)

            _, html_body = clemail.ClEmail.generate_mail_jinja2(
                html_template, templ_data=report.to_html_template(period, self.notification_type),
                subject=subject)

        except clemail.jinja2.exceptions.TemplateError as e:
            raise WmtEmailException(f'cannot generate email, reason: {e}')

        return Email(subject=subject, text_sample=text_body, html_sample=html_body)

    def _generate_final_message(self, template_data: EmailTemplate):
        email = self._generate_email(template_data)
        text_body = email.text_sample.encode('utf-8', 'xmlcharrefreplace').decode('utf-8')
        html_body = email.html_sample.encode('utf-8', 'xmlcharrefreplace').decode('utf-8')
        message = MIMEMultipart('alternative')
        message.attach(MIMEText(text_body, 'plain', 'utf-8'))
        message.attach(MIMEText(html_body, 'html', 'utf-8'))

        message['Subject'] = Header(email.subject, 'utf-8').encode()
        message['From'] = template_data.from_mail
        message['To'] = template_data.target_mail
        return message
b IDATxytVսϓ22 A@IR :hCiZ[v*E:WũZA ^dQeQ @ !jZ'>gsV仿$|?g)&x-EIENT ;@xT.i%-X}SvS5.r/UHz^_$-W"w)Ɗ/@Z &IoX P$K}JzX:;` &, ŋui,e6mX ԵrKb1ԗ)DADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADADA݀!I*]R;I2$eZ#ORZSrr6mteffu*((Pu'v{DIߔ4^pIm'77WEEE;vƎ4-$]'RI{\I&G :IHJ DWBB=\WR޽m o$K(V9ABB.}jѢv`^?IOȅ} ڶmG}T#FJ`56$-ھ}FI&v;0(h;Б38CӧOWf!;A i:F_m9s&|q%=#wZprrrla A &P\\СC[A#! {olF} `E2}MK/vV)i{4BffV\|ۭX`b@kɶ@%i$K z5zhmX[IXZ` 'b%$r5M4º/l ԃߖxhʔ)[@=} K6IM}^5k㏷݆z ΗÿO:gdGBmyT/@+Vɶ纽z񕏵l.y޴it뭷zV0[Y^>Wsqs}\/@$(T7f.InݺiR$푔n.~?H))\ZRW'Mo~v Ov6oԃxz! S,&xm/yɞԟ?'uaSѽb,8GלKboi&3t7Y,)JJ c[nzӳdE&KsZLӄ I?@&%ӟ۶mSMMњ0iؐSZ,|J+N ~,0A0!5%Q-YQQa3}$_vVrf9f?S8`zDADADADADADADADADAdqP,تmMmg1V?rSI꒟]u|l RCyEf٢9 jURbztѰ!m5~tGj2DhG*{H9)꒟ר3:(+3\?/;TUݭʴ~S6lڧUJ*i$d(#=Yݺd{,p|3B))q:vN0Y.jkק6;SɶVzHJJЀ-utѹսk>QUU\޲~]fFnK?&ߡ5b=z9)^|u_k-[y%ZNU6 7Mi:]ۦtk[n X(e6Bb."8cۭ|~teuuw|ήI-5"~Uk;ZicEmN/:]M> cQ^uiƞ??Ңpc#TUU3UakNwA`:Y_V-8.KKfRitv޲* 9S6ֿj,ՃNOMߤ]z^fOh|<>@Å5 _/Iu?{SY4hK/2]4%it5q]GGe2%iR| W&f*^]??vq[LgE_3f}Fxu~}qd-ږFxu~I N>\;͗O֊:̗WJ@BhW=y|GgwܷH_NY?)Tdi'?խwhlmQi !SUUsw4kӺe4rfxu-[nHtMFj}H_u~w>)oV}(T'ebʒv3_[+vn@Ȭ\S}ot}w=kHFnxg S 0eޢm~l}uqZfFoZuuEg `zt~? b;t%>WTkķh[2eG8LIWx,^\thrl^Ϊ{=dž<}qV@ ⠨Wy^LF_>0UkDuʫuCs$)Iv:IK;6ֲ4{^6եm+l3>݆uM 9u?>Zc }g~qhKwڭeFMM~pМuqǿz6Tb@8@Y|jx](^]gf}M"tG -w.@vOqh~/HII`S[l.6nØXL9vUcOoB\xoǤ'T&IǍQw_wpv[kmO{w~>#=P1Pɞa-we:iǏlHo׈꒟f9SzH?+shk%Fs:qVhqY`jvO'ρ?PyX3lх]˾uV{ݞ]1,MzYNW~̈́ joYn}ȚF߾׮mS]F z+EDxm/d{F{-W-4wY듏:??_gPf ^3ecg ҵs8R2מz@TANGj)}CNi/R~}c:5{!ZHӋӾ6}T]G]7W6^n 9*,YqOZj:P?Q DFL|?-^.Ɵ7}fFh׶xe2Pscz1&5\cn[=Vn[ĶE鎀uˌd3GII k;lNmشOuuRVfBE]ۣeӶu :X-[(er4~LHi6:Ѻ@ԅrST0trk%$Č0ez" *z"T/X9|8.C5Feg}CQ%͞ˣJvL/?j^h&9xF`њZ(&yF&Iݻfg#W;3^{Wo^4'vV[[K';+mӍִ]AC@W?1^{එyh +^]fm~iԵ]AB@WTk̏t uR?l.OIHiYyԶ]Aˀ7c:q}ힽaf6Z~қm(+sK4{^6}T*UUu]n.:kx{:2 _m=sAߤU@?Z-Vކеz왍Nэ{|5 pڶn b p-@sPg]0G7fy-M{GCF'%{4`=$-Ge\ eU:m+Zt'WjO!OAF@ik&t݆ϥ_ e}=]"Wz_.͜E3leWFih|t-wZۍ-uw=6YN{6|} |*={Ѽn.S.z1zjۻTH]흾 DuDvmvK.`V]yY~sI@t?/ϓ. m&["+P?MzovVЫG3-GRR[(!!\_,^%?v@ҵő m`Y)tem8GMx.))A]Y i`ViW`?^~!S#^+ѽGZj?Vģ0.))A꨷lzL*]OXrY`DBBLOj{-MH'ii-ϰ ok7^ )쭡b]UXSְmռY|5*cֽk0B7镹%ڽP#8nȎq}mJr23_>lE5$iwui+ H~F`IjƵ@q \ @#qG0".0" l`„.0! ,AQHN6qzkKJ#o;`Xv2>,tێJJ7Z/*A .@fفjMzkg @TvZH3Zxu6Ra'%O?/dQ5xYkU]Rֽkق@DaS^RSּ5|BeHNN͘p HvcYcC5:y #`οb;z2.!kr}gUWkyZn=f Pvsn3p~;4p˚=ē~NmI] ¾ 0lH[_L hsh_ғߤc_њec)g7VIZ5yrgk̞W#IjӪv>՞y睝M8[|]\շ8M6%|@PZڨI-m>=k='aiRo-x?>Q.}`Ȏ:Wsmu u > .@,&;+!!˱tﭧDQwRW\vF\~Q7>spYw$%A~;~}6¾ g&if_=j,v+UL1(tWake:@Ș>j$Gq2t7S?vL|]u/ .(0E6Mk6hiۺzښOrifޱxm/Gx> Lal%%~{lBsR4*}{0Z/tNIɚpV^#Lf:u@k#RSu =S^ZyuR/.@n&΃z~B=0eg뺆#,Þ[B/?H uUf7y Wy}Bwegל`Wh(||`l`.;Ws?V@"c:iɍL֯PGv6zctM̠':wuW;d=;EveD}9J@B(0iհ bvP1{\P&G7D޴Iy_$-Qjm~Yrr&]CDv%bh|Yzni_ˆR;kg}nJOIIwyuL}{ЌNj}:+3Y?:WJ/N+Rzd=hb;dj͒suݔ@NKMԄ jqzC5@y°hL m;*5ezᕏ=ep XL n?מ:r`۵tŤZ|1v`V뽧_csج'ߤ%oTuumk%%%h)uy]Nk[n 'b2 l.=͜E%gf$[c;s:V-͞WߤWh-j7]4=F-X]>ZLSi[Y*We;Zan(ӇW|e(HNNP5[= r4tP &0<pc#`vTNV GFqvTi*Tyam$ߏWyE*VJKMTfFw>'$-ؽ.Ho.8c"@DADADADADADADADADA~j*֘,N;Pi3599h=goضLgiJ5փy~}&Zd9p֚ e:|hL``b/d9p? fgg+%%hMgXosج, ΩOl0Zh=xdjLmhݻoO[g_l,8a]٭+ӧ0$I]c]:粹:Teꢢ"5a^Kgh,&= =՟^߶“ߢE ܹS J}I%:8 IDAT~,9/ʃPW'Mo}zNƍ쨓zPbNZ~^z=4mswg;5 Y~SVMRXUյڱRf?s:w ;6H:ºi5-maM&O3;1IKeamZh͛7+##v+c ~u~ca]GnF'ټL~PPPbn voC4R,ӟgg %hq}@#M4IÇ Oy^xMZx ) yOw@HkN˖-Sǎmb]X@n+i͖!++K3gd\$mt$^YfJ\8PRF)77Wא!Cl$i:@@_oG I{$# 8磌ŋ91A (Im7֭>}ߴJq7ޗt^ -[ԩSj*}%]&' -ɓ'ꫯVzzvB#;a 7@GxI{j޼ƌ.LÇWBB7`O"I$/@R @eee@۷>}0,ɒ2$53Xs|cS~rpTYYY} kHc %&k.], @ADADADADADADADADA@lT<%''*Lo^={رc5h %$+CnܸQ3fҥK}vUVVs9G R,_{xˇ3o߾;TTTd}馛]uuuG~iԩ@4bnvmvfϞ /Peeeq}}za I~,誫{UWW뮻}_~YƍSMMMYχ֝waw\ďcxꩧtEƍկ_?۷5@u?1kNׯWzz/wy>}zj3 k(ٺuq_Zvf̘:~ ABQ&r|!%KҥKgԞ={<_X-z !CyFUUz~ ABQIIIjݺW$UXXDٳZ~ ABQƍecW$<(~<RSSvZujjjԧOZQu@4 8m&&&jԩg$ď1h ͟?_{768@g =@`)))5o6m3)ѣƌJ;wҿUTT /KZR{~a=@0o<*狔iFɶ[ˎ;T]]OX@?K.ۈxN pppppppppppppppppPfl߾] ,{ァk۶mڿo5BTӦMӴiӴ|r DB2e|An!Dy'tkΝ[A $***t5' "!駟oaDnΝ:t֭[gDШQ06qD;@ x M6v(PiizmZ4ew"@̴ixf [~-Fٱc&IZ2|n!?$@{[HTɏ#@hȎI# _m(F /6Z3z'\r,r!;w2Z3j=~GY7"I$iI.p_"?pN`y DD?: _  Gÿab7J !Bx@0 Bo cG@`1C[@0G @`0C_u V1 aCX>W ` | `!<S `"<. `#c`?cAC4 ?c p#~@0?:08&_MQ1J h#?/`7;I  q 7a wQ A 1 Hp !#<8/#@1Ul7=S=K.4Z?E_$i@!1!E4?`P_  @Bă10#: "aU,xbFY1 [n|n #'vEH:`xb #vD4Y hi.i&EΖv#O H4IŶ}:Ikh @tZRF#(tXҙzZ ?I3l7q@õ|ۍ1,GpuY Ꮿ@hJv#xxk$ v#9 5 }_$c S#=+"K{F*m7`#%H:NRSp6I?sIՖ{Ap$I$I:QRv2$Z @UJ*$]<FO4IENDB`