mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 11:42:29 +08:00
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
from typing import Optional
|
|
|
|
import resend
|
|
from flask import Flask
|
|
|
|
|
|
class Mail:
|
|
def __init__(self):
|
|
self._client = None
|
|
self._default_send_from = None
|
|
|
|
def is_inited(self) -> bool:
|
|
return self._client is not None
|
|
|
|
def init_app(self, app: Flask):
|
|
if app.config.get('MAIL_TYPE'):
|
|
if app.config.get('MAIL_DEFAULT_SEND_FROM'):
|
|
self._default_send_from = app.config.get('MAIL_DEFAULT_SEND_FROM')
|
|
|
|
if app.config.get('MAIL_TYPE') == 'resend':
|
|
api_key = app.config.get('RESEND_API_KEY')
|
|
if not api_key:
|
|
raise ValueError('RESEND_API_KEY is not set')
|
|
|
|
api_url = app.config.get('RESEND_API_URL')
|
|
if api_url:
|
|
resend.api_url = api_url
|
|
|
|
resend.api_key = api_key
|
|
self._client = resend.Emails
|
|
else:
|
|
raise ValueError('Unsupported mail type {}'.format(app.config.get('MAIL_TYPE')))
|
|
|
|
def send(self, to: str, subject: str, html: str, from_: Optional[str] = None):
|
|
if not self._client:
|
|
raise ValueError('Mail client is not initialized')
|
|
|
|
if not from_ and self._default_send_from:
|
|
from_ = self._default_send_from
|
|
|
|
if not from_:
|
|
raise ValueError('mail from is not set')
|
|
|
|
if not to:
|
|
raise ValueError('mail to is not set')
|
|
|
|
if not subject:
|
|
raise ValueError('mail subject is not set')
|
|
|
|
if not html:
|
|
raise ValueError('mail html is not set')
|
|
|
|
self._client.send({
|
|
"from": from_,
|
|
"to": to,
|
|
"subject": subject,
|
|
"html": html
|
|
})
|
|
|
|
|
|
def init_app(app: Flask):
|
|
mail.init_app(app)
|
|
|
|
|
|
mail = Mail()
|