更新插件功能

applications/dev/mail.py

@@ -0,0 +1,83 @@
+"""
+集成了对 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)
+    # orm查询
+    # 使用分页获取data需要.items
+    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 来表示当前登录用户
+    :return: 成功与否
+    """
+    try:
+        msg = Message(subject=subject, recipients=receiver.split(";"), html=content)
+        flask_mail.send(msg)
+    except BaseException as e:
+        current_app.log_exception(e)
+        return False
+
+    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: 成功与否
+    """
+    res = Mail.query.filter_by(id=id).delete()
+    if not res:
+        return False
+    db.session.commit()
+    return True