Coverage for collectra/logger.py: 100.00%

1import logging 

2 

3from rich.logging import RichHandler 

4 

5 

6def setup_logging(verbose: bool = False) -> None: 

7 level = logging.DEBUG if verbose else logging.WARNING 

8 handler = RichHandler(rich_tracebacks=True, show_path=False, markup=True) 

9 handler.setLevel(level) 

10 pkg_logger = logging.getLogger("collectra") 

11 pkg_logger.setLevel(level) 

12 pkg_logger.handlers.clear() 

13 pkg_logger.addHandler(handler) 

14 

15 

16def get_logger(name: str) -> logging.Logger: 

17 return logging.getLogger(name)