104 lines
3.7 KiB
Python
104 lines
3.7 KiB
Python
from functools import wraps
|
||
from flask import request, g
|
||
import json
|
||
|
||
from flask_jwt_extended import current_user
|
||
from app.models import OperationHistory, User
|
||
|
||
from app import db,jwt
|
||
|
||
def log_operation(func):
|
||
@wraps(func)
|
||
def decorated(*args, **kwargs):
|
||
log_entry = OperationHistory(
|
||
api_path=request.path,
|
||
http_method=request.method,
|
||
ip_address=request.remote_addr,
|
||
request_params=None
|
||
# 用户信息暂留空
|
||
)
|
||
# 记录GET参数
|
||
if request.method == "GET":
|
||
log_entry.request_params = request.args.to_dict()
|
||
|
||
# 记录POST参数（支持JSON/表单/原始数据）
|
||
elif request.method == "POST":
|
||
if request.is_json:
|
||
res=request.get_json(silent=True) or None
|
||
json_str = json.dumps(res,ensure_ascii=False)
|
||
log_entry.request_params = json_str
|
||
elif request.form:
|
||
json_str = json.dumps(request.form.to_dict(),ensure_ascii=False)
|
||
log_entry.request_params =json_str
|
||
elif request.data:
|
||
json_str = json.dumps({"raw_data": request.data.decode('utf-8')[:500]},ensure_ascii=False)
|
||
# 原始数据截取前500字符
|
||
log_entry.request_params =json_str
|
||
|
||
|
||
try:
|
||
response = func(*args, **kwargs) # 执行原函数（触发 JWT 验证）
|
||
log_entry.user_id = current_user.id # 此时 current_user 已就绪
|
||
log_entry.username = current_user.username
|
||
log_entry.response_code = response.status_code
|
||
return response
|
||
except Exception as e:
|
||
log_entry.error_message = str(e)
|
||
log_entry.operation_result = 0
|
||
raise
|
||
finally:
|
||
db.session.add(log_entry)
|
||
db.session.commit()
|
||
return decorated
|
||
|
||
|
||
# def log_operation(func):
|
||
# @wraps(func)
|
||
# def decorated(*args, **kwargs):
|
||
# # 初始化日志对象（示例结构）
|
||
# log_entry = {
|
||
# "path": request.path,
|
||
# "method": request.method,
|
||
# "ip": request.remote_addr,
|
||
# "params": None # 待填充
|
||
# }
|
||
|
||
# try:
|
||
# # ---- 核心：记录请求参数 ----
|
||
# if request.method == "GET":
|
||
# params = request.args.to_dict()
|
||
# log_entry["params"] = filter_sensitive_data(params)
|
||
|
||
# elif request.method == "POST":
|
||
# if request.is_json:
|
||
# json_data = request.get_json(silent=True) or {}
|
||
# log_entry["params"] = filter_sensitive_data(json_data)
|
||
# elif request.form:
|
||
# form_data = request.form.to_dict()
|
||
# log_entry["params"] = filter_sensitive_data(form_data)
|
||
# elif request.files:
|
||
# log_entry["files"] = [f.filename for f in request.files.values()]
|
||
|
||
# # ---- 执行原函数 ----
|
||
# start_time = time()
|
||
# response = func(*args, **kwargs)
|
||
# duration = time() - start_time
|
||
|
||
# # ---- 记录响应 ----
|
||
# log_entry["status"] = response.status_code
|
||
# log_entry["duration"] = f"{duration:.3f}s"
|
||
# return response
|
||
|
||
# except Exception as e:
|
||
# log_entry["error"] = str(e)
|
||
# raise
|
||
# finally:
|
||
# # 实际存储到数据库/文件（此处打印示例）
|
||
# current_app.logger.info(json.dumps(log_entry, ensure_ascii=False))
|
||
|
||
# return decorated
|
||
|
||
# JWT 用户回调
|
||
@jwt.user_lookup_loader
|
||
def load_user(jwt_header, jwt_data):
|
||
return User.query.get(jwt_data["sub"]) |