新增 消息定时推送功能

admin/dashboard/blueprints/message_push.py

@@ -1,4 +1,4 @@
-from flask import Blueprint, render_template, jsonify, request, current_app
+from flask import Blueprint, render_template, jsonify, request, current_app, session
 from .auth import login_required
 from loguru import logger
 import json
@@ -56,7 +56,7 @@ def api_create_task():
             
         # 生成任务ID
         data['task_id'] = str(uuid.uuid4())
-        data['creator_id'] = request.user.get('id')
+        data['creator_id'] = session.get('user_id')  # 从session中获取用户ID
         
         # 创建任务
         db = current_app.dashboard_server.task_db
@@ -187,7 +187,7 @@ def api_preview_task(task_id):
             
         # 发送预览
         message_push = current_app.dashboard_server.message_push_task.message_push
-        if not message_push.send_preview(task, [request.user.get('id')]):
+        if not message_push.send_preview(task, [session.get('user_id')]):  # 从session中获取用户ID
             return jsonify({"success": False, "error": "发送预览失败"}), 500
             
         return jsonify({