python：logging模块封装，快速添加日志

小文大数据

已于 2023-05-08 09:50:38 修改

阅读量216

点赞数

文章标签： python 开发语言

于 2023-04-25 11:38:32 首次发布

本文链接：https://blog.csdn.net/weixin_46707493/article/details/130361470

版权

import os
import sys

BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, os.path.join(BASE_DIR))
import logging
from logging.handlers import TimedRotatingFileHandler

parent = os.path.dirname(os.path.realpath("__file__"))
garder = os.path.dirname(parent)

class CreateLog(object):
    def get_log(self, str, level='DEBUG'):
        log_dir = parent + '/logs'
        if not os.path.exists(log_dir):
            os.mkdir(log_dir)

        LOG_FORMAT = "%(asctime)s - %(name).30s %(thread)d - %(levelname)s - %(message)s"
        DATE_FORMAT = "%Y/%m/%d %H:%M:%S"
        filename = os.path.join(log_dir, '{}.log'.format(str))

        my_log = logging.getLogger(str)
        rHandler = TimedRotatingFileHandler(filename, when='D', interval=1, backupCount=5)
        # rHandler = RotatingFileHandler(filename, maxBytes=268435456, backupCount=8)

        level = getattr(logging, level)
        my_log.setLevel(level)
        rHandler.setLevel(level)

        formatter = logging.Formatter(fmt=LOG_FORMAT, datefmt=DATE_FORMAT)
        rHandler.setFormatter(formatter)
        my_log.addHandler(rHandler)
        sh = logging.StreamHandler()
        sh.setFormatter(fmt=formatter)
        my_log.addHandler(rHandler)
        my_log.addHandler(sh)
        return my_log


if __name__ == '__main__':
    create_log = CreateLog()
    logger = create_log.get_log('test3','INFO')
    logger.info('start')