mirror of
https://github.com/langgenius/dify.git
synced 2025-12-19 14:12:29 +00:00
05fe92a541
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
from typing import Any
|
|
|
|
from flask_restx import Resource
|
|
from pydantic import BaseModel, Field
|
|
|
|
from controllers.common.schema import register_schema_model
|
|
from controllers.console.wraps import setup_required
|
|
from controllers.inner_api import inner_api_ns
|
|
from controllers.inner_api.wraps import billing_inner_api_only, enterprise_inner_api_only
|
|
from tasks.mail_inner_task import send_inner_email_task
|
|
|
|
|
|
class InnerMailPayload(BaseModel):
|
|
to: list[str] = Field(description="Recipient email addresses", min_length=1)
|
|
subject: str
|
|
body: str
|
|
substitutions: dict[str, Any] | None = None
|
|
|
|
|
|
register_schema_model(inner_api_ns, InnerMailPayload)
|
|
|
|
|
|
class BaseMail(Resource):
|
|
"""Shared logic for sending an inner email."""
|
|
|
|
@inner_api_ns.doc("send_inner_mail")
|
|
@inner_api_ns.doc(description="Send internal email")
|
|
@inner_api_ns.expect(inner_api_ns.models[InnerMailPayload.__name__])
|
|
def post(self):
|
|
args = InnerMailPayload.model_validate(inner_api_ns.payload or {})
|
|
send_inner_email_task.delay(
|
|
to=args.to,
|
|
subject=args.subject,
|
|
body=args.body,
|
|
substitutions=args.substitutions, # type: ignore
|
|
)
|
|
return {"message": "success"}, 200
|
|
|
|
|
|
@inner_api_ns.route("/enterprise/mail")
|
|
class EnterpriseMail(BaseMail):
|
|
method_decorators = [setup_required, enterprise_inner_api_only]
|
|
|
|
@inner_api_ns.doc("send_enterprise_mail")
|
|
@inner_api_ns.doc(description="Send internal email for enterprise features")
|
|
@inner_api_ns.expect(inner_api_ns.models[InnerMailPayload.__name__])
|
|
@inner_api_ns.doc(
|
|
responses={200: "Email sent successfully", 401: "Unauthorized - invalid API key", 404: "Service not available"}
|
|
)
|
|
def post(self):
|
|
"""Send internal email for enterprise features.
|
|
|
|
This endpoint allows sending internal emails for enterprise-specific
|
|
notifications and communications.
|
|
|
|
Returns:
|
|
dict: Success message with status code 200
|
|
"""
|
|
return super().post()
|
|
|
|
|
|
@inner_api_ns.route("/billing/mail")
|
|
class BillingMail(BaseMail):
|
|
method_decorators = [setup_required, billing_inner_api_only]
|
|
|
|
@inner_api_ns.doc("send_billing_mail")
|
|
@inner_api_ns.doc(description="Send internal email for billing notifications")
|
|
@inner_api_ns.expect(inner_api_ns.models[InnerMailPayload.__name__])
|
|
@inner_api_ns.doc(
|
|
responses={200: "Email sent successfully", 401: "Unauthorized - invalid API key", 404: "Service not available"}
|
|
)
|
|
def post(self):
|
|
"""Send internal email for billing notifications.
|
|
|
|
This endpoint allows sending internal emails for billing-related
|
|
notifications and alerts.
|
|
|
|
Returns:
|
|
dict: Success message with status code 200
|
|
"""
|
|
return super().post()
|