toolkit_outlook.py

import win32com.client as win32
import sys
import logging

default_recipients = 'chdu@xx.com'


class Outlook():
    def __init__(self):
        app = 'Outlook'
        self.app = win32.gencache.EnsureDispatch("%s.Application" % app)
        self.outlookAPI = self.app.GetNamespace("MAPI")
        self.inbox = self.outlookAPI.GetDefaultFolder(
            win32.constants.olFolderInbox)
        self.deletedBox = self.outlookAPI.GetDefaultFolder(
            win32.constants.olFolderDeletedItems)

    def folderPathConvert(self, folderPath):
        ''' Input folder path seperated by '/'
            Return folder instance'''
        toFolderPathStr = 'self.inbox' + \
            ''.join(['''.Folders['{}']'''.format(i)
                     for i in folderPath.split('/') if i])
        return eval(toFolderPathStr)

    def send_mail(self, subj, body, recipients=None, attachment_path=None, mail_type='PlainText'):
        '''
        Attachment can be list or string
        mail_type: PlainText/html
        '''
        if recipients is None:
            recipients = ['chdu@merkleinc.com']

        print("Subject: " + subj)
        print("Body: " + body)
        print("Send to: " + str(recipients))

        logging.info("Subject: " + subj)
        logging.info("Body: " + body)
        logging.info("Send to: " + str(recipients))
        mail = self.app.CreateItem(win32.constants.olMailItem)
        for i in recipients:
            mail.Recipients.Add(i)

        if attachment_path is not None:
            if isinstance(attachment_path, str):
                mail.Attachments.Add(Source=attachment_path)
            elif isinstance(attachment_path, list):
                for i in attachment_path:
                    logging.info('Add {} as attachment'.format(i))
                    mail.Attachments.Add(Source=i)

        mail.Subject = subj
        if mail_type == 'html':
            mail.HTMLBody = body
        else:
            mail.HTMLBody = body
        try:
            mail.Send()
        except Exception as e:
            logging.error('Failed to send email!')
        else:
            print('Mail sent')
            logging.info('Mail sent')

    def fetch_inbox_folders(self):
        '''
        Return the list of outlook folders
        '''
        fldr_iterator = self.inbox.Folders
        return [i.Name for i in fldr_iterator]

    def fetch_folder(self, folderName='INBOX'):
        '''
        Fetch the subject of mail in a folder
        '''
        if folderName.upper() == 'INBOX':
            # Fetch inbox
            folder = self.outlookAPI.GetDefaultFolder(6)
        else:
            folder = self.folderPathConvert(folderName)

        # folder = self.inbox.Folders[folderName]
        messages = folder.Items
        return [message.Subject for message in messages]

    def fetch_mail(self, folderName='INBOX', subject='keywords'):
        '''
        Fetch the HTMLBody of mail in a folder
        '''
        if folderName.upper() == 'INBOX':
            # Fetch inbox
            folder = self.outlookAPI.GetDefaultFolder(6)
        else:
            folder = self.folderPathConvert(folderName)

        # folder = self.inbox.Folders[folderName]
        messages = folder.Items
        return [message.HTMLBody for message in messages if subject in message.Subject]

    def empty_folder(self, folderName):
        '''
        Put all mails in the folder into deleted box
        '''
        folder = self.folderPathConvert(folderName)
        messages = folder.Items
        deletedItem = ['init']
        while len(deletedItem):
            deletedItem = []
            for message in messages:
                logging.info('Delete {}'.format(message.Subject))
                deletedItem.append(message.Subject)
                try:
                    message.Delete()
                except Exception as e:
                    logging.error("Err, try again")
                    self.empty_folder(folderName)

    def empty_junkbox(self):
        '''
        Purge all mails in the deleted box
        '''
        deletedItem = ['init']
        while len(deletedItem):
            deletedItem = []
            for i in self.deletedBox.Items:
                deletedItem.append(i.Subject)
                i.Delete()
            logging.info('Deleted {} mails'.format(len(deletedItem)))

        logging.info("Purge done")


outlook = Outlook()


def outlook_send_mail(subj, body, recipients=None, attachment_path=None):
    outlook.send_mail(subj, body, recipients=recipients,
                      attachment_path=attachment_path)


if __name__ == '__main__':
    # folder = outlook.empty_folder('[ Notification ] Filebridge')
    clear_list = ['[ Notification ] Carters', '[ Notification ] iBase',
                  '[ Notification ] Bose', '[ Notification ] System', '[ Notification ] Filebridge', ]
    # folder = outlook.empty_folder('[ Notification ] Carters')

    # outlook_send_mail('23', '33')
    #