realtimex-deeptutor 0.5.0.post1__py3-none-any.whl

src/logging/handlers/websocket.py

@@ -0,0 +1,127 @@
+# -*- coding: utf-8 -*-
+"""
+WebSocket Log Handler
+=====================
+
+Handlers for streaming logs to WebSocket clients.
+"""
+
+import asyncio
+import logging
+
+
+class WebSocketLogHandler(logging.Handler):
+    """
+    A logging handler that streams log records to a WebSocket via asyncio Queue.
+
+    Usage:
+        queue = asyncio.Queue()
+        handler = WebSocketLogHandler(queue)
+        logger.addHandler(handler)
+
+        # In WebSocket endpoint:
+        while True:
+            log_entry = await queue.get()
+            await websocket.send_json(log_entry)
+    """
+
+    # Symbols for different log types (matching ConsoleFormatter)
+    SYMBOLS = {
+        "DEBUG": "·",
+        "INFO": "●",
+        "SUCCESS": "✓",
+        "WARNING": "⚠",
+        "ERROR": "✗",
+        "CRITICAL": "✗",
+    }
+
+    def __init__(self, queue: asyncio.Queue, include_module: bool = True):
+        """
+        Initialize WebSocket log handler.
+
+        Args:
+            queue: asyncio.Queue to put log entries into
+            include_module: Whether to include module name in output
+        """
+        super().__init__()
+        self.queue = queue
+        self.include_module = include_module
+        self.setFormatter(logging.Formatter("%(message)s"))
+
+    def emit(self, record: logging.LogRecord):
+        """Emit a log record to the queue."""
+        try:
+            msg = self.format(record)
+
+            # Get symbol
+            display_level = getattr(record, "display_level", record.levelname)
+            symbol = getattr(record, "symbol", self.SYMBOLS.get(display_level, "●"))
+
+            # Get module name
+            module_name = getattr(record, "module_name", record.name)
+
+            # Build formatted content
+            if self.include_module:
+                content = f"[{module_name}] {symbol} {msg}"
+            else:
+                content = f"{symbol} {msg}"
+
+            # Construct structured message
+            log_entry = {
+                "type": "log",
+                "level": display_level,
+                "module": module_name,
+                "symbol": symbol,
+                "content": content,
+                "message": msg,
+                "timestamp": record.created,
+            }
+
+            # Put into queue non-blocking
+            try:
+                self.queue.put_nowait(log_entry)
+            except asyncio.QueueFull:
+                pass  # Drop log if queue is full
+
+        except Exception:
+            self.handleError(record)
+
+
+class LogInterceptor:
+    """
+    Context manager to temporarily attach a WebSocketLogHandler to a logger.
+
+    Usage:
+        queue = asyncio.Queue()
+        logger = logging.getLogger("ai_tutor.Solver")
+
+        with LogInterceptor(logger, queue):
+            # All logs from this logger will be streamed to queue
+            solver.solve(problem)
+    """
+
+    def __init__(
+        self,
+        logger: logging.Logger,
+        queue: asyncio.Queue,
+        include_module: bool = True,
+    ):
+        """
+        Initialize log interceptor.
+
+        Args:
+            logger: Logger to intercept
+            queue: Queue to stream logs to
+            include_module: Whether to include module name in output
+        """
+        self.logger = logger
+        self.handler = WebSocketLogHandler(queue, include_module)
+
+    def __enter__(self):
+        """Attach handler on context enter."""
+        self.logger.addHandler(self.handler)
+        return self.handler
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Remove handler on context exit."""
+        self.logger.removeHandler(self.handler)