26 lines
690 B
Python
26 lines
690 B
Python
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
|