Function setup_server_logging

logging_utils/setup.py:76–207

Setup server logging system Args: logger_instance: Main logger instance log_ws_manager: WebSocket connection manager log_level_name: Log level name redirect_print_str: Whether to redirect print output Returns: Tuple[object, object]: Original std

(
    logger_instance: logging.Logger,
    log_ws_manager: Optional[WebSocketConnectionManager],
    log_level_name: str = "INFO",
    redirect_print_str: str = "false",
)

def setup_server_logging(
    logger_instance: logging.Logger,
    log_ws_manager: Optional[WebSocketConnectionManager],
    log_level_name: str = "INFO",
    redirect_print_str: str = "false",
) -> Tuple[object, object]:
    """
    Setup server logging system

    Args:
        logger_instance: Main logger instance
        log_ws_manager: WebSocket connection manager
        log_level_name: Log level name
        redirect_print_str: Whether to redirect print output

    Returns:
        Tuple[object, object]: Original stdout and stderr streams
    """
    log_level = getattr(logging, log_level_name.upper(), logging.INFO)
    redirect_print = redirect_print_str.lower() in ("true", "1", "yes")

    # Create necessary directories
    os.makedirs(LOG_DIR, exist_ok=True)
    os.makedirs(ACTIVE_AUTH_DIR, exist_ok=True)
    os.makedirs(SAVED_AUTH_DIR, exist_ok=True)

    # Clear existing handlers
    if logger_instance.hasHandlers():
        logger_instance.handlers.clear()
    logger_instance.setLevel(log_level)
    logger_instance.propagate = False

    # Remove old log file
    if os.path.exists(APP_LOG_FILE_PATH):
        try:
            os.remove(APP_LOG_FILE_PATH)
        except OSError as e:
            print(
                f"Warning (setup_server_logging): Failed to remove old app.log file '{APP_LOG_FILE_PATH}': {e}. Will rely on mode='w' for truncation.",
                file=sys.__stderr__,
            )

    # Use JSONFormatter for file logging if JSON_LOGS_ENABLED, otherwise PlainGridFormatter
    if JSON_LOGS_ENABLED:
        file_log_formatter = JSONFormatter()
    else:
        file_log_formatter = PlainGridFormatter()

    file_handler = logging.handlers.RotatingFileHandler(
        APP_LOG_FILE_PATH,
        maxBytes=LOG_FILE_MAX_BYTES,
        backupCount=LOG_FILE_BACKUP_COUNT,
        encoding="utf-8",
        mode="w",
    )
    file_handler.setFormatter(file_log_formatter)
    file_handler.setLevel(log_level)
    logger_instance.addHandler(file_handler)