47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
# log.py
|
|
import logging
|
|
|
|
class Log:
|
|
# name, level no need to record
|
|
def __init__(self, name):
|
|
self.__log_name = name
|
|
self.__log_levlel = None
|
|
# deal duplicate handler
|
|
self.__handlers = []
|
|
# init default logger
|
|
self.set_console()
|
|
|
|
def set_console(self):
|
|
self.logger = logging.getLogger(self.__log_name)
|
|
# self.logger.setLevel(eval("logging." + level.upper()))
|
|
fh = logging.StreamHandler()
|
|
log_format = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
,datefmt='%Y-%m-%d %H:%M:%S')
|
|
fh.setFormatter(log_format)
|
|
if fh not in self.__handlers:
|
|
self.logger.addHandler(fh)
|
|
self.__handlers.append(fh)
|
|
|
|
def set_file(self, file_name):
|
|
self.logger = logging.getLogger(self.__log_name)
|
|
# self.logger.setLevel(eval("logging." + level.upper()))
|
|
fh = logging.FileHandler(file_name,encoding='utf-8') # FileHandler 的默认模式是 'a'，用 'a+' 在某些平台上可能导致文件指针行为异常。
|
|
log_format = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
,datefmt='%Y-%m-%d %H:%M:%S')
|
|
fh.setFormatter(log_format)
|
|
if fh not in self.__handlers:
|
|
self.logger.addHandler(fh)
|
|
self.__handlers.append(fh)
|
|
|
|
def message(self,level,message):
|
|
# self.logger.log(eval("logging." + level.upper()),message) # not safe for eval
|
|
self.logger.log(getattr(logging,level.upper()),message)
|
|
|
|
|
|
|
|
l = Log('ai-model')
|
|
l.message("warning","test Log class to console")
|
|
|
|
# l.set_file('libplus.log')
|
|
# l.message("info","test info Log to file")
|