import logging from rich.console import Console from rich.logging import RichHandler def setup_logging(terminal_width: int | None = None, level: int = logging.INFO) -> None: logger = logging.getLogger("fastapi_cli") console = Console(width=terminal_width) if terminal_width else None rich_handler = RichHandler( show_time=False, rich_tracebacks=True, tracebacks_show_locals=True, markup=True, show_path=False, console=console, ) rich_handler.setFormatter(logging.Formatter("%(message)s")) logger.addHandler(rich_handler) logger.setLevel(level) logger.propagate = False