90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
|
from flask import Flask, jsonify, request
|
|||
|
|
from flask_cors import CORS
|
|||
|
|
import mysql.connector
|
|||
|
|
import database
|
|||
|
|
|
|||
|
|
app = Flask(__name__)
|
|||
|
|
# 启用CORS，允许前端访问
|
|||
|
|
CORS(app, resources={r"/api/*": {"origins": "*"}}, supports_credentials=True)
|
|||
|
|
|
|||
|
|
# 添加 v1 前缀
|
|||
|
|
@app.route('/api/v1/auth/login', methods=['POST'])
|
|||
|
|
def login():
|
|||
|
|
# 实现登录逻辑
|
|||
|
|
return {"code": 0, "data": {"token": "your-token"}, "message": "登录成功"}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.route('/api/v1/users/me', methods=['GET'])
|
|||
|
|
def get_current_user():
|
|||
|
|
return jsonify({
|
|||
|
|
"code": 0,
|
|||
|
|
"data": {
|
|||
|
|
"username": "admin",
|
|||
|
|
"roles": ["admin"]
|
|||
|
|
},
|
|||
|
|
"message": "获取用户信息成功"
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.route('/api/v1/users', methods=['GET'])
|
|||
|
|
def get_users():
|
|||
|
|
"""获取用户的API端点,支持分页和条件查询"""
|
|||
|
|
try:
|
|||
|
|
# 获取查询参数
|
|||
|
|
current_page = int(request.args.get('currentPage', 1))
|
|||
|
|
page_size = int(request.args.get('size', 10))
|
|||
|
|
username = request.args.get('username', '')
|
|||
|
|
email = request.args.get('email', '')
|
|||
|
|
|
|||
|
|
# 调用数据库函数获取分页和筛选后的用户数据
|
|||
|
|
users, total = database.get_users_with_pagination(current_page, page_size, username, email)
|
|||
|
|
|
|||
|
|
# 返回符合前端期望格式的数据
|
|||
|
|
return jsonify({
|
|||
|
|
"code": 0, # 成功状态码
|
|||
|
|
"data": {
|
|||
|
|
"list": users,
|
|||
|
|
"total": total
|
|||
|
|
},
|
|||
|
|
"message": "获取用户列表成功"
|
|||
|
|
})
|
|||
|
|
except Exception as e:
|
|||
|
|
# 错误处理
|
|||
|
|
return jsonify({
|
|||
|
|
"code": 500,
|
|||
|
|
"message": f"获取用户列表失败: {str(e)}"
|
|||
|
|
}), 500
|
|||
|
|
|
|||
|
|
@app.route('/api/v1/users/<string:user_id>', methods=['DELETE'])
|
|||
|
|
def delete_user(user_id):
|
|||
|
|
"""删除用户的API端点"""
|
|||
|
|
database.delete_user(user_id)
|
|||
|
|
return jsonify({
|
|||
|
|
"code": 0,
|
|||
|
|
"message": f"用户 {user_id} 删除成功"
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
@app.route('/api/v1/users', methods=['POST'])
|
|||
|
|
def create_user():
|
|||
|
|
"""创建用户的API端点"""
|
|||
|
|
data = request.json
|
|||
|
|
# 创建用户
|
|||
|
|
database.create_user(user_data=data)
|
|||
|
|
return jsonify({
|
|||
|
|
"code": 0,
|
|||
|
|
"message": "用户创建成功"
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
@app.route('/api/v1/users/<string:user_id>', methods=['PUT'])
|
|||
|
|
def update_user(user_id):
|
|||
|
|
"""更新用户的API端点"""
|
|||
|
|
data = request.json
|
|||
|
|
user_id = data.get('id')
|
|||
|
|
database.update_user(user_id=user_id, user_data=data)
|
|||
|
|
return jsonify({
|
|||
|
|
"code": 0,
|
|||
|
|
"message": f"用户 {user_id} 更新成功"
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
app.run(host='0.0.0.0', port=5000, debug=True)
|