logger.py 428 B

123456789101112131415161718
  1. import logging
  2. from rich.logging import RichHandler
  3. logger = logging.getLogger()
  4. logger.setLevel(logging.INFO)
  5. def add_handlers(logger: logging.Logger):
  6. handler = RichHandler()
  7. formatter = logging.Formatter(
  8. fmt="[%(levelname)s %(asctime)s] %(filename)s:%(lineno)d - %(message)s"
  9. )
  10. handler.setFormatter(formatter)
  11. logger.addHandler(handler)
  12. if not logger.hasHandlers():
  13. add_handlers(logger)