322563dab0
- tb_user 用户表、tb_log 日志表 - Flask-Login 认证（login/logout/权限装饰器） - 用户管理页（admin 专有）：增删改查、改密、角色设置 - 操作日志页：分页查询、按用户/类型筛选 - 测试操作区指令自动记录日志 - 所有页面加 @login_required 保护 - 默认管理员 admin/admin123（首次启动自动创建）
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
"""测试信息 API"""
|
|
|
|
import csv
|
|
import io
|
|
from flask import Blueprint, jsonify, render_template, request, Response
|
|
from flask_login import login_required
|
|
from app.models import get_test_data, get_all_test_data_for_export
|
|
|
|
bp = Blueprint("test_data", __name__)
|
|
|
|
|
|
@bp.route("/test-data")
|
|
def test_data_page():
|
|
"""测试信息页"""
|
|
return render_template("test_data.html")
|
|
|
|
|
|
@bp.route("/api/test-data")
|
|
def api_test_data():
|
|
"""分页查询测试数据"""
|
|
page = request.args.get("page", 1, type=int)
|
|
per_page = request.args.get("per_page", 20, type=int)
|
|
serial = request.args.get("serial", "", type=str)
|
|
date_from = request.args.get("date_from", "", type=str)
|
|
date_to = request.args.get("date_to", "", type=str)
|
|
|
|
records, total = get_test_data(page, per_page, serial, date_from, date_to)
|
|
return jsonify({
|
|
"records": records,
|
|
"total": total,
|
|
"page": page,
|
|
"per_page": per_page,
|
|
"pages": (total + per_page - 1) // per_page if total > 0 else 1,
|
|
})
|
|
|
|
|
|
@bp.route("/api/test-data/export")
|
|
def api_export():
|
|
"""导出测试数据为 CSV"""
|
|
serial = request.args.get("serial", "", type=str)
|
|
date_from = request.args.get("date_from", "", type=str)
|
|
date_to = request.args.get("date_to", "", type=str)
|
|
|
|
records = get_all_test_data_for_export(serial, date_from, date_to)
|
|
|
|
output = io.StringIO()
|
|
writer = csv.writer(output)
|
|
|
|
# 表头
|
|
if records:
|
|
headers = [k for k in records[0].keys()]
|
|
writer.writerow(headers)
|
|
for r in records:
|
|
writer.writerow(r.values())
|
|
|
|
output.seek(0)
|
|
return Response(
|
|
output.getvalue(),
|
|
mimetype="text/csv",
|
|
headers={"Content-Disposition": "attachment; filename=test_data.csv"},
|
|
)
|