Coverage for src/fastapi_cloud_cli/logging.py: 100%

1import logging 7 ctx1abcdefg

2import os 7 ctx1abcdefg

3 

4from rich.console import Console 7 ctx1abcdefg

5from rich.logging import RichHandler 7 ctx1abcdefg

6 

7 

8def setup_logging(terminal_width: int | None = None, level: int | None = None) -> None: 7 ctx1abcdefg

9 if level is None: 7 ctx1abcdefg

10 level = ( 7 ctx1abcdefg

11 logging.DEBUG if os.getenv("FASTAPI_CLOUD_DEBUG") == "1" else logging.INFO 

12 ) 

13 

14 logger = logging.getLogger("fastapi_cloud_cli") 7 ctx1abcdefg

15 console = Console(width=terminal_width) if terminal_width else None 7 ctx1abcdefg

16 rich_handler = RichHandler( 7 ctx1abcdefg

17 show_time=False, 

18 rich_tracebacks=True, 

19 tracebacks_show_locals=True, 

20 markup=True, 

21 show_path=False, 

22 console=console, 

23 ) 

24 rich_handler.setFormatter(logging.Formatter("{message}", style="{")) 7 ctx1abcdefg

25 logger.addHandler(rich_handler) 7 ctx1abcdefg

26 

27 logger.setLevel(level) 7 ctx1abcdefg

28 logger.propagate = False 7 ctx1abcdefg