|
|
# -*- coding: utf-8 -*-
|
|
|
# @date：2024/1/9 16:49
|
|
|
# @Author：LiuYiJie
|
|
|
# @file： test_log
|
|
|
|
|
|
import os
|
|
|
from os.path import dirname, abspath, join
|
|
|
import logging
|
|
|
from logging import handlers
|
|
|
import datetime
|
|
|
|
|
|
|
|
|
class ColoredFormatter(logging.Formatter):
|
|
|
COLORS = {
|
|
|
"black": "\033[40m", # 黑色
|
|
|
"red": "\033[91m", # 红色
|
|
|
"green": "\033[92m", # 绿色
|
|
|
"yellow": "\033[93m", # 黄色
|
|
|
"blue": "\033[34m", # 蓝色
|
|
|
"purple": "\033[95m", # 紫色
|
|
|
"dgreen": "\033[96m", # 深绿
|
|
|
"white": "\033[97m", # 白色
|
|
|
"reset": '\033[0m', # 默认
|
|
|
}
|
|
|
|
|
|
DEFAULT_STYLES = {
|
|
|
"spam": COLORS['green'],
|
|
|
"DEBUG": COLORS['blue'],
|
|
|
"verbose": COLORS['blue'],
|
|
|
"INFO": COLORS['white'],
|
|
|
"WARNING": COLORS['yellow'],
|
|
|
"success": COLORS['green'],
|
|
|
"ERROR": COLORS['red'],
|
|
|
"CRITICAL": COLORS['red'],
|
|
|
"EXCEPTION": COLORS['red'],
|
|
|
|
|
|
"asctime": COLORS['green'],
|
|
|
"message": COLORS['green'],
|
|
|
"lineno": COLORS['purple'],
|
|
|
"threadName": COLORS['red'],
|
|
|
"module": COLORS['red'],
|
|
|
"levelname": COLORS['white'],
|
|
|
"name": COLORS['blue'],
|
|
|
"default": COLORS['blue'],
|
|
|
}
|
|
|
|
|
|
def __init__(self, styles=None):
|
|
|
super().__init__()
|
|
|
self.styles = styles or self.DEFAULT_STYLES
|
|
|
|
|
|
def set_color(self, levelname: str = None):
|
|
|
return self.styles.get(levelname, "reset")
|
|
|
|
|
|
def format(self, record):
|
|
|
levelname = record.levelname
|
|
|
asctime = f"{self.styles.get('default')}{datetime.datetime.fromtimestamp(record.created).strftime('%Y-%m-%d %H:%M:%S')}{self.COLORS['reset']}"
|
|
|
threadName = f"{self.styles.get('default')}{record.threadName}{self.COLORS['reset']}"
|
|
|
pathname = f"{self.styles.get('default')}{record.pathname}{self.COLORS['reset']}"
|
|
|
lineno = f"{self.styles.get('default')}{record.lineno}{self.COLORS['reset']}"
|
|
|
funcName = f"{self.styles.get('default')}{record.funcName}{self.COLORS['reset']}"
|
|
|
module = f"{self.styles.get('default')}{record.module}{self.COLORS['reset']}"
|
|
|
message = super().format(record)
|
|
|
|
|
|
levelcolor = self.set_color(levelname)
|
|
|
levelname = f"{levelcolor}{levelname}{self.COLORS['reset']}"
|
|
|
message = f"{levelcolor}{message}{self.COLORS['reset']}"
|
|
|
|
|
|
formatted_message = f"{threadName} - {asctime} - {levelname} - {pathname} - {module}:{funcName}:{lineno} - {message}"
|
|
|
return formatted_message
|
|
|
|
|
|
|
|
|
class ColoredConsoleHandler(logging.StreamHandler):
|
|
|
def __init__(self, formatter=None):
|
|
|
super().__init__()
|
|
|
self.formatter = formatter or ColoredFormatter()
|
|
|
|
|
|
|
|
|
def setup_logger():
|
|
|
log_obj = logging.getLogger()
|
|
|
log_obj.setLevel(logging.INFO)
|
|
|
log_obj.handlers.clear()
|
|
|
|
|
|
# 自定义输出格式，控制台输出
|
|
|
console_handler = ColoredConsoleHandler()
|
|
|
console_handler.setLevel(logging.INFO)
|
|
|
log_obj.addHandler(console_handler)
|
|
|
|
|
|
root_dir = dirname(dirname(abspath(__file__)))
|
|
|
LOG_DIR = join(root_dir, 'logs')
|
|
|
os.makedirs(LOG_DIR, exist_ok=True)
|
|
|
log_file = join(LOG_DIR, 'log_file.log')
|
|
|
|
|
|
# 文件输出，每天一个文件
|
|
|
file_handler = handlers.TimedRotatingFileHandler(log_file, when="midnight", interval=1, backupCount=10,
|
|
|
encoding='utf-8')
|
|
|
file_handler.setLevel(logging.INFO)
|
|
|
file_handler.suffix = "%Y-%m-%d.log"
|
|
|
file_handler.setFormatter(logging.Formatter(
|
|
|
'%(threadName)-10s - %(asctime)s - %(module)s - %(funcName)s:%(lineno)d - %(levelname)s - %(message)s'))
|
|
|
log_obj.addHandler(file_handler)
|
|
|
|
|
|
return log_obj
|
|
|
|
|
|
|
|
|
logger = setup_logger()
|