PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-11-16 12:51:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

重构 logger

Browse Source
This commit is contained in:
xtaodada 2022-05-27 11:38:27 +08:00 committed by 洛水.山岭居室
parent f2c9116902
commit acd265a38c
1 changed files with 23 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
logger.py
View File

@ -4,68 +4,52 @@ import colorlog
import os
current_path = os.path.realpath(os.getcwd())
log_path = os.path.join(current_path, 'logs')
log_path = os.path.join(current_path, "logs")
if not os.path.exists(log_path):
os.mkdir(log_path)
log_file_name = os.path.join(log_path, 'log.log')
log_file_name = os.path.join(log_path, "log.log")
log_colors_config = {
'DEBUG': 'cyan',
'INFO': 'green',
'WARNING': 'yellow',
'ERROR': 'red',
'CRITICAL': 'red',
"DEBUG": "cyan",
"INFO": "green",
"WARNING": "yellow",
"ERROR": "red",
"CRITICAL": "red",
}
class Logger:
def __init__(self):
self.logger = logging.getLogger("TGPaimonBot")
root_logger = logging.getLogger()
root_logger.setLevel(logging.CRITICAL)
self.logger.setLevel(logging.INFO)
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')
"%(log_color)s[%(asctime)s] [%(levelname)s] - %(message)s", log_colors=log_colors_config)
self.formatter2 = logging.Formatter("[%(asctime)s] [%(levelname)s] - %(message)s")
fh = RotatingFileHandler(filename=log_file_name, maxBytes=1024 * 1024 * 5, backupCount=5,
encoding="utf-8")
fh.setFormatter(self.formatter2)
root_logger.addHandler(fh)
ch = colorlog.StreamHandler()
ch.setFormatter(self.formatter)
root_logger.addHandler(ch)
def getLogger(self):
return self.logger
def _console(self, level, message, exc_info=None):
fh = RotatingFileHandler(filename=log_file_name, maxBytes=1024 * 1024 * 5, backupCount=5,
encoding='utf-8')
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)
self.logger.debug(msg=msg, exc_info=exc_info)
def info(self, msg, exc_info=None):
self._console('info', msg, exc_info)
self.logger.info(msg=msg, exc_info=exc_info)
def warning(self, msg, exc_info=None):
self._console('warning', msg, exc_info)
self.logger.warning(msg=msg, exc_info=exc_info)
def error(self, msg, exc_info=None):
self._console('error', msg, exc_info)
self.logger.error(msg=msg, exc_info=exc_info)
Log = Logger()