Ruff fix

src/webui/logs_ws.py

@@ -1,4 +1,5 @@
 """WebSocket 日志推送模块"""
+
 from fastapi import APIRouter, WebSocket, WebSocketDisconnect
 from typing import Set
 import json
@@ -14,30 +15,30 @@ active_connections: Set[WebSocket] = set()
 
 def load_recent_logs(limit: int = 100) -> list[dict]:
     """从日志文件中加载最近的日志
-    
+
     Args:
         limit: 返回的最大日志条数
-        
+
     Returns:
         日志列表
     """
     logs = []
     log_dir = Path("logs")
-    
+
     if not log_dir.exists():
         return logs
-    
+
     # 获取所有日志文件,按修改时间排序
     log_files = sorted(log_dir.glob("app_*.log.jsonl"), key=lambda f: f.stat().st_mtime, reverse=True)
-    
+
     # 用于生成唯一 ID 的计数器
     log_counter = 0
-    
+
     # 从最新的文件开始读取
     for log_file in log_files:
         if len(logs) >= limit:
             break
-            
+
         try:
             with open(log_file, "r", encoding="utf-8") as f:
                 lines = f.readlines()
@@ -49,7 +50,9 @@ def load_recent_logs(limit: int = 100) -> list[dict]:
                         log_entry = json.loads(line.strip())
                         # 转换为前端期望的格式
                         # 使用时间戳 + 计数器生成唯一 ID
-                        timestamp_id = log_entry.get("timestamp", "0").replace("-", "").replace(" ", "").replace(":", "")
+                        timestamp_id = (
+                            log_entry.get("timestamp", "0").replace("-", "").replace(" ", "").replace(":", "")
+                        )
                         formatted_log = {
                             "id": f"{timestamp_id}_{log_counter}",
                             "timestamp": log_entry.get("timestamp", ""),
@@ -64,7 +67,7 @@ def load_recent_logs(limit: int = 100) -> list[dict]:
         except Exception as e:
             logger.error(f"读取日志文件失败 {log_file}: {e}")
             continue
-    
+
     # 反转列表，使其按时间顺序排列（旧到新）
     return list(reversed(logs))
 
@@ -72,35 +75,35 @@ def load_recent_logs(limit: int = 100) -> list[dict]:
 @router.websocket("/ws/logs")
 async def websocket_logs(websocket: WebSocket):
     """WebSocket 日志推送端点
-    
+
     客户端连接后会持续接收服务器端的日志消息
     """
     await websocket.accept()
     active_connections.add(websocket)
     logger.info(f"📡 WebSocket 客户端已连接，当前连接数: {len(active_connections)}")
-    
+
     # 连接建立后，立即发送历史日志
     try:
         recent_logs = load_recent_logs(limit=100)
         logger.info(f"发送 {len(recent_logs)} 条历史日志到客户端")
-        
+
         for log_entry in recent_logs:
             await websocket.send_text(json.dumps(log_entry, ensure_ascii=False))
     except Exception as e:
         logger.error(f"发送历史日志失败: {e}")
-    
+
     try:
         # 保持连接，等待客户端消息或断开
         while True:
             # 接收客户端消息（用于心跳或控制指令）
             data = await websocket.receive_text()
-            
+
             # 可以处理客户端的控制消息，例如：
             # - "ping" -> 心跳检测
             # - {"filter": "ERROR"} -> 设置日志级别过滤
             if data == "ping":
                 await websocket.send_text("pong")
-                
+
     except WebSocketDisconnect:
         active_connections.discard(websocket)
         logger.info(f"📡 WebSocket 客户端已断开，当前连接数: {len(active_connections)}")
@@ -111,19 +114,19 @@ async def websocket_logs(websocket: WebSocket):
 
 async def broadcast_log(log_data: dict):
     """广播日志到所有连接的 WebSocket 客户端
-    
+
     Args:
         log_data: 日志数据字典
     """
     if not active_connections:
         return
-    
+
     # 格式化为 JSON
     message = json.dumps(log_data, ensure_ascii=False)
-    
+
     # 记录需要断开的连接
     disconnected = set()
-    
+
     # 广播到所有客户端
     for connection in active_connections:
         try:
@@ -131,7 +134,7 @@ async def broadcast_log(log_data: dict):
         except Exception:
             # 发送失败，标记为断开
             disconnected.add(connection)
-    
+
     # 清理断开的连接
     if disconnected:
         active_connections.difference_update(disconnected)