lt
72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, Request
|
|
from sqlalchemy.orm import Session
|
|
from pathlib import Path
|
|
import subprocess
|
|
from ..database import get_db
|
|
from ..models import User
|
|
from .auth import get_current_admin
|
|
|
|
router = APIRouter(prefix="/api", tags=["optimizer_logs"])
|
|
|
|
PROJECT_ROOT = Path(__file__).parent.parent.parent.parent.parent
|
|
|
|
@router.post("/optimizer/run")
|
|
def run_optimizer(
|
|
request: Request,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_admin)
|
|
):
|
|
"""
|
|
触发合规优化器运行（管理员）
|
|
"""
|
|
try:
|
|
body = request.json()
|
|
except Exception:
|
|
raise HTTPException(status_code=400, detail="Invalid JSON")
|
|
topic_id = body.get("topic_id")
|
|
if not topic_id:
|
|
raise HTTPException(status_code=400, detail="topic_id is required")
|
|
|
|
script_path = PROJECT_ROOT / "automation" / "scripts" / "compliance_optimizer.py"
|
|
if not script_path.exists():
|
|
raise HTTPException(status_code=500, detail="Optimizer script not found")
|
|
|
|
try:
|
|
result = subprocess.run(
|
|
["python", str(script_path), "--topic-id", topic_id],
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=600,
|
|
cwd=PROJECT_ROOT
|
|
)
|
|
if result.returncode != 0:
|
|
raise HTTPException(status_code=500, detail=f"Optimizer failed: {result.stderr}")
|
|
return {"ok": True, "message": "Optimization completed", "output": result.stdout}
|
|
except subprocess.TimeoutExpired:
|
|
raise HTTPException(status_code=500, detail="Optimizer timed out")
|
|
|
|
@router.get("/logs")
|
|
def get_logs(
|
|
request: Request,
|
|
type: str = None, # creator, optimizer, collector
|
|
date: str = None, # YYYY-MM-DD
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_admin)
|
|
):
|
|
"""
|
|
读取系统日志文件内容（管理员）
|
|
"""
|
|
if not type or not date:
|
|
raise HTTPException(status_code=400, detail="type and date parameters are required")
|
|
logs_dir = PROJECT_ROOT / "automation" / "logs"
|
|
filename = f"{type}_{date}.log"
|
|
log_path = logs_dir / filename
|
|
if not log_path.exists():
|
|
# 日志文件不存在返回空内容
|
|
return {"type": type, "date": date, "content": ""}
|
|
try:
|
|
content = log_path.read_text(encoding="utf-8")
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to read log: {str(e)}")
|
|
return {"type": type, "date": date, "content": content}
|