mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
|
import logging
|
|||
|
|
from logging.handlers import RotatingFileHandler # 按文件大小滚动备份
|
|||
|
|
import colorlog # 控制台日志输入颜色
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
cur_path = os.path.realpath(os.getcwd()) # log_path是存放日志的路径
|
|||
|
|
log_path = os.path.join(cur_path, 'logs')
|
|||
|
|
if not os.path.exists(log_path):
|
|||
|
|
os.mkdir(log_path) # 如果不存在这个logs文件夹，就自动创建一个
|
|||
|
|
logName = os.path.join(log_path, 'log.log') # 文件的命名
|
|||
|
|
|
|||
|
|
log_colors_config = {
|
|||
|
|
'DEBUG': 'cyan',
|
|||
|
|
'INFO': 'green',
|
|||
|
|
'WARNING': 'yellow',
|
|||
|
|
'ERROR': 'red',
|
|||
|
|
'CRITICAL': 'red',
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Logger:
|
|||
|
|
def __init__(self):
|
|||
|
|
self.logName = logName
|
|||
|
|
self.logger = logging.getLogger()
|
|||
|
|
self.logger.setLevel(logging.INFO)
|
|||
|
|
# self.formatter = colorlog.ColoredFormatter( '%(log_color)s[%(asctime)s] [%(filename)s:%(lineno)d] [%(
|
|||
|
|
# module)s:%(funcName)s] [%(levelname)s] - %(message)s', log_colors=log_colors_config)
|
|||
|
|
self.formatter = colorlog.ColoredFormatter(
|
|||
|
|
'%(log_color)s[%(asctime)s] [%(levelname)s] - %(message)s', log_colors=log_colors_config)
|
|||
|
|
self.formatter2 = colorlog.ColoredFormatter(
|
|||
|
|
'[%(asctime)s] [%(levelname)s] - %(message)s')
|
|||
|
|
|
|||
|
|
def getLogger(self):
|
|||
|
|
return self.logger
|
|||
|
|
|
|||
|
|
def __console(self, level, message, exc_info=None):
|
|||
|
|
# 创建一个FileHandler，用于写到本地
|
|||
|
|
fh = RotatingFileHandler(filename=self.logName, maxBytes=1024 * 1024 * 5, backupCount=5,
|
|||
|
|
encoding='utf-8') # 使用RotatingFileHandler类，滚动备份日志
|
|||
|
|
fh.setLevel(logging.INFO)
|
|||
|
|
fh.setFormatter(self.formatter2)
|
|||
|
|
self.logger.addHandler(fh)
|
|||
|
|
|
|||
|
|
ch = colorlog.StreamHandler()
|
|||
|
|
ch.setLevel(logging.INFO)
|
|||
|
|
ch.setFormatter(self.formatter)
|
|||
|
|
self.logger.addHandler(ch)
|
|||
|
|
|
|||
|
|
if level == 'info':
|
|||
|
|
self.logger.info(msg=message, exc_info=exc_info)
|
|||
|
|
elif level == 'debug':
|
|||
|
|
self.logger.debug(msg=message, exc_info=exc_info)
|
|||
|
|
elif level == 'warning':
|
|||
|
|
self.logger.warning(msg=message, exc_info=exc_info)
|
|||
|
|
elif level == 'error':
|
|||
|
|
self.logger.error(msg=message, exc_info=exc_info)
|
|||
|
|
# 这两行代码是为了避免日志输出重复问题
|
|||
|
|
self.logger.removeHandler(ch)
|
|||
|
|
self.logger.removeHandler(fh)
|
|||
|
|
fh.close() # 关闭打开的文件
|
|||
|
|
|
|||
|
|
def debug(self, msg, exc_info=None):
|
|||
|
|
self.__console('debug', msg, exc_info)
|
|||
|
|
|
|||
|
|
def info(self, msg, exc_info=None):
|
|||
|
|
self.__console('info', msg, exc_info)
|
|||
|
|
|
|||
|
|
def warning(self, msg, exc_info=None):
|
|||
|
|
self.__console('warning', msg, exc_info)
|
|||
|
|
|
|||
|
|
def error(self, msg, exc_info=None):
|
|||
|
|
self.__console('error', msg, exc_info)
|
|||
|
|
|
|||
|
|
|
|||
|
|
Log = Logger()
|