mirror of
https://github.com/langgenius/dify.git
synced 2025-07-31 05:18:40 +00:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import logging
|
|
import time
|
|
from collections.abc import Mapping
|
|
|
|
import click
|
|
from celery import shared_task # type: ignore
|
|
from flask import render_template_string
|
|
|
|
from extensions.ext_mail import mail
|
|
from libs.email_i18n import get_email_i18n_service
|
|
|
|
|
|
@shared_task(queue="mail")
|
|
def send_enterprise_email_task(to: list[str], subject: str, body: str, substitutions: Mapping[str, str]):
|
|
if not mail.is_inited():
|
|
return
|
|
|
|
logging.info(click.style("Start enterprise mail to {} with subject {}".format(to, subject), fg="green"))
|
|
start_at = time.perf_counter()
|
|
|
|
try:
|
|
html_content = render_template_string(body, **substitutions)
|
|
|
|
email_service = get_email_i18n_service()
|
|
email_service.send_raw_email(to=to, subject=subject, html_content=html_content)
|
|
|
|
end_at = time.perf_counter()
|
|
logging.info(
|
|
click.style("Send enterprise mail to {} succeeded: latency: {}".format(to, end_at - start_at), fg="green")
|
|
)
|
|
except Exception:
|
|
logging.exception("Send enterprise mail to {} failed".format(to))
|