mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-05-24 05:28:43 +08:00
d186daa131
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: Hash Brown <hi@xzd.me> Co-authored-by: crazywoola <427733928@qq.com> Co-authored-by: GareArc <chen4851@purdue.edu> Co-authored-by: Byron.wang <byron@dify.ai> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: -LAN- <laipz8200@outlook.com> Co-authored-by: Garfield Dai <dai.hai@foxmail.com> Co-authored-by: KVOJJJin <jzongcode@gmail.com> Co-authored-by: Alexi.F <654973939@qq.com> Co-authored-by: Xiyuan Chen <52963600+GareArc@users.noreply.github.com> Co-authored-by: kautsar_masuara <61046989+izon-masuara@users.noreply.github.com> Co-authored-by: achmad-kautsar <achmad.kautsar@insignia.co.id> Co-authored-by: Xin Zhang <sjhpzx@gmail.com> Co-authored-by: kelvintsim <83445753+kelvintsim@users.noreply.github.com> Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Zixuan Cheng <61724187+Theysua@users.noreply.github.com>
85 lines
3.3 KiB
Python
85 lines
3.3 KiB
Python
import logging
|
|
import time
|
|
|
|
import click
|
|
from celery import shared_task # type: ignore
|
|
from flask import render_template
|
|
|
|
from configs import dify_config
|
|
from extensions.ext_mail import mail
|
|
from services.feature_service import FeatureService
|
|
|
|
|
|
@shared_task(queue="mail")
|
|
def send_invite_member_mail_task(language: str, to: str, token: str, inviter_name: str, workspace_name: str):
|
|
"""
|
|
Async Send invite member mail
|
|
:param language
|
|
:param to
|
|
:param token
|
|
:param inviter_name
|
|
:param workspace_name
|
|
|
|
Usage: send_invite_member_mail_task.delay(language, to, token, inviter_name, workspace_name)
|
|
"""
|
|
if not mail.is_inited():
|
|
return
|
|
|
|
logging.info(
|
|
click.style("Start send invite member mail to {} in workspace {}".format(to, workspace_name), fg="green")
|
|
)
|
|
start_at = time.perf_counter()
|
|
|
|
# send invite member mail using different languages
|
|
try:
|
|
url = f"{dify_config.CONSOLE_WEB_URL}/activate?token={token}"
|
|
if language == "zh-Hans":
|
|
template = "invite_member_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/invite_member_mail_template_zh-CN.html"
|
|
html_content = render_template(
|
|
template,
|
|
to=to,
|
|
inviter_name=inviter_name,
|
|
workspace_name=workspace_name,
|
|
url=url,
|
|
application_title=application_title,
|
|
)
|
|
mail.send(to=to, subject=f"立即加入 {application_title} 工作空间", html=html_content)
|
|
else:
|
|
html_content = render_template(
|
|
template, to=to, inviter_name=inviter_name, workspace_name=workspace_name, url=url
|
|
)
|
|
mail.send(to=to, subject="立即加入 Dify 工作空间", html=html_content)
|
|
else:
|
|
template = "invite_member_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/invite_member_mail_template_en-US.html"
|
|
html_content = render_template(
|
|
template,
|
|
to=to,
|
|
inviter_name=inviter_name,
|
|
workspace_name=workspace_name,
|
|
url=url,
|
|
application_title=application_title,
|
|
)
|
|
mail.send(to=to, subject=f"Join {application_title} Workspace Now", html=html_content)
|
|
else:
|
|
html_content = render_template(
|
|
template, to=to, inviter_name=inviter_name, workspace_name=workspace_name, url=url
|
|
)
|
|
mail.send(to=to, subject="Join Dify Workspace Now", html=html_content)
|
|
|
|
end_at = time.perf_counter()
|
|
logging.info(
|
|
click.style(
|
|
"Send invite member mail to {} succeeded: latency: {}".format(to, end_at - start_at), fg="green"
|
|
)
|
|
)
|
|
except Exception:
|
|
logging.exception("Send invite member mail to {} failed".format(to))
|