101 lines
3.7 KiB
Python
101 lines
3.7 KiB
Python
import logging
|
||
import os
|
||
import os.path
|
||
import datetime
|
||
import inspect
|
||
|
||
|
||
class ServiceLoggerHandler(logging.Handler):
|
||
def __init__(self, file_fmt=None):
|
||
self.file_fmt = file_fmt
|
||
if file_fmt is None:
|
||
self.file_fmt = os.path.join('logs', '%Y-%m-%d.log')
|
||
logging.Handler.__init__(self)
|
||
|
||
def emit(self, record):
|
||
msg = self.format(record)
|
||
fpath = datetime.datetime.now().strftime(self.file_fmt)
|
||
fdir = os.path.dirname(fpath)
|
||
try:
|
||
if not os.path.exists(fdir):
|
||
os.makedirs(fdir)
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
try:
|
||
f = open(fpath, 'a')
|
||
f.write(msg)
|
||
f.write("\n")
|
||
f.flush()
|
||
f.close()
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
|
||
class Logger:
|
||
def __init__(self, service_name, log_level=logging.DEBUG, log_dir=None, print_terminal=True):
|
||
self.service_name = service_name
|
||
self.log_level = log_level
|
||
self.log_dir = log_dir
|
||
self.print_terminal = print_terminal
|
||
|
||
logging.addLevelName(10, 'DEBG')
|
||
logging.addLevelName(20, 'INFO')
|
||
logging.addLevelName(30, 'WARN')
|
||
logging.addLevelName(40, 'EROR')
|
||
logging.addLevelName(50, 'CRIT')
|
||
|
||
self.logger = logging.getLogger(self.service_name)
|
||
self.logger.setLevel(self.log_level)
|
||
|
||
# 创建一个handler，用于写入日志文件
|
||
file_fmt = os.path.join(self.log_dir, self.service_name, self.service_name + "-%Y-%m-%d.log")
|
||
fh = ServiceLoggerHandler(file_fmt)
|
||
fh.setLevel(self.log_level)
|
||
|
||
# 再创建一个handler，用于输出到控制台
|
||
ch = logging.StreamHandler()
|
||
ch.setLevel(self.log_level)
|
||
|
||
# 定义handler的输出格式
|
||
formatter = logging.Formatter(
|
||
'[%(asctime)s] [%(levelname).4s] (%(name)s, %(call_fname)s:%(call_line)d) %(message)s')
|
||
fh.setFormatter(formatter)
|
||
if print_terminal:
|
||
ch.setFormatter(formatter)
|
||
|
||
# 给logger添加handler
|
||
self.logger.addHandler(fh)
|
||
if print_terminal:
|
||
self.logger.addHandler(ch)
|
||
|
||
def Debug(self, msg):
|
||
previous_frame = inspect.currentframe().f_back
|
||
pathname, line_number, func_name, lines, index = inspect.getframeinfo(previous_frame)
|
||
filename = os.path.basename(pathname)
|
||
self.logger.debug(msg, extra={'call_fname': filename, 'call_line': line_number})
|
||
|
||
def Info(self, msg):
|
||
previous_frame = inspect.currentframe().f_back
|
||
pathname, line_number, func_name, lines, index = inspect.getframeinfo(previous_frame)
|
||
filename = os.path.basename(pathname)
|
||
self.logger.info(msg, extra={'call_fname': filename, 'call_line': line_number})
|
||
|
||
def Warn(self, msg):
|
||
previous_frame = inspect.currentframe().f_back
|
||
pathname, line_number, func_name, lines, index = inspect.getframeinfo(previous_frame)
|
||
filename = os.path.basename(pathname)
|
||
self.logger.warning(msg, extra={'call_fname': filename, 'call_line': line_number})
|
||
|
||
def Error(self, msg):
|
||
previous_frame = inspect.currentframe().f_back
|
||
pathname, line_number, func_name, lines, index = inspect.getframeinfo(previous_frame)
|
||
filename = os.path.basename(pathname)
|
||
self.logger.error(msg, extra={'call_fname': filename, 'call_line': line_number})
|
||
|
||
def Critical(self, msg):
|
||
previous_frame = inspect.currentframe().f_back
|
||
pathname, line_number, func_name, lines, index = inspect.getframeinfo(previous_frame)
|
||
filename = os.path.basename(pathname)
|
||
self.logger.critical(msg, extra={'call_fname': filename, 'call_line': line_number})
|