python 封装logging

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Time    : 2020/9/25 9:55
# @Author  : cheng
# @Email   : 771535427@qq.com
# @File    : log.py
# @Software: PyCharm
import logging
from logging.handlers import RotatingFileHandler


class BasicLog(object):

    def __init__(self, level="INFO", filename="./log.txt", format="%(asctime)s - %(levelname)s - %(message)s"):
        """
        封装logging日志类
        :param level: 日志级别 [info, debug, warning, error, critical, fatal]
        :param filename: 日志存储文件
        :param format: 日志输出格式
        """
        self._level = getattr(logging, level.upper())
        self._filename = filename
        self._format = format
        self._logger = logging.getLogger(__name__)
        self._logger.setLevel(level=self._level)
        self.init_logging()

    def init_logging(self):
        logging.basicConfig(level=self._level, filename=self._filename, format=self._format)

    def info(self, message):
        self._logger.info(message)

    def debug(self, message):
        self._logger.debug(message)

    def warning(self, message):
        self._logger.warning(message)

    def error(self, message):
        self._logger.error(message)

    def critical(self, message):
        self._logger.critical(message)

    def fatal(self, message):
        self._logger.fatal(message)


class RotatingLog(BasicLog):
    def __init__(self, level, filename="./log.txt",
                 format="%(asctime)s - %(levelname)s - %(message)s",
                 maxBytes=(100 * 1024),
                 backupCount=3
                 ):
        """
        封装logging日志回滚类
        :param level: 日志级别 [info, debug, warning, error, critical, fatal]
        :param filename: 日志存储文件
        :param format: 日志输出格式
        :param maxBytes: 单个日志文件大小
        :param backupCount: 日志回滚数量
        """
        self._maxBytes = maxBytes
        self._backupCount = backupCount
        super().__init__(level=level, filename=filename, format=format)

    def init_logging(self):
        rotatingHandler = RotatingFileHandler(filename=self._filename, maxBytes=self._maxBytes,
                                              backupCount=self._backupCount)

        rotatingHandler.setFormatter(logging.Formatter(self._format))

        self._logger.addHandler(rotatingHandler)


if __name__ == "__main__":
    log = RotatingLog(level="debug")
    log.info("Start print log")
    log.debug("Do something")
    log.warning("Something maybe fail.")
    log.info("Finish")