207 lines
6.7 KiB
Python
207 lines
6.7 KiB
Python
# users.py
|
||
|
||
from flask import Blueprint, render_template, request, redirect, url_for, flash, session, jsonify
|
||
import requests
|
||
import math
|
||
from frontend.config import Config
|
||
|
||
users_bp = Blueprint('users', __name__, url_prefix='/users')
|
||
|
||
# 显示用户列表，带分页
|
||
@users_bp.route('/')
|
||
def list_users():
|
||
"""显示用户列表，带分页"""
|
||
if not session.get('token'):
|
||
flash("请先登录", "warning")
|
||
return redirect(url_for('auth.login'))
|
||
|
||
# 1. 获取用户总数
|
||
response = requests.post(f'{Config.BASE_API_URL}/admin/users/sum',
|
||
json={"token": session.get('token')})
|
||
total_users = 0
|
||
if response.status_code == 200:
|
||
try:
|
||
total_users = int(response.json().get('message', 0))
|
||
except (ValueError, TypeError):
|
||
total_users = 0
|
||
|
||
# 2. 分页查询用户
|
||
page = request.args.get('page', 1, type=int)
|
||
start = (page - 1) * 20 + 1
|
||
end = page * 20
|
||
|
||
coupon_resp = requests.get(
|
||
f"{Config.BASE_API_URL}/admin/coupons",
|
||
params={"token": session['token']}
|
||
)
|
||
coupons = coupon_resp.json() if coupon_resp.status_code == 200 else []
|
||
|
||
response2 = requests.post(f'{Config.BASE_API_URL}/admin/users/query',
|
||
json={"token": session.get('token'), "start": start, "end": end})
|
||
users = response2.json() if response2.status_code == 200 else []
|
||
|
||
total_pages = math.ceil(total_users / 20)
|
||
return render_template("users.html",
|
||
users=users,
|
||
coupons=coupons,
|
||
page=page,
|
||
total_users=total_users,
|
||
total_pages=total_pages)
|
||
|
||
# 搜索接口
|
||
@users_bp.route('/search', methods=['POST'])
|
||
def search_users():
|
||
"""
|
||
前端在 /users 页面点击搜索时 POST 到 /users/search
|
||
不需要从前端传 token，直接使用 session['token']
|
||
"""
|
||
if not session.get('token'):
|
||
return jsonify([]) # 或返回 401
|
||
|
||
data = request.get_json() or {}
|
||
query_mode = data.get('query_mode')
|
||
query_value = data.get('query_value')
|
||
|
||
# 若搜索值为空，直接返回空
|
||
if not query_value:
|
||
return jsonify([])
|
||
|
||
payload = {
|
||
"token": session['token'],
|
||
"query_mode": query_mode,
|
||
"query_value": query_value
|
||
}
|
||
resp = requests.post(f"{Config.BASE_API_URL}/admin/users/search", json=payload)
|
||
if resp.status_code == 200:
|
||
return jsonify(resp.json())
|
||
return jsonify([])
|
||
|
||
# 删除用户
|
||
@users_bp.route('/delete_user', methods=['POST'])
|
||
def delete_user():
|
||
if not session.get('token'):
|
||
flash("请先登录", "warning")
|
||
return redirect(url_for('auth.login'))
|
||
|
||
uid = request.form.get("uid")
|
||
page = request.args.get('page', 1, type=int)
|
||
payload = {
|
||
"token": session['token'],
|
||
"uid": int(uid)
|
||
}
|
||
resp = requests.post(f"{Config.BASE_API_URL}/admin/users/del", json=payload)
|
||
if resp.status_code == 200:
|
||
flash("删除成功", "success")
|
||
else:
|
||
flash("删除失败", "danger")
|
||
return redirect(url_for('users.list_users', page=page))
|
||
|
||
# 更新用户
|
||
@users_bp.route('/update_user', methods=['POST'])
|
||
def update_user():
|
||
if not session.get('token'):
|
||
flash("请先登录", "warning")
|
||
return redirect(url_for('auth.login'))
|
||
|
||
uid = request.form.get('uid')
|
||
page = request.args.get('page', 1, type=int)
|
||
payload = {
|
||
"token": session['token'],
|
||
"uid": int(uid),
|
||
"username": request.form.get('username'),
|
||
"email": request.form.get('email'),
|
||
"phone_number": request.form.get('phone_number'),
|
||
"gender": request.form.get('gender'),
|
||
"user_type": request.form.get('user_type')
|
||
}
|
||
resp = requests.post(f"{Config.BASE_API_URL}/admin/users/update", json=payload)
|
||
if resp.status_code == 200:
|
||
flash("更新成功", "success")
|
||
else:
|
||
flash("更新失败", "danger")
|
||
|
||
return redirect(url_for('users.list_users', page=page))
|
||
|
||
# 重置密码（实现具体逻辑）
|
||
@users_bp.route('/reset_password', methods=['POST'])
|
||
def reset_password():
|
||
"""
|
||
这个路由接收新密码并调用后端 API 进行密码重置。
|
||
"""
|
||
if not session.get('token'):
|
||
flash("请先登录", "warning")
|
||
return redirect(url_for('auth.login'))
|
||
|
||
uid = request.form.get('uid')
|
||
new_password = request.form.get('password')
|
||
# 确保 uid 和 new_password 不为空
|
||
if not uid or not new_password:
|
||
flash("缺少必要参数", "danger")
|
||
return redirect(url_for('users.list_users', page=request.args.get('page', 1, type=int)))
|
||
|
||
payload = {
|
||
"token": session['token'],
|
||
"uid": int(uid),
|
||
"new_password": new_password
|
||
}
|
||
resp = requests.post(f"{Config.BASE_API_URL}/admin/users/update_password", json=payload)
|
||
if resp.status_code == 200:
|
||
flash("重置密码成功", "success")
|
||
else:
|
||
flash("重置密码失败", "danger")
|
||
return redirect(url_for('users.list_users', page=request.args.get('page', 1, type=int)))
|
||
|
||
# 在users_bp蓝图下添加新路由
|
||
@users_bp.route('/adjust_points', methods=['POST'])
|
||
def adjust_points():
|
||
if not session.get('token'):
|
||
flash("请先登录", "warning")
|
||
return redirect(url_for('auth.login'))
|
||
|
||
uid = request.form.get('uid')
|
||
points = int(request.form.get('points'))
|
||
reason = request.form.get('reason')
|
||
|
||
payload = {
|
||
"token": session['token'],
|
||
"uid": uid,
|
||
"points": points,
|
||
"reason": reason
|
||
}
|
||
|
||
resp = requests.post(f"{Config.BASE_API_URL}/admin/users/update_points", json=payload)
|
||
if resp.status_code == 200:
|
||
flash("积分更新成功", "success")
|
||
else:
|
||
flash("积分更新失败", "danger")
|
||
|
||
return redirect(url_for('users.list_users', page=request.args.get('page', 1)))
|
||
|
||
@users_bp.route('/issue-coupon', methods=['POST'])
|
||
def issue_coupon():
|
||
if not session.get('token'):
|
||
return redirect(url_for('auth.login'))
|
||
|
||
try:
|
||
user_id = int(request.form['user_id'])
|
||
coupon_id = int(request.form['coupon_id'])
|
||
|
||
# 通过Flask中转请求到FastAPI
|
||
response = requests.post(
|
||
f"{Config.BASE_API_URL}/admin/coupons/issue",
|
||
json={
|
||
"token": session['token'],
|
||
"user_id": user_id,
|
||
"coupon_id": coupon_id
|
||
}
|
||
)
|
||
print(request.form)
|
||
if response.status_code == 200:
|
||
flash("优惠券发放成功", "success")
|
||
else:
|
||
flash(response.json().get('detail', '发放失败'), "danger")
|
||
except requests.exceptions.ConnectionError:
|
||
flash("后端服务不可用", "danger")
|
||
|
||
return redirect(url_for('users.list_users'))
|