51 lines
1.6 KiB
Python
Executable File
51 lines
1.6 KiB
Python
Executable File
import os
|
|
import logging
|
|
from datetime import datetime
|
|
|
|
|
|
def get_logger(root, name=None, debug=True):
|
|
"""
|
|
创建带时间戳的日志记录器
|
|
:param root: 日志文件目录
|
|
:param name: 日志名称
|
|
:param debug: 是否开启调试模式
|
|
"""
|
|
logger = logging.getLogger(name)
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
# 避免重复添加 Handler
|
|
if logger.hasHandlers():
|
|
logger.handlers.clear()
|
|
|
|
# 时间格式改为 年/月/日 时:分:秒
|
|
formatter = logging.Formatter("%(asctime)s - %(message)s", "%Y/%m/%d %H:%M:%S")
|
|
|
|
# 控制台输出
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setFormatter(formatter)
|
|
console_handler.setLevel(logging.DEBUG if debug else logging.INFO)
|
|
logger.addHandler(console_handler)
|
|
|
|
# 文件输出（仅非 debug 模式）
|
|
if not debug:
|
|
os.makedirs(root, exist_ok=True)
|
|
logfile = os.path.join(root, "run.log")
|
|
print(f"Create Log File in: {logfile}")
|
|
file_handler = logging.FileHandler(logfile, mode="w", encoding="utf-8")
|
|
file_handler.setLevel(logging.DEBUG)
|
|
file_handler.setFormatter(formatter)
|
|
logger.addHandler(file_handler)
|
|
|
|
return logger
|
|
|
|
|
|
if __name__ == "__main__":
|
|
time_str = datetime.now().strftime("%Y%m%d%H%M%S")
|
|
print(time_str)
|
|
|
|
logger = get_logger("./logs", debug=False) # 改成 False 测试文件输出
|
|
logger.debug("this is a {} debug message".format(1))
|
|
logger.info("this is an info message")
|
|
logger.debug("this is a debug message")
|
|
logger.info("this is an info message")
|