128 lines
4.1 KiB
Python
128 lines
4.1 KiB
Python
# !/usr/bin/python
|
||
# -*- coding:utf-8 -*-
|
||
# @Time : 2022/5/24 20:53
|
||
# @Author : wangjie
|
||
# @File : base_log.py
|
||
# @project : SensoroApi
|
||
|
||
import logging
|
||
import os.path
|
||
import sys
|
||
import time
|
||
|
||
import colorlog
|
||
|
||
from common.settings import LOG_DEBUG
|
||
from configs.dir_path_config import LOGS_DIR
|
||
|
||
|
||
class ColoredFormatter(logging.Formatter):
|
||
""" 设置日志颜色 """
|
||
COLOR_CODES = {
|
||
'DEBUG': '\033[1;34m', # Blue
|
||
'INFO': '\033[1;32m', # Green
|
||
'WARNING': '\033[1;33m', # Yellow
|
||
'ERROR': '\033[1;31m', # Red
|
||
'CRITICAL': '\033[1;41m' # Red (background)
|
||
}
|
||
|
||
def format(self, record):
|
||
levelname = record.levelname
|
||
msg = super().format(record)
|
||
color_code = self.COLOR_CODES.get(levelname, '')
|
||
reset_code = '\033[0m' # Reset color
|
||
return f'{color_code}{msg}{reset_code}'
|
||
|
||
|
||
class Logger:
|
||
"""用于全局记录log"""
|
||
_logger_instance = None
|
||
_log_path = None
|
||
|
||
@classmethod
|
||
def get_logger(cls):
|
||
"""单例模式,使多文件或多次调用时,始终使用这一个logger"""
|
||
if cls._logger_instance:
|
||
return cls._logger_instance
|
||
else:
|
||
logger = logging.getLogger()
|
||
|
||
# 设置日志默认可输出的最低等级
|
||
logger.setLevel(logging.INFO)
|
||
|
||
# 设置日志输入格式
|
||
# 设置写入文件时的格式
|
||
formatter_file = logging.Formatter(
|
||
'%(levelname)-8s%(asctime)s-%(filename)s[line:%(lineno)d]：%(message)s', '%Y-%m-%d %H:%m:%S'
|
||
)
|
||
# 设置终端上显示带颜色的格式
|
||
# formatter_stream = ColoredFormatter(
|
||
# '%(asctime)s-%(filename)s[line:%(lineno)d]-%(levelname)s：%(message)s', '%Y-%m-%d %H:%m:%S'
|
||
# )
|
||
formatter_stream = cls.log_color()
|
||
|
||
# 设置日志存储路径
|
||
if not cls._log_path:
|
||
# file_path = os.path.abspath(
|
||
# os.path.join(os.path.dirname(os.path.abspath(__file__)), '../outFiles/logs'))
|
||
file_path = LOGS_DIR
|
||
dir_name = time.strftime('%Y-%m-%d')
|
||
if sys.platform in ('win32', 'win64'): # 兼容window文件命名时不支持":"的方式
|
||
file_name = time.strftime('%Y-%m-%d_%H点%M分%S秒') + '-' + 'log.log'
|
||
else:
|
||
file_name = time.strftime('%Y-%m-%d_%H:%M:%S') + '-' + 'log.log'
|
||
if not os.path.exists(os.path.join(file_path, dir_name)):
|
||
os.makedirs(os.path.join(file_path, dir_name))
|
||
cls._log_path = os.path.join(file_path, dir_name, file_name)
|
||
|
||
# 创建FileHandler,用于写入日志
|
||
fh = logging.FileHandler(cls._log_path, encoding='utf-8')
|
||
fh.setFormatter(formatter_file)
|
||
fh.setLevel(logging.INFO)
|
||
|
||
# 创建StreamHandler,用于输出到控制台
|
||
ch = logging.StreamHandler()
|
||
ch.setFormatter(formatter_stream)
|
||
ch.setLevel(logging.INFO)
|
||
|
||
# 添加输出
|
||
logger.addHandler(ch)
|
||
logger.addHandler(fh)
|
||
|
||
# 控制日志级别
|
||
if LOG_DEBUG:
|
||
logger.setLevel(logging.DEBUG)
|
||
ch.setLevel(logging.DEBUG)
|
||
fh.setLevel(logging.DEBUG)
|
||
|
||
cls._logger_instance = logger
|
||
|
||
return cls._logger_instance
|
||
|
||
@classmethod
|
||
def log_color(cls):
|
||
""" 设置日志颜色 """
|
||
log_colors_config = {
|
||
'DEBUG': 'cyan',
|
||
'INFO': 'green',
|
||
'WARNING': 'yellow',
|
||
'ERROR': 'red',
|
||
'CRITICAL': 'bold_red,bg_bold_white',
|
||
}
|
||
|
||
formatter = colorlog.ColoredFormatter(
|
||
'%(log_color)s%(asctime)s-%(filename)s[line:%(lineno)d]-%(levelname)s：%(message)s', '%Y-%m-%d %H:%m:%S',
|
||
log_colors=log_colors_config
|
||
)
|
||
return formatter
|
||
|
||
|
||
logger = Logger().get_logger()
|
||
|
||
if __name__ == '__main__':
|
||
logger.debug("测试日志")
|
||
logger.info("测试日志")
|
||
logger.warning("测试日志")
|
||
logger.error("测试日志")
|
||
logger.critical("测试日志")
|