添加邮件管理模块

applications/extensions/__init__.py

@@ -5,7 +5,7 @@ from .init_login import init_login_manager
 from .init_debug_tool import init_debug_tool
 from .init_template_directives import init_template_directives
 from .init_error_views import init_error_views
-from .init_mail import init_mail
+from .init_mail import init_mail, mail as flask_mail
 from .init_apscheduler import init_scheduler
 from .init_upload import init_upload
 from .init_dotenv import init_dotenv

applications/models/__init__.py

@@ -6,4 +6,5 @@ from .admin_power import Power
 from .admin_role import Role
 from .admin_role_power import role_power
 from .admin_user import User
-from .admin_user_role import user_role
+from .admin_user_role import user_role
+from .admin_mail import Mail

applications/models/admin_mail.py

@@ -0,0 +1,12 @@
+import datetime
+from applications.extensions import db
+
+
+class Mail(db.Model):
+    __tablename__ = 'admin_mail'
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True, comment='邮件编号')
+    receiver = db.Column(db.String(1024), comment='收件人邮箱')
+    subject = db.Column(db.String(128), comment='邮件主题')
+    content = db.Column(db.Text(), comment='邮件正文')
+    user_id = db.Column(db.Integer, comment='发送人id')
+    create_at = db.Column(db.DateTime, default=datetime.datetime.now, comment='创建时间')

applications/schemas/__init__.py

@@ -5,3 +5,4 @@ from .admin_dict import DictDataOutSchema, DictTypeOutSchema
 from .admin_dept import DeptOutSchema
 from .admin_log import LogOutSchema
 from .admin_photo import PhotoOutSchema
+from .admin_mail import MailOutSchema

applications/schemas/admin_mail.py

@@ -0,0 +1,19 @@
+from applications.extensions import ma
+from marshmallow import fields
+from applications.models import User
+
+
+# 用户models的序列化类
+class MailOutSchema(ma.Schema):
+    id = fields.Integer()
+    receiver = fields.Str()
+    subject = fields.Str()
+    content = fields.Str()
+    realname = fields.Method("get_realname")
+    create_at = fields.DateTime()
+
+    def get_realname(self, obj):
+        if obj.user_id != None:
+            return User.query.filter_by(id=obj.user_id).first().realname
+        else:
+            return None

applications/view/admin/__init__.py

@@ -9,6 +9,7 @@ from applications.view.admin.role import admin_role
 from applications.view.admin.user import admin_user
 from applications.view.admin.monitor import admin_monitor_bp
 from applications.view.admin.task import admin_task
+from applications.view.admin.mail import admin_mail
 
 
 def register_admin_views(app: Flask):
@@ -21,3 +22,4 @@ def register_admin_views(app: Flask):
     app.register_blueprint(admin_role)
     app.register_blueprint(admin_dict)
     app.register_blueprint(admin_task)
+    app.register_blueprint(admin_mail)

applications/view/admin/mail.py

@@ -0,0 +1,99 @@
+from flask import Blueprint, render_template, request, current_app
+from flask_login import current_user
+from flask_mail import Message
+
+from applications.common.curd import model_to_dicts
+from applications.common.helper import ModelFilter
+from applications.common.utils.http import table_api, fail_api, success_api
+from applications.common.utils.rights import authorize
+from applications.common.utils.validate import xss_escape
+from applications.extensions import db, flask_mail
+from applications.models import Mail
+from applications.schemas import MailOutSchema
+
+admin_mail = Blueprint('adminMail', __name__, url_prefix='/admin/mail')
+
+
+# 用户管理
+@admin_mail.get('/')
+@authorize("admin:mail:main", log=True)
+def main():
+    return render_template('admin/mail/main.html')
+
+
+#   用户分页查询
+@admin_mail.get('/data')
+@authorize("admin:mail:main", log=True)
+def data():
+    # 获取请求参数
+    receiver = xss_escape(request.args.get("receiver"))
+    subject = xss_escape(request.args.get('subject'))
+    content = xss_escape(request.args.get('content'))
+    # 查询参数构造
+    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()
+    count = mail.total
+    # 返回api
+    return table_api(data=model_to_dicts(schema=MailOutSchema, data=mail.items), count=count)
+
+
+# 用户增加
+@admin_mail.get('/add')
+@authorize("admin:mail:add", log=True)
+def add():
+    return render_template('admin/mail/add.html')
+
+
+@admin_mail.post('/save')
+@authorize("admin:mail:add", log=True)
+def save():
+    req_json = request.json
+    receiver = xss_escape(req_json.get("receiver"))
+    subject = xss_escape(req_json.get('subject'))
+    content = xss_escape(req_json.get('content'))
+    user_id = current_user.id
+
+    try:
+        msg = Message(subject=subject, recipients=receiver.split(";"), body=content)
+        flask_mail.send(msg)
+    except Exception as e:
+        current_app.log_exception(e)
+        return fail_api(msg="发送失败，请检查邮件配置或发送人邮箱是否写错")
+
+    mail = Mail(receiver=receiver, subject=subject, content=content, user_id=user_id)
+
+    db.session.add(mail)
+    db.session.commit()
+    return success_api(msg="增加成功")
+
+
+# 删除用户
+@admin_mail.delete('/remove/<int:id>')
+@authorize("admin:mail:remove", log=True)
+def delete(id):
+    res = Mail.query.filter_by(id=id).delete()
+    if not res:
+        return fail_api(msg="删除失败")
+    db.session.commit()
+    return success_api(msg="删除成功")
+
+
+# 批量删除
+@admin_mail.delete('/batchRemove')
+@authorize("admin:mail:remove", log=True)
+def batch_remove():
+    ids = request.form.getlist('ids[]')
+    for id in ids:
+        res = Mail.query.filter_by(id=id).delete()
+        if not res:
+            return fail_api(msg="批量删除失败")
+    db.session.commit()
+    return success_api(msg="批量删除成功")