liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增 消息定时推送功能

Browse Source
This commit is contained in:
liuwei
2025-06-10 12:01:19 +08:00
parent 2f2071394f
commit cd849d2323
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
admin/dashboard/blueprints/auth.py
View File

@@ -1,5 +1,6 @@
from flask import Blueprint, render_template, request, redirect, url_for, session, current_app from flask import Blueprint, render_template, request, redirect, url_for, session, current_app
from functools import wraps from functools import wraps
from loguru import logger
# 创建认证蓝图 # 创建认证蓝图
auth_bp = Blueprint('auth', __name__) auth_bp = Blueprint('auth', __name__)
@@ -30,6 +31,7 @@ def login():
if username == server.username and password == server.password: if username == server.username and password == server.password:
session['logged_in'] = True session['logged_in'] = True
session['username'] = username # 存储用户名到session session['username'] = username # 存储用户名到session
logger.debug(f"Login successful. Session after login: {dict(session)}")
return redirect(url_for('main.index')) return redirect(url_for('main.index'))
else: else:
error = '用户名或密码错误' error = '用户名或密码错误'

3
admin/dashboard/blueprints/message_push.py
View File

@@ -55,8 +55,9 @@ def api_create_task():
return jsonify({"success": False, "error": "无效的请求数据"}), 400 return jsonify({"success": False, "error": "无效的请求数据"}), 400
# 获取用户名 # 获取用户名
logger.debug(f"session:{session}") logger.debug(f"Session before getting username: {dict(session)}")
username = session.get('username') username = session.get('username')
logger.debug(f"Username from session: {username}")
if not username: if not username:
return jsonify({"success": False, "error": "未登录或会话已过期"}), 401 return jsonify({"success": False, "error": "未登录或会话已过期"}), 401