94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
"""
|
|
集成了对 Pear Admin Flask 二次开发的邮件操作模块，并提供了相应的示例。
|
|
"""
|
|
from flask import current_app
|
|
from flask_mail import Message
|
|
|
|
from applications.common.curd import model_to_dicts
|
|
from applications.common.helper import ModelFilter
|
|
from applications.extensions import db, flask_mail
|
|
from applications.models import Mail
|
|
from applications.schemas import MailOutSchema
|
|
|
|
|
|
def get_all(receiver=None, subject=None, content=None):
|
|
"""
|
|
获取邮件列表，支持根据接收者、主题和内容进行筛选。
|
|
|
|
返回的列表中的字典结构如下::
|
|
|
|
{
|
|
"content": "", # HTML 内容
|
|
"create_at": "2022-12-25T10:51:17", # 创建时间
|
|
"id": 17, # 邮件ID
|
|
"realname": "超级管理", # 创建者姓名
|
|
"receiver": "", # 接收者
|
|
"subject": "" # 邮件主题
|
|
}
|
|
|
|
:param receiver: 接收者邮箱地址，支持模糊查询。
|
|
:param subject: 邮件主题，支持模糊查询。
|
|
:param content: 邮件内容，支持模糊查询。
|
|
:return: 返回符合条件的邮件列表。
|
|
"""
|
|
# 构造查询条件
|
|
mf = ModelFilter()
|
|
if receiver:
|
|
mf.contains(field_name="receiver", value=receiver)
|
|
if subject:
|
|
mf.contains(field_name="subject", value=subject)
|
|
if content:
|
|
mf.exact(field_name="content", value=content)
|
|
|
|
# 查询邮件数据并分页
|
|
mail = Mail.query.filter(mf.get_filter(Mail)).layui_paginate()
|
|
return model_to_dicts(schema=MailOutSchema, data=mail.items)
|
|
|
|
|
|
def add(receiver, subject, content, user_id):
|
|
"""
|
|
发送一封邮件，并将发送记录保存到数据库。 **该方法被邮件发送的视图函数调用。**
|
|
|
|
:param receiver: 接收者邮箱地址，多个邮箱用英文分号隔开。
|
|
:param subject: 邮件主题。
|
|
:param content: 邮件内容（HTML 格式）。
|
|
:param user_id: 发送者用户ID，表示谁发送了这封邮件。
|
|
可以使用 `from flask_login import current_user; current_user.id` 获取当前登录用户的ID。
|
|
:return: 发送成功返回 True，失败报错。
|
|
"""
|
|
send_mail(subject=subject, recipients=receiver.split(";"), content=content)
|
|
|
|
# 保存邮件记录到数据库
|
|
mail = Mail(receiver=receiver, subject=subject, content=content, user_id=user_id)
|
|
db.session.add(mail)
|
|
db.session.commit()
|
|
return True
|
|
|
|
|
|
def delete(id):
|
|
"""
|
|
删除指定的邮件记录。
|
|
|
|
:param id: 邮件ID。
|
|
:return: 删除成功返回 True，失败返回 False。
|
|
"""
|
|
res = Mail.query.filter_by(id=id).delete()
|
|
if not res:
|
|
return False
|
|
db.session.commit()
|
|
return True
|
|
|
|
|
|
def send_mail(subject, recipients, content):
|
|
"""
|
|
发送邮件（不记录发送日志）。
|
|
|
|
注意：如果发送失败会抛出异常，请使用 try-except 进行捕获。
|
|
|
|
:param subject: 邮件主题。
|
|
:param recipients: 接收者邮箱地址，多个邮箱用英文分号隔开。
|
|
:param content: 邮件内容（HTML 格式）。
|
|
"""
|
|
message = Message(subject=subject, recipients=recipients, html=content)
|
|
flask_mail.send(message)
|