18dce66443
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: -LAN- <laipz8200@outlook.com>
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from flask_restx 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")
|