【Python】自己写日志功能

Python有自带的logging模块，用于日志记录，功能很强大，但不好用，使用挺麻烦的，而且发现了几个bug，调用了一个logger.warning()一次，结果日志文件中出现了n行记录，且逐渐变成n+1,n+2,...越来越多。也不想去查原因，索性自己写一个日志功能。

我想要的日志的核心功能就只有两点：

1. 保存触发时间、异常堆栈信息。

2. 用文件形式保存。

上面第1点可以用Python自带的traceback模块实现，文件读写就很简单了。

MyLogger.py

import os
import time
import traceback

class MyLogger:
    def __init__(self,filename='./log/myapp.log'):#默认的日志文件为当前目录的子目录下的myapp.log文件
        self.logfile=filename
        path = filename[0:filename.rfind("/")]
        if not os.path.isdir(path):  # 无文件夹时创建
            os.makedirs(path)
        if not os.path.isfile(filename):
            f=open(self.logfile,'w',encoding='utf-8')
            f.close()
        self.default_formator='{0},File {1},{2}():{3}
'
            
    def get_invoker_info(self):
        invoker_file=traceback.extract_stack()[-3].filename#调用本logger的代码的所在文件
        invoker_func=traceback.extract_stack()[-3].name#调用本logger的代码（其实就是下面的log方法和error方法）所在的函数
        return invoker_file,invoker_func
    '''普通日志，保存任意信息，且也有堆栈信息'''
    def log(self,msg=''):
        t=time.strftime('%Y-%m-%d %H:%M:%S')
        print('Log at:',t)
        formator='Log at '+self.default_formator
        #获取调用者信息
        invoker_file,invoker_func=self.get_invoker_info()
        content=formator.format(t,os.path.basename(invoker_file),invoker_func,msg)
        self.append_log(content)
    ''''错误日志，报错的时候可以使用，加上了堆栈的形式'''
    def error(self,msg=''):
        t=time.strftime('%Y-%m-%d %H:%M:%S')
        print('Error at:',t)
        trace=traceback.format_exc()
        #获取调用者信息
        invoker_file,invoker_func=self.get_invoker_info()
        formator='Error at '+self.default_formator+'{4}'
        content=formator.format(t,os.path.basename(invoker_file),invoker_func,msg,trace)
        self.append_log(content)
        
    def append_log(self,content):
        f=open(self.logfile,'a',encoding='utf-8')
        f.write(content)
        #f.write('
')
        f.close()

logger=MyLogger()

用法示例

logger_test.py

from MyLogger import logger

def test():
    try:
        b=0
        a=1/b
    except Exception as e:
        #logger.error('UU',exc_info=True)
        logger.error(e)
        logger.log('calling exception')

if __name__=='__main__':
    test()
    logger.log('info')

打印的日志在log/myapp.log中的内容如下：

Error at 2019-12-31 19:18:48,File logger_test.py,test():division by zero
Traceback (most recent call last):
  File "/Users/admin/Desktop/code/spyder_workspace/logger_test.py", line 13, in test
    a=1/b
ZeroDivisionError: division by zero
Log at 2019-12-31 19:18:48,File logger_test.py,test():calling exception
Log at 2019-12-31 19:18:48,File logger_test.py,<module>():info