mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
d68a4099ba
* ✨ 新增插件商店api
* chore(version): Update version to v0.2.2-7e15f20
---------
Co-authored-by: HibiKier <HibiKier@users.noreply.github.com>
30 lines
851 B
Python
30 lines
851 B
Python
from loguru import logger
|
|
from fastapi import APIRouter
|
|
from nonebot.utils import escape_tag
|
|
from starlette.websockets import WebSocket, WebSocketState, WebSocketDisconnect
|
|
|
|
from .log_manager import LOG_STORAGE
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.websocket("/logs")
|
|
async def system_logs_realtime(websocket: WebSocket):
|
|
await websocket.accept()
|
|
|
|
async def log_listener(log: str):
|
|
await websocket.send_text(log)
|
|
|
|
LOG_STORAGE.listeners.add(log_listener)
|
|
try:
|
|
while websocket.client_state == WebSocketState.CONNECTED:
|
|
recv = await websocket.receive()
|
|
logger.trace(
|
|
f"{system_logs_realtime.__name__!r} received "
|
|
f"<e>{escape_tag(repr(recv))}</e>"
|
|
)
|
|
except WebSocketDisconnect:
|
|
pass
|
|
finally:
|
|
LOG_STORAGE.listeners.remove(log_listener)
|