32 lines
959 B
Python
32 lines
959 B
Python
|
import logging
|
|||
|
|
from conf.logConfig import LOG_GENERATE_FILE
|
|||
|
|
|
|||
|
|
generate_logger = logging.getLogger("generate_document_logger")
|
|||
|
|
|
|||
|
|
class GenerateLogger(object):
|
|||
|
|
instance = None
|
|||
|
|
|
|||
|
|
# 单例模式
|
|||
|
|
def __new__(cls, *args, **kwargs):
|
|||
|
|
if cls.instance is None:
|
|||
|
|
cls.instance = object.__new__(cls)
|
|||
|
|
return cls.instance
|
|||
|
|
else:
|
|||
|
|
return cls.instance
|
|||
|
|
|
|||
|
|
def __init__(self, model: str = '通用文档'):
|
|||
|
|
self.logger = generate_logger
|
|||
|
|
# 模块属性
|
|||
|
|
self.model = model
|
|||
|
|
|
|||
|
|
def write_warning_log(self, fragment: str, message: str):
|
|||
|
|
"""警告日志记录，暂时简单点：model和message"""
|
|||
|
|
whole_message = f"[{self.model}模块][{fragment}]片段:{message}"
|
|||
|
|
self.logger.warning(whole_message)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def delete_one_logs():
|
|||
|
|
"""删除生成文档logger的日志记录"""
|
|||
|
|
with open(LOG_GENERATE_FILE, 'w') as f:
|
|||
|
|
f.truncate()
|