99 lines
3.1 KiB
Python
99 lines
3.1 KiB
Python
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 str_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")
|
|
def main():
|
|
return render_template('admin/mail/main.html')
|
|
|
|
|
|
# 用户分页查询
|
|
@admin_mail.get('/data')
|
|
@authorize("admin:mail:main")
|
|
def data():
|
|
# 获取请求参数
|
|
receiver = str_escape(request.args.get("receiver"))
|
|
subject = str_escape(request.args.get('subject'))
|
|
content = str_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 = str_escape(req_json.get("receiver"))
|
|
subject = str_escape(req_json.get('subject'))
|
|
content = str_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="批量删除成功")
|