mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 03:15:06 +08:00
209f16af76
* style: remove necessary imports * style: fix F841 * style: fix F401 * style: fix F811 * style: fix E402 * style: fix E721 * style: fix E722 * style: fix E722 * style: fix F541 * style: ruff format * style: all passed * style: add ruff in deps * style: more ignores in ruff.toml * style: add pre-commit
72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
import logging
|
||
import sys
|
||
import time
|
||
|
||
import colorlog
|
||
|
||
from ...utils import constants
|
||
|
||
|
||
log_colors_config = {
|
||
'DEBUG': 'green', # cyan white
|
||
'INFO': 'white',
|
||
'WARNING': 'yellow',
|
||
'ERROR': 'red',
|
||
'CRITICAL': 'cyan',
|
||
}
|
||
|
||
|
||
async def init_logging(extra_handlers: list[logging.Handler] = None) -> logging.Logger:
|
||
# 删除所有现有的logger
|
||
for handler in logging.root.handlers[:]:
|
||
logging.root.removeHandler(handler)
|
||
|
||
level = logging.INFO
|
||
|
||
if constants.debug_mode:
|
||
level = logging.DEBUG
|
||
|
||
log_file_name = 'data/logs/langbot-%s.log' % time.strftime(
|
||
'%Y-%m-%d', time.localtime()
|
||
)
|
||
|
||
qcg_logger = logging.getLogger('langbot')
|
||
|
||
qcg_logger.setLevel(level)
|
||
|
||
color_formatter = colorlog.ColoredFormatter(
|
||
fmt='%(log_color)s[%(asctime)s.%(msecs)03d] %(filename)s (%(lineno)d) - [%(levelname)s] : %(message)s',
|
||
datefmt='%m-%d %H:%M:%S',
|
||
log_colors=log_colors_config,
|
||
)
|
||
|
||
stream_handler = logging.StreamHandler(sys.stdout)
|
||
# stream_handler.setLevel(level)
|
||
# stream_handler.setFormatter(color_formatter)
|
||
stream_handler.stream = open(
|
||
sys.stdout.fileno(), mode='w', encoding='utf-8', buffering=1
|
||
)
|
||
|
||
log_handlers: list[logging.Handler] = [
|
||
stream_handler,
|
||
logging.FileHandler(log_file_name, encoding='utf-8'),
|
||
]
|
||
log_handlers += extra_handlers if extra_handlers is not None else []
|
||
|
||
for handler in log_handlers:
|
||
handler.setLevel(level)
|
||
handler.setFormatter(color_formatter)
|
||
qcg_logger.addHandler(handler)
|
||
|
||
qcg_logger.debug('日志初始化完成，日志级别：%s' % level)
|
||
logging.basicConfig(
|
||
level=logging.CRITICAL, # 设置日志输出格式
|
||
format='[DEPR][%(asctime)s.%(msecs)03d] %(pathname)s (%(lineno)d) - [%(levelname)s] :\n%(message)s',
|
||
# 日志输出的格式
|
||
# -8表示占位符，让输出左对齐，输出长度都为8位
|
||
datefmt='%Y-%m-%d %H:%M:%S', # 时间输出的格式
|
||
handlers=[logging.NullHandler()],
|
||
)
|
||
|
||
return qcg_logger
|