python工具类--打印日志

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2019/7/16 21:16
# @Author  : herry
# @Site    : 
# @File    : logger.py
# @Software: PyCharm




import logging
import time
import os


# 日志


# log_path是存放日志的路径
cur_path = os.path.dirname(os.path.realpath(__file__))
log_path = os.path.join(os.path.dirname(cur_path), 'logs')


# 如果不存在这个logs文件夹，就自动创建一个
if not os.path.exists(log_path):
    os.mkdir(log_path)




class Log():
    def __init__(self, logger=None):
        '''
            指定保存日志的文件路径，日志级别，以及调用文件
            将日志存入到指定的文件中
        '''


        # 创建一个logger
        self.logger = logging.getLogger(logger)
        self.logger.handlers.clear()  ##先清理已有Handler，避免重复添加，导致日志重复输出
        self.logger.setLevel(logging.INFO)
        # 创建一个handler，用于写入日志文件
        self.log_time = time.strftime("%Y_%m_%d_")
        self.log_path = cur_path
        self.log_name = os.path.join(log_path, '%s.log' % time.strftime('%Y_%m_%d'))


        fh = logging.FileHandler(self.log_name, 'a', encoding='utf-8')  # 追加模式
        fh.setLevel(logging.INFO)


        # 再创建一个handler，用于输出到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.INFO)


        # 定义handler的输出格式
        formatter = logging.Formatter(
            '[%(asctime)s] %(filename)s->%(funcName)s line:%(lineno)d [%(levelname)s]%(message)s')
        fh.setFormatter(formatter)
        ch.setFormatter(formatter)


        # 给logger添加handler
        self.logger.addHandler(fh)
        self.logger.addHandler(ch)
        # 关闭打开的文件
        fh.close()
        ch.close()






    def getlog(self):
        return self.logger




if __name__ == "__main__":
    log = Log().getlog()
    log.info("---测试开始----")
    log.info("操作步骤1,2,3")
    log.warning("----测试结束----")

-------------------------------------最后---------------------------------

 更多软件测试相关内容请关注“软件测试道与术”公众号或扫描下方二维码