mail.py

import os
import requests

from classes.mail_wrapper import MailWrapper
from classes.imap import ImapGmailWrapper

from config import mail, app_password, url_telegram, chat_id

def check_gmail():

    imap = ImapGmailWrapper(mail, app_password)
    imap.select_mail_box('inbox')
    possible_new_email_id = int(imap.get_last_email_id('UNSEEN'))

    if not os.path.exists('recent_mail.txt'):
        save_latest_email_id(possible_new_email_id)
        return

    file_reader = open('recent_mail.txt', 'r')
    if file_reader.readable():
        current_mail_id = int(file_reader.read())

        if possible_new_email_id > current_mail_id:
            latest_email_id = possible_new_email_id
            number_of_new_mails = latest_email_id - current_mail_id

            file_reader.close()

            save_latest_email_id(latest_email_id)

            raw_email = imap.fetchs([latest_email_id])

            mail_parser = MailWrapper(raw_email)
            sender = mail_parser.get_sender()

            subject = mail_parser.get_subject()

            if number_of_new_mails > 1:
                message = 'Shalom Adonai, irmãos. {} novos emails, deem uma olhada.'.format(number_of_new_mails) 
            else:
                message = 'Shalom Adonai, irmãos. Novo email, deem uma olhada.\n\nTítulo: {}\n\nRemetente: {}'.format(subject, sender) 

            send_message(message)

def save_latest_email_id(latest_email_id):
    file_writer = open('recent_mail.txt', 'w')
    file_writer.write(str(latest_email_id))
    file_writer.close()

def send_message(message):
    url = url_telegram + 'sendMessage?chat_id=' + chat_id + '&text=' + message
    requests.post(url)

check_gmail()