diff --git a/Utils/log_handler.py b/Utils/log_handler.py new file mode 100644 index 0000000000000000000000000000000000000000..1bea4cf8d0abd9929ca3d760ce13ff152bd3e3e6 --- /dev/null +++ b/Utils/log_handler.py @@ -0,0 +1,41 @@ +import logging +import datetime +import os +import inspect + +class LogHandler: + def __init__(self, name='autotest_logger', log_dir='../TestLog'): + self.logger = logging.getLogger(name) + self.logger.setLevel(logging.DEBUG) + + if not os.path.exists(log_dir): + os.makedirs(log_dir) + + today = datetime.datetime.now().strftime("%Y-%m-%d") + log_file = os.path.join(log_dir, f"{today}_log.log") + file_handler = logging.FileHandler(log_file) + + formatter = logging.Formatter( + '%(asctime)s [%(levelname)s] in %(module)s (line %(lineno)d): %(message)s' + ) + file_handler.setFormatter(formatter) + + self.logger.addHandler(file_handler) + + def info(self, message, *args, **kwargs): + frame = inspect.stack()[1] + module = inspect.getmodule(frame[0]) + title = f"{module.__name__}" + formatted_message = f"{title}\n{message}" + self.logger.info(formatted_message, *args, **kwargs) + + def error(self, message, *args, **kwargs): + frame = inspect.stack()[1] + module = inspect.getmodule(frame[0]) + title = f"{module.__name__}" + formatted_message = f"{title}\n{message}" + self.logger.error(formatted_message, *args, **kwargs) + +# 使用示例: +log_dir = '../TestLog' # 指定日志存放目录 +logger = LogHandler(name='autotest_logger', log_dir=log_dir) diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000000000000000000000000000000000000..f75243b21a7d3f4d958e56a134e360bc0f56014a Binary files /dev/null and b/requirements.txt differ