mirror of
https://gitee.com/dify_ai/dify.git
synced 2024-12-05 12:48:32 +08:00
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import logging
|
|
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
|
|
elif app.config.get('MAIL_TYPE') == 'smtp':
|
|
from libs.smtp import SMTPClient
|
|
if not app.config.get('SMTP_SERVER') or not app.config.get('SMTP_PORT'):
|
|
raise ValueError('SMTP_SERVER and SMTP_PORT are required for smtp mail type')
|
|
if not app.config.get('SMTP_USE_TLS') and app.config.get('SMTP_OPPORTUNISTIC_TLS'):
|
|
raise ValueError('SMTP_OPPORTUNISTIC_TLS is not supported without enabling SMTP_USE_TLS')
|
|
self._client = SMTPClient(
|
|
server=app.config.get('SMTP_SERVER'),
|
|
port=app.config.get('SMTP_PORT'),
|
|
username=app.config.get('SMTP_USERNAME'),
|
|
password=app.config.get('SMTP_PASSWORD'),
|
|
_from=app.config.get('MAIL_DEFAULT_SEND_FROM'),
|
|
use_tls=app.config.get('SMTP_USE_TLS'),
|
|
opportunistic_tls=app.config.get('SMTP_OPPORTUNISTIC_TLS')
|
|
)
|
|
else:
|
|
raise ValueError('Unsupported mail type {}'.format(app.config.get('MAIL_TYPE')))
|
|
else:
|
|
logging.warning('MAIL_TYPE is not set')
|
|
|
|
|
|
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()
|