如何在 ThreadPool 执行函数中只发送一次电子邮件？

Question

这是我用来使 API 操作快 10 倍的函数：

def load_url(req, id, data, timeout):
    headers = {'Authorization': 'AT-API 111111222222333333344444445555555'}
    r = req.post("https://service.com/api/v1/compare", headers=headers, data=data, timeout=timeout)
    data = r.json()
    print id
    if data['error']:
        print data['error']
    else:
        c.execute("UPDATE offers SET valid = ? WHERE id = ?", ('valid' if data['data']['success'] else 'invalid', id))
        print data['data']['success']
        print data['data']['count']
    return r.json()


if __name__ == '__main__':
    ...
    with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
        future_to_url = {executor.submit(load_url, s, data['id'], data, 120): data for data in datas}
        for future in concurrent.futures.as_completed(future_to_url):
            url = future_to_url[future]
            try:
                data = future.result()

我们运行每 4 小时定期执行此脚本。在 load_url 中，我们打印并检查请求的状态，data['data']['count'] - 是一个月内剩余的调用量。我想在调用次数达到 5000 次或更少后发送电子邮件通知，但只能发送一次。如何实现只发一条消息，不发5条不发50条？我们正在使用 Sqlite3 来存储数据。

我们正在使用 Mailgun 发送电子邮件：

def send_simple_message(email_list):
    for email in email_list:
        response = requests.post(
            "https://api.mailgun.net/v3/newsletter.company.com",
            auth=("api", "key-1234567"),
            data={"from": "Mailgun Sandbox <postmaster@newsletter.mobupps.com>",
                "to": email,
                "subject": "Agent - we reached the limit by API",
                "html": "We reached the limit for agent" })

Answer 1

你可以使用锁机制来防止多个线程执行send_simple_message函数，并设置一个同步值来跟踪邮件是否已发送。

import threading
lock = threading.Lock()
has_been_sent = False

# then in your load_url function you could do something like
if condition on count:
    with lock:
        if not has_beend_sent:
            # send mail
            has_been_sent = True

如何在 ThreadPool 执行函数中只发送一次电子邮件？

How to send email message only once in ThreadPool executed function?

python

email

multithreading

threadpool

threadpoolexecutor