227 lines
7.6 KiB
Python
227 lines
7.6 KiB
Python
from flask import Blueprint, render_template, jsonify, request, current_app, session
|
|
from .auth import login_required
|
|
from loguru import logger
|
|
import json
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
# 创建消息推送管理蓝图
|
|
message_push_bp = Blueprint('message_push', __name__, url_prefix='/message_push')
|
|
|
|
# 消息推送管理页面
|
|
@message_push_bp.route('/')
|
|
@login_required
|
|
def message_push_management():
|
|
"""消息推送管理页面"""
|
|
return render_template('message_push_management.html')
|
|
|
|
# API路由
|
|
@message_push_bp.route('/api/tasks', methods=['GET'])
|
|
@login_required
|
|
def api_tasks_list():
|
|
"""获取任务列表API"""
|
|
try:
|
|
# 获取查询参数
|
|
status = request.args.get('status')
|
|
start_time = request.args.get('start_time')
|
|
end_time = request.args.get('end_time')
|
|
page = int(request.args.get('page', 1))
|
|
limit = int(request.args.get('limit', 20))
|
|
|
|
# 获取任务列表
|
|
db = current_app.dashboard_server.task_db
|
|
tasks, total = db.get_tasks_list(status, start_time, end_time, page, limit)
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"tasks": tasks,
|
|
"total": total,
|
|
"page": page,
|
|
"limit": limit
|
|
}
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"获取任务列表失败: {e}")
|
|
return jsonify({"success": False, "error": str(e)}), 500
|
|
|
|
@message_push_bp.route('/api/tasks', methods=['POST'])
|
|
@login_required
|
|
def api_create_task():
|
|
"""创建任务API"""
|
|
try:
|
|
data = request.json
|
|
if not data:
|
|
return jsonify({"success": False, "error": "无效的请求数据"}), 400
|
|
|
|
# 获取用户名
|
|
logger.debug(f"Session before getting username: {dict(session)}")
|
|
username = session.get('username')
|
|
logger.debug(f"Username from session: {username}")
|
|
if not username:
|
|
return jsonify({"success": False, "error": "未登录或会话已过期"}), 401
|
|
|
|
# 生成任务ID
|
|
data['task_id'] = str(uuid.uuid4())
|
|
data['creator_id'] = username
|
|
|
|
# 创建任务
|
|
db = current_app.dashboard_server.task_db
|
|
task = db.create_task(data)
|
|
if not task:
|
|
return jsonify({"success": False, "error": "创建任务失败"}), 500
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"task": task
|
|
}
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"创建任务失败: {e}")
|
|
return jsonify({"success": False, "error": str(e)}), 500
|
|
|
|
@message_push_bp.route('/api/tasks/<task_id>', methods=['PUT'])
|
|
@login_required
|
|
def api_update_task(task_id):
|
|
"""更新任务API"""
|
|
try:
|
|
data = request.json
|
|
if not data:
|
|
return jsonify({"success": False, "error": "无效的请求数据"}), 400
|
|
|
|
# 获取任务
|
|
db = current_app.dashboard_server.task_db
|
|
task = db.get_task(task_id)
|
|
if not task:
|
|
return jsonify({"success": False, "error": "任务不存在"}), 404
|
|
|
|
# 更新任务
|
|
if not db.update_task(task_id, data):
|
|
return jsonify({"success": False, "error": "更新任务失败"}), 500
|
|
|
|
# 获取更新后的任务
|
|
updated_task = db.get_task(task_id)
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"task": updated_task
|
|
}
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"更新任务失败: {e}")
|
|
return jsonify({"success": False, "error": str(e)}), 500
|
|
|
|
@message_push_bp.route('/api/tasks/<task_id>', methods=['DELETE'])
|
|
@login_required
|
|
def api_delete_task(task_id):
|
|
"""删除任务API"""
|
|
try:
|
|
# 获取任务
|
|
db = current_app.dashboard_server.task_db
|
|
task = db.get_task(task_id)
|
|
if not task:
|
|
return jsonify({"success": False, "error": "任务不存在"}), 404
|
|
|
|
# 删除任务
|
|
if not db.delete_task(task_id):
|
|
return jsonify({"success": False, "error": "删除任务失败"}), 500
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"message": "任务已删除"
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"删除任务失败: {e}")
|
|
return jsonify({"success": False, "error": str(e)}), 500
|
|
|
|
@message_push_bp.route('/api/tasks/<task_id>/pause', methods=['POST'])
|
|
@login_required
|
|
def api_pause_task(task_id):
|
|
"""暂停任务API"""
|
|
try:
|
|
# 获取任务
|
|
db = current_app.dashboard_server.task_db
|
|
task = db.get_task(task_id)
|
|
if not task:
|
|
return jsonify({"success": False, "error": "任务不存在"}), 404
|
|
|
|
# 暂停任务
|
|
if not db.update_task(task_id, {'status': 'paused'}):
|
|
return jsonify({"success": False, "error": "暂停任务失败"}), 500
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"message": "任务已暂停"
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"暂停任务失败: {e}")
|
|
return jsonify({"success": False, "error": str(e)}), 500
|
|
|
|
@message_push_bp.route('/api/tasks/<task_id>/resume', methods=['POST'])
|
|
@login_required
|
|
def api_resume_task(task_id):
|
|
"""恢复任务API"""
|
|
try:
|
|
# 获取任务
|
|
db = current_app.dashboard_server.task_db
|
|
task = db.get_task(task_id)
|
|
if not task:
|
|
return jsonify({"success": False, "error": "任务不存在"}), 404
|
|
|
|
# 恢复任务
|
|
if not db.update_task(task_id, {'status': 'scheduled'}):
|
|
return jsonify({"success": False, "error": "恢复任务失败"}), 500
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"message": "任务已恢复"
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"恢复任务失败: {e}")
|
|
return jsonify({"success": False, "error": str(e)}), 500
|
|
|
|
@message_push_bp.route('/api/tasks/<task_id>/preview', methods=['POST'])
|
|
@login_required
|
|
def api_preview_task(task_id):
|
|
"""预览任务API"""
|
|
try:
|
|
# 获取任务
|
|
db = current_app.dashboard_server.task_db
|
|
task = db.get_task(task_id)
|
|
if not task:
|
|
return jsonify({"success": False, "error": "任务不存在"}), 404
|
|
|
|
# 发送预览
|
|
message_push = current_app.dashboard_server.message_push_task.message_push
|
|
if not message_push.send_preview(task, [session.get('user_id')]): # 从session中获取用户ID
|
|
return jsonify({"success": False, "error": "发送预览失败"}), 500
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"message": "预览已发送"
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"发送预览失败: {e}")
|
|
return jsonify({"success": False, "error": str(e)}), 500
|
|
|
|
@message_push_bp.route('/api/tasks/<task_id>/logs', methods=['GET'])
|
|
@login_required
|
|
def api_task_logs(task_id):
|
|
"""获取任务日志API"""
|
|
try:
|
|
# 获取查询参数
|
|
page = int(request.args.get('page', 1))
|
|
limit = int(request.args.get('limit', 20))
|
|
|
|
# 查询日志
|
|
db = current_app.dashboard_server.task_db
|
|
logs_data = db.get_task_logs_with_pagination(task_id, page, limit)
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"data": logs_data
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"获取任务日志失败: {e}")
|
|
return jsonify({"success": False, "error": str(e)}), 500 |