import logging

from typing import Union

from rich.console import Console

from rich.logging import RichHandler

def setup_logging(

terminal_width: Union[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