kami_walmart_card/logger.py

from loguru import logger
import sys
import pytz
from datetime import datetime

# 移除默认的日志处理器
logger.remove()

# 设置中国时区
china_tz = pytz.timezone("Asia/Shanghai")


# 自定义时间格式函数
def custom_time():
    return datetime.now(china_tz).strftime("%Y-%m-%d %H:%M:%S")


# 添加新的处理器，输出到控制台
logger.add(sys.stdout, format="{time:YYYY-MM-DD HH:mm:ss} {level} {message}", level="INFO", serialize=False)

# 添加文件日志处理器
logger.add("walmart.log", rotation="20 MB", retention="30 days", format="{time:YYYY-MM-DD HH:mm:ss} {level} {message}")

# 创建一个带有时间信息的 Logger
logger = logger.bind(time=custom_time())


def get_logger():
    return logger