mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-05-08 16:32:41 +08:00
YUNSHEN XIE
3a6058e445
* add stable ci * fix * update * fix * rename tests dir;fix stable ci bug * add timeout limit * update
100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
#!/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# @author DDDivano
|
|
# encoding=utf-8 vi:ts=4:sw=4:expandtab:ft=python
|
|
"""
|
|
ServeTest
|
|
"""
|
|
import logging
|
|
import os
|
|
from datetime import datetime
|
|
|
|
import pytz
|
|
|
|
|
|
class Logger(object):
|
|
"""
|
|
日志记录配置的基础类。
|
|
"""
|
|
|
|
SAVE_LEVELS = ["both", "file", "channel"]
|
|
LOG_FORMAT = "%(asctime)s - %(name)s - [%(levelname)s] - %(message)s"
|
|
|
|
def __init__(self, loggername, save_level="both", log_path=None):
|
|
"""
|
|
使用指定名称和保存级别初始化日志记录器。
|
|
|
|
Args:
|
|
loggername (str): 日志记录器的名称。
|
|
save_level (str): 日志保存的级别。默认为"both"。file: 仅保存到文件，channel: 仅保存到控制台。
|
|
log_path (str, optional): 日志文件保存路径。默认为None。
|
|
"""
|
|
|
|
if save_level not in self.SAVE_LEVELS:
|
|
raise ValueError(f"Invalid save level: {save_level}. Allowed values: {self.SAVE_LEVELS}")
|
|
|
|
self.logger = logging.getLogger(loggername)
|
|
self.logger.setLevel(logging.DEBUG)
|
|
|
|
# 设置时区为东八区
|
|
tz = pytz.timezone("Asia/Shanghai")
|
|
|
|
# 自定义时间格式化器，指定时区为东八区
|
|
class CSTFormatter(logging.Formatter):
|
|
"""
|
|
自定义时间格式化器，指定时区为东八区
|
|
"""
|
|
|
|
def converter(self, timestamp):
|
|
"""
|
|
自定义时间转换函数，加上时区信息
|
|
Args:
|
|
timestamp (int): 时间戳。
|
|
Returns:
|
|
tuple: 格式化后的时间元组。
|
|
"""
|
|
dt = datetime.utcfromtimestamp(timestamp)
|
|
dt = pytz.utc.localize(dt).astimezone(tz)
|
|
return dt.timetuple()
|
|
|
|
formatter = CSTFormatter(self.LOG_FORMAT)
|
|
log_name = None
|
|
if save_level == "both" or save_level == "file":
|
|
os.makedirs(log_path, exist_ok=True)
|
|
log_filename = f"out_{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.log"
|
|
log_name = os.path.join(log_path, log_filename)
|
|
file_handler = logging.FileHandler(log_name, encoding="utf-8")
|
|
file_handler.setLevel(logging.DEBUG)
|
|
file_handler.setFormatter(formatter)
|
|
self.logger.addHandler(file_handler)
|
|
|
|
if save_level == "both" or save_level == "channel":
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(logging.DEBUG)
|
|
console_handler.setFormatter(formatter)
|
|
self.logger.addHandler(console_handler)
|
|
|
|
if log_name is None:
|
|
self.logger.info(
|
|
f"Logger initialized. Log level: {save_level}. "
|
|
f"Log path ({log_path}) is unused according to the level."
|
|
)
|
|
else:
|
|
self.logger.info(f"Logger initialized. Log level: {save_level}. Log path: {log_name}")
|
|
# Adjusting the timezone offset
|
|
|
|
def get_logger(self):
|
|
"""
|
|
Get the logger object
|
|
"""
|
|
return self.logger
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Test the logger
|
|
logger = Logger("test_logger", save_level="channel").get_logger()
|
|
logger.info("the is the beginning")
|
|
logger.debug("the is the beginning")
|
|
logger.warning("the is the beginning")
|
|
logger.error("the is the beginning")
|