mirror of https://github.com/langgenius/dify.git
59 lines
2.5 KiB
Python
59 lines
2.5 KiB
Python
import logging
|
|
import time
|
|
|
|
import click
|
|
from celery import shared_task # type: ignore
|
|
from flask import render_template
|
|
|
|
from extensions.ext_mail import mail
|
|
from services.feature_service import FeatureService
|
|
|
|
|
|
@shared_task(queue="mail")
|
|
def send_reset_password_mail_task(language: str, to: str, code: str):
|
|
"""
|
|
Async Send reset password mail
|
|
:param language: Language in which the email should be sent (e.g., 'en', 'zh')
|
|
:param to: Recipient email address
|
|
:param code: Reset password code
|
|
"""
|
|
if not mail.is_inited():
|
|
return
|
|
|
|
logging.info(click.style("Start password reset mail to {}".format(to), fg="green"))
|
|
start_at = time.perf_counter()
|
|
|
|
# send reset password mail using different languages
|
|
try:
|
|
if language == "zh-Hans":
|
|
template = "reset_password_mail_template_zh-CN.html"
|
|
system_features = FeatureService.get_system_features()
|
|
if system_features.branding.enabled:
|
|
application_title = system_features.branding.application_title
|
|
template = "without-brand/reset_password_mail_template_zh-CN.html"
|
|
html_content = render_template(template, to=to, code=code, application_title=application_title)
|
|
mail.send(to=to, subject=f"设置您的 {application_title} 密码", html=html_content)
|
|
else:
|
|
html_content = render_template(template, to=to, code=code)
|
|
mail.send(to=to, subject="设置您的 Dify 密码", html=html_content)
|
|
else:
|
|
template = "reset_password_mail_template_en-US.html"
|
|
system_features = FeatureService.get_system_features()
|
|
if system_features.branding.enabled:
|
|
application_title = system_features.branding.application_title
|
|
template = "without-brand/reset_password_mail_template_en-US.html"
|
|
html_content = render_template(template, to=to, code=code, application_title=application_title)
|
|
mail.send(to=to, subject=f"Set Your {application_title} Password", html=html_content)
|
|
else:
|
|
html_content = render_template(template, to=to, code=code)
|
|
mail.send(to=to, subject="Set Your Dify Password", html=html_content)
|
|
|
|
end_at = time.perf_counter()
|
|
logging.info(
|
|
click.style(
|
|
"Send password reset mail to {} succeeded: latency: {}".format(to, end_at - start_at), fg="green"
|
|
)
|
|
)
|
|
except Exception:
|
|
logging.exception("Send password reset mail to {} failed".format(to))
|