diff --git a/api/services/feature_service.py b/api/services/feature_service.py index e2298cc573..1a3a4026ad 100644 --- a/api/services/feature_service.py +++ b/api/services/feature_service.py @@ -38,6 +38,7 @@ class LicenseModel(BaseModel): class BrandingModel(BaseModel): enabled: bool = False + application_title: str = "" login_page_logo: str = "" workspace_logo: str = "" favicon: str = "" diff --git a/api/tasks/mail_email_code_login.py b/api/tasks/mail_email_code_login.py index ac4b8f054f..e5d433b7e4 100644 --- a/api/tasks/mail_email_code_login.py +++ b/api/tasks/mail_email_code_login.py @@ -7,6 +7,7 @@ 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") @@ -28,14 +29,16 @@ def send_email_code_login_mail_task(language: str, to: str, code: str): if language == "zh-Hans": template = "email_code_login_mail_template_zh-CN.html" if dify_config.ENTERPRISE_ENABLED: + application_title = FeatureService.get_system_features().get("application_title", "Dify") template = "without-brand/email_code_login_mail_template_zh-CN_enterprise.html" - html_content = render_template(template, to=to, code=code) + html_content = render_template(template, to=to, code=code, application_title=application_title) mail.send(to=to, subject="邮箱验证码", html=html_content) else: template = "email_code_login_mail_template_en-US.html" if dify_config.ENTERPRISE_ENABLED: + application_title = FeatureService.get_system_features().get("application_title", "Dify") template = "without-brand/email_code_login_mail_template_en-US_enterprise.html" - html_content = render_template(template, to=to, code=code) + html_content = render_template(template, to=to, code=code, application_title=application_title) mail.send(to=to, subject="Email Code", html=html_content) end_at = time.perf_counter() diff --git a/api/tasks/mail_invite_member_task.py b/api/tasks/mail_invite_member_task.py index b4fcf7caef..33ec128f2f 100644 --- a/api/tasks/mail_invite_member_task.py +++ b/api/tasks/mail_invite_member_task.py @@ -7,6 +7,7 @@ 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") @@ -35,6 +36,7 @@ def send_invite_member_mail_task(language: str, to: str, token: str, inviter_nam if language == "zh-Hans": template = "invite_member_mail_template_zh-CN.html" if dify_config.ENTERPRISE_ENABLED: + application_title = FeatureService.get_system_features().get("application_title", "Dify") template = "without-brand/invite_member_mail_template_zh-CN.html" html_content = render_template( template, @@ -42,11 +44,13 @@ def send_invite_member_mail_task(language: str, to: str, token: str, inviter_nam inviter_name=inviter_name, workspace_name=workspace_name, url=url, + application_title=application_title, ) mail.send(to=to, subject="立即加入 Dify 工作空间", html=html_content) else: template = "invite_member_mail_template_en-US.html" if dify_config.ENTERPRISE_ENABLED: + application_title = FeatureService.get_system_features().get("application_title", "Dify") template = "without-brand/invite_member_mail_template_en-US.html" html_content = render_template( template, @@ -54,6 +58,7 @@ def send_invite_member_mail_task(language: str, to: str, token: str, inviter_nam inviter_name=inviter_name, workspace_name=workspace_name, url=url, + application_title=application_title, ) mail.send(to=to, subject="Join Dify Workspace Now", html=html_content) diff --git a/api/tasks/mail_reset_password_task.py b/api/tasks/mail_reset_password_task.py index e434bc30c0..1c260209d6 100644 --- a/api/tasks/mail_reset_password_task.py +++ b/api/tasks/mail_reset_password_task.py @@ -7,6 +7,7 @@ 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") @@ -28,14 +29,16 @@ def send_reset_password_mail_task(language: str, to: str, code: str): if language == "zh-Hans": template = "reset_password_mail_template_zh-CN.html" if dify_config.ENTERPRISE_ENABLED: + application_title = FeatureService.get_system_features().get("application_title", "Dify") template = "without-brand/reset_password_mail_template_zh-CN.html" - html_content = render_template(template, to=to, code=code) + html_content = render_template(template, to=to, code=code, application_title=application_title) mail.send(to=to, subject="设置您的 Dify 密码", html=html_content) else: template = "reset_password_mail_template_en-US.html" if dify_config.ENTERPRISE_ENABLED: + application_title = FeatureService.get_system_features().get("application_title", "Dify") template = "without-brand/reset_password_mail_template_en-US.html" - html_content = render_template(template, to=to, code=code) + html_content = render_template(template, to=to, code=code, application_title=application_title) mail.send(to=to, subject="Set Your Dify Password", html=html_content) end_at = time.perf_counter()