python3+requests接口自動化-日志封裝

1.logger.py這個文件放到common目錄下，封裝日志文件的讀取

2.日志保存到logs文件夾

3.封裝代碼

import logging
import os
import time


# log_path是日志存放路徑地址
get_path = os.path.dirname(os.path.abspath(__file__))
log_path = os.path.join(os.path.dirname(get_path),"log")


# 如果不存在這個logs文件夾，就自動創建一個
if not os.path.exists(log_path):os.mkdir(log_path)

class Log():
    def __init__(self):
        # 文件的命名
        self.logname = os.path.join(log_path,"%s.log"%time.strftime("%Y:%m:%d"))
        self.logger = logging.getLogger()
        self.logger.setLevel(logging.DEBUG)
        # 日志輸出格式
        self.formatter = logging.Formatter('[%(asctime)s] - %(filename)s] - %(levelname)s: %(message)s')

    def __console(self,level,message):

    # 創建一個FileHandler，用於寫到本地
        fh = logging.FileHandler(self.logname,"a",encoding='utf-8') # 追加模式
        fh.setLevel(logging.DEBUG)
        fh.setFormatter(self.formatter)
        self.logger.addHandler(fh)

        # 創建一個StreamHandler,用於輸出到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.DEBUG)
        ch.setFormatter(self.formatter)
        self.logger.addHandler(ch)

        if level == "info":
            self.logger.info(message)
        elif level == "debug":
            self.logger.debug(message)
        elif level == "warning":
            self.logger.warning(message)
        elif level == "error":
            self.logger.error(message)

        # 這兩行代碼是為了避免日志輸出重復問題
        self.logger.removeHandler(ch)
        self.logger.removeHandler(fh)

        # 關閉打開的文件
        fh.close()

    def debug(self,message):
        self.__console("debug",message)

    def info(self,message):
        self.__console("info",message)
    def warning(self,message):
        self.__console("warning",message)
    def error(self,message):
        self.__console("error",message)

if __name__ == "__main__":
    log = Log()
    log.info("--測試開始--")
    log.info("操作步驟1，2,3")
    log.warning("--測試結束--")