dify/api/controllers/inner_api/mail.py

from flask_restful import Resource, reqparse

from controllers.console.wraps import setup_required
from controllers.inner_api import api
from controllers.inner_api.wraps import billing_inner_api_only, enterprise_inner_api_only
from tasks.mail_inner_task import send_inner_email_task

_mail_parser = reqparse.RequestParser()
_mail_parser.add_argument("to", type=str, action="append", required=True)
_mail_parser.add_argument("subject", type=str, required=True)
_mail_parser.add_argument("body", type=str, required=True)
_mail_parser.add_argument("substitutions", type=dict, required=False)


class BaseMail(Resource):
    """Shared logic for sending an inner email."""

    def post(self):
        args = _mail_parser.parse_args()
        send_inner_email_task.delay(
            to=args["to"],
            subject=args["subject"],
            body=args["body"],
            substitutions=args["substitutions"],
        )
        return {"message": "success"}, 200


class EnterpriseMail(BaseMail):
    method_decorators = [setup_required, enterprise_inner_api_only]


class BillingMail(BaseMail):
    method_decorators = [setup_required, billing_inner_api_only]


api.add_resource(EnterpriseMail, "/enterprise/mail")
api.add_resource(BillingMail, "/billing/mail")