logging_utils.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. from datetime import datetime
  2. def print_with_level(level, message):
  3. """打印带有时间戳和级别的消息"""
  4. timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
  5. print(f"{timestamp} - {level} - {message}")
  6. def debug(message):
  7. print_with_level("DEBUG", message)
  8. def info(message):
  9. print_with_level("INFO", message)
  10. def warning(message):
  11. print_with_level("WARNING", message)
  12. def error(message):
  13. print_with_level("ERROR", message)
  14. def critical(message):
  15. print_with_level("CRITICAL", message)
  16. def exception(message):
  17. """打印异常信息"""
  18. import traceback
  19. print_with_level("ERROR", f"{message}\n{traceback.format_exc()}")
  20. # 兼容性函数
  21. def get_logger(name=None, level=None, log_dir=None, log_to_file=True):
  22. """返回一个模拟logger的对象,实际上只是打印到控制台"""
  23. return type('Logger', (), {
  24. 'debug': debug,
  25. 'info': info,
  26. 'warning': warning,
  27. 'error': error,
  28. 'critical': critical,
  29. 'exception': exception
  30. })()
  31. def set_global_level(level):
  32. """为了保持兼容性,这个函数实际上不做任何事"""
  33. pass