修改登录退出逻辑

This commit is contained in:
dxc
2026-02-04 14:29:59 +08:00
parent 13590b1fac
commit fd5600b65b
12 changed files with 411 additions and 235 deletions

View File

@ -1,5 +1,5 @@
# app/api/v1/auth.py
from flask import Blueprint, request, jsonify
from flask import Blueprint, request, jsonify, current_app
from flask_jwt_extended import jwt_required, get_jwt
from app.services.auth_service import AuthService
@ -10,13 +10,38 @@ auth_bp = Blueprint('auth', __name__)
def login():
try:
data = request.get_json()
if not data:
return jsonify({'msg': '无效的请求数据'}), 400
if not data.get('username') or not data.get('password'):
return jsonify({'msg': '请输入用户名和密码'}), 400
# 调用 Service 层逻辑
result = AuthService.login(data)
return jsonify({'msg': '登录成功', 'data': result}), 200
# [关键修复]
# 前端 store 代码写的是: token.value = res.data.access_token
# 所以我们这里不能把 access_token 包裹在 data 字段里,
# 而是应该直接合并返回,或者让前端去 data.data 里面取。
# 为了不改前端,我们这里做解构返回:
response_data = {
'msg': '登录成功',
'access_token': result.get('access_token'),
'user': result.get('user')
}
return jsonify(response_data), 200
except ValueError as ve:
# 捕获已知的业务错误(如密码错误、用户不存在)
return jsonify({'msg': str(ve)}), 401
except Exception as e:
return jsonify({'msg': str(e)}), 401
# [关键修复] 打印详细报错到控制台,方便排查 500 错误
# (例如数据库连接失败、表不存在等)
current_app.logger.error(f"Login Failed Error: {str(e)}")
# 生产环境不建议直接把 error 返回给前端,但调试阶段很有用
return jsonify({'msg': f'服务器内部错误: {str(e)}'}), 500
# 新增:创建用户 (替代了原来的注册)
@ -30,8 +55,13 @@ def create_user():
claims = get_jwt()
operator_role = claims.get('role')
# 增加一个简单的权限判断(可选)
if operator_role not in ['super_admin', 'supervisor']:
return jsonify({'msg': '权限不足,无法创建用户'}), 403
result = AuthService.create_user(data, operator_role)
return jsonify({'msg': '用户创建成功', 'data': result}), 201
except Exception as e:
# 这里虽然返回 400但实际可能包含 403 的含义,具体看前端处理
current_app.logger.error(f"User Create Failed: {str(e)}")
return jsonify({'msg': str(e)}), 400

View File

@ -1,8 +1,12 @@
from flask import Blueprint, jsonify
# 定义蓝图,名字叫 'transactions'
# 定义蓝图
# 注意:这个变量名 trans_bp 必须与 app/__init__.py 中注册时引用的名字一致
trans_bp = Blueprint('transactions', __name__)
@trans_bp.route('/test', methods=['GET'])
def test_transaction():
return jsonify({"message": "Transaction module is working"})
"""
测试接口:用于验证 Transaction 模块是否加载成功
"""
return jsonify({"message": "Transaction module is working", "status": "success"})