Python-Projekte/mailserver/imap.py

from imap_tools import MailBox, AND
import re
import json
import os
from datetime import datetime, date


whitelist = {}


def cleanup_folder():
    date_now = datetime.now()
    date_criteria = date(date_now.year, date_now.month - 1, 1)
    msg_limit = 100

    with MailBox('mail.global-cube.com').login('archiv', 'Gc01am64!') as mb:
        folder_list = [f.name for f in mb.folder.list() if 'Archive.' in f.name]

        for folder in folder_list:
            msg_count = mb.folder.status(folder)['MESSAGES']
            if msg_count < msg_limit:
                continue
            mb.folder.set(folder)
            messages = mb.fetch(criteria=AND(date_lt=date_criteria), mark_seen=True, headers_only=True)
            uids = [msg.uid for msg in messages]
            mb.delete(uids[0:(msg_count - msg_limit)])


def move_mails():
    with MailBox('mail.global-cube.com').login('archiv', 'Gc01am64!') as mb:
        messages = mb.fetch(criteria=AND(from_="@global-cube.com"), mark_seen=True, bulk=True, limit=1000, headers_only=True)

        for msg in messages:
            match = re.findall(r'\+(.*)@', msg.from_)
            if not match:
                print(msg.from_, msg.to, msg.subject)
            else:
                domain = '@' + match[0]

                subfolder = 'Archive.' + domain.replace('@', '').replace('.', '_')
                if not mb.folder.exists(subfolder):
                    mb.folder.create(subfolder)

                if is_valid_message(msg, domain):
                    mb.move(msg.uid, subfolder)
                else:
                    print(domain, ', '.join(msg.to), msg.subject)
                # print(msg.text)
                # print([att.filename for att in msg.attachments if att.filename.endswith('.pdf')])


def is_valid_message(msg, domain):
    for to in msg.to:
        if domain in to or '@global-cube.de' in to or msg.from_ == to:
            continue
        if domain not in whitelist:
            return False
        valid_domain = [entry in to for entry in whitelist[domain]]
        if not any(valid_domain):
            return False
    return True


if __name__ == '__main__':
    whitelist = json.load(open(os.path.dirname(__file__) + '/whitelist.json', 'r'))
    move_mails()
    cleanup_folder()