mailalert/mail_alert.py

'''Created Sep 28, 2021 Levi'''

import configparser
import sys
import traceback
import ssl
from time import sleep
from collections import deque
from threading import Event
from imaplib import IMAP4

import telegram
import keyring
from bs4 import BeautifulSoup
from telegram.error import BadRequest
from imap_tools import MailBox, AND, OR

# -> for python >=3.10, the dh key too small problem
context = ssl.SSLContext(ssl.PROTOCOL_TLSv1_2)
context.set_ciphers('DEFAULT@SECLEVEL=1')

config = configparser.ConfigParser()
config.read('bot')
bot = telegram.Bot(token=config['AUTH']['token'])
bot2 = telegram.Bot(token=config['AUTH']['bank'])
mailbox = MailBox(host=config['email']['host'], ssl_context=context)
mailbox2 = MailBox(host=config['email']['host'], ssl_context=context)
mailbox3 = MailBox(host=config['email']['host'], ssl_context=context)

mailbox.login(config['email']['user'], keyring.get_password('yagmail', 'levente.marton@mzk.ro'), initial_folder='Inbox')
mailbox2.login(config['email']['user'], keyring.get_password('yagmail', 'levente.marton@mzk.ro'), initial_folder='bank_events')
mailbox3.login(config['email']['user'], keyring.get_password('yagmail', 'levente.marton@mzk.ro'), initial_folder='API')


def pretty_exc():
    exc_type, _exc_obj, exc_tb = sys.exc_info()
    # fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
    for tb in list(traceback.format_exception(exc_type, _exc_obj, exc_tb)):
            print(tb)

def mask(msg, to_):
    return any([to_ == msg.from_,
                to_ in msg.to,
                to_ in msg.cc,
                to_ in msg.bcc])

def mask2(msg, to_):
    return any([to_ == msg.from_,
                to_ in msg.to,
                to_ in msg.cc,
                to_ in msg.bcc,
                to_ in msg.subject])


def mail_alert(event=None):
    '''mail alert Inbox'''
    container = deque(maxlen=500)
    # mailbox.login(config['email']['user'], keyring.get_password('yagmail', 'levente.marton@mzk.ro'), initial_folder='Inbox')
    while not event.is_set():
        try:
            sleep(15)
            msgs = [msgs for msgs in mailbox.fetch(AND(seen=False), mark_seen=False)]
            for msg in msgs:
                chk = mask(msg, 'levente.marton@mzk.ro')
                if chk:
                    msg_text = msg.text
                    if 'end of day extras/intraday' in msg.subject:
                        msg_text = msg.text.replace('<br>', '\n')
                    if msg.uid not in container:
                        try:
                            if len(msg.text) != 0:
                                bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\n{msg_text}')
                            else:
                                bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\n{msg.html}')
                            container.appendleft(msg.uid)
                            print('message appended', len(container))
                        except BadRequest:
                            try:
                                if len(msg.text) != 0:
                                    bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\n{msg_text[:len(msg_text) // 5]}')
                                else:
                                    bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\n{msg.html[:len(msg.html) // 5]}')
                                container.appendleft(msg.uid)
                                print('message appended', len(container))
                            except BadRequest:
                                bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\nMessage is too long')
                                container.appendleft(msg.uid)
                                print('message appended', len(container), '!! last message was to long !!')
                        except Exception:
                            try:
                                mailbox.logout()
                                pretty_exc()
                            except IMAP4.abort:
                                pass
                            # break
        except Exception:
            try:
                mailbox.logout()
            except IMAP4.abort:
                pass
            # break
        # finally:
        #     mailbox.logout()

def mail_alert2(event=None):
    container = deque(maxlen=500)
    # mailbox2.login(config['email']['user'], keyring.get_password('yagmail', 'levente.marton@mzk.ro'), initial_folder='bank_events')
    while not event.is_set():
        try:
            sleep(15)
            msgs = [msgs for msgs in mailbox2.fetch(AND(seen=False), mark_seen=True)]
            for msg in msgs:
                chk = mask2(msg, 'account')
                if chk:
                    if msg.uid not in container:
                        soup = BeautifulSoup(msg.html, 'html.parser')
                        msg_text = soup.prettify().replace('<br/>', '').replace('<html>', '').replace('<head>', '').replace('</head>', '').replace('<body>', '').replace('<div>', '').replace('</div>', '').replace('</body>', '').replace('</html>', '')
                        try:
                            if len(msg.text) != 0:
                                bot2.send_message(chat_id=config['AUTH']['chatid'], text=msg_text)
                            else:
                                bot2.send_message(chat_id=config['AUTH']['chatid'], text=msg_text)
                            container.appendleft(msg.uid)
                            print('message appended', len(container))
                        except BadRequest:
                            try:
                                if len(msg.text) != 0:
                                    bot2.send_message(chat_id=config['AUTH']['chatid'], text=msg_text // 5)
                                else:
                                    bot2.send_message(chat_id=config['AUTH']['chatid'], text=msg_text // 5)
                                container.appendleft(msg.uid)
                                print('message appended', len(container))
                            except BadRequest:
                                bot2.send_message(chat_id=config['AUTH']['chatid'], text=f'Message is too long')
                                container.appendleft(msg.uid)
                                print('message appended', len(container), '!! last message was to long !!')
                        except Exception:
                            try:
                                mailbox2.logout()
                                pretty_exc()
                            except IMAP4.abort:
                                pass
                            # break
        except Exception:
            # pretty_exc()
            try:
                mailbox2.logout()
            except IMAP4.abort:
                pass
            # break
        # finally:
        #     mailbox2.logout()

def mail_alert3(event=None):
    '''mail alert Inbox'''
    container = deque(maxlen=500)
    # mailbox3.login(config['email']['user'], keyring.get_password('yagmail', 'levente.marton@mzk.ro'), initial_folder='API')
    while not event.is_set():
        try:
            sleep(15)
            msgs = [msgs for msgs in mailbox3.fetch(AND(seen=False), mark_seen=True)]
            for msg in msgs:
                chk = mask(msg, 'Ciel_notifier@mzk.ro')
                if chk:
                    msg_text = msg.text
                    # if 'end of day extras/intraday' in msg.subject:
                    #     msg_text = msg.text.replace('<br>', '\n')
                    if msg.uid not in container:
                        try:
                            if len(msg.text) != 0:
                                bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\n{msg_text}')
                            else:
                                bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\n{msg.html}')
                            container.appendleft(msg.uid)
                            print('message appended', len(container))
                        except BadRequest:
                            try:
                                if len(msg.text) != 0:
                                    bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\n{msg_text[:len(msg_text) // 5]}')
                                else:
                                    bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\n{msg.html[:len(msg.html) // 5]}')
                                container.appendleft(msg.uid)
                                print('message appended', len(container))
                            except BadRequest:
                                bot.send_message(chat_id=config['AUTH']['chatid'], text=f'{msg.from_}\nsubj: {msg.subject}\n\nMessage is too long')
                                container.appendleft(msg.uid)
                                print('message appended', len(container), '!! last message was to long !!')
                        except Exception:
                            try:
                                mailbox3.logout()
                                pretty_exc()
                            except IMAP4.abort:
                                pass
                            # break
        except Exception:
            try:
                mailbox3.logout()
            except IMAP4.abort:
                pass
            # break
        # finally:
        #     mailbox3.logout()


if __name__ == '__main__':
    try:
        event = Event()
        mail_alert2(event=event)
    except KeyboardInterrupt:
        event.set()