Python基础之logging日志模块封装

发表于:2022-3-09 09:44

字体: | 上一篇 | 下一篇 | 我要投稿

 作者:全栈测试开发日记    来源:CSDN

  引言
  使用python做自动化测试的时候,对于关键信息输出和记录是必要的,方便调试和记录查找。python有个logging模块,我们简单对其进行封装。
  设计流程
  话不多说,流程图已经画,自己领会。
   
  源码
  """
  封装log方法
  """
   
  import os
  import time
  import logging.handlers
   
  # 日志打印等级
  LEVELS = {
      'debug': logging.DEBUG,
      'info': logging.INFO,
      'warning': logging.WARNING,
      'error': logging.ERROR,
      'critical': logging.CRITICAL
  }
   
  # 创建一个日志
  logger = logging.getLogger()
  level = 'default'
   
   
  # 创建日志文件方法
  def create_file(filename):
      path = filename[0:filename.rfind('/')]
      if not os.path.isdir(path):
          os.makedirs(path)
      if not os.path.isfile(filename):
          fd = open(filename, mode='w', encoding='utf-8')
          fd.close()
      else:
          pass
   
   
   
  # 给logger添加handler 添加内容到日志句柄中
  def set_handler(levels):
      if levels == 'error':
          logger.addHandler(MyLog.err_handler)
      logger.addHandler(MyLog.handler)
   
  # 在记录日志之后移除句柄
  def remove_handler(levels):
      if levels == 'error':
          logger.removeHandler(MyLog.err_handler)
      logger.removeHandler(MyLog.handler)
   
   
  def get_current_time():
      return time.strftime(MyLog.date, time.localtime(time.time()))
   
   
  class MyLog:
      path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
      log_file = path+'/Log/log.log'
      err_file = path+'/Log/err.log'
      logger.setLevel(LEVELS.get(level, logging.NOTSET))
      create_file(log_file)
      create_file(err_file)
      date = '%Y-%m-%d %H:%M:%S'
   
      # 创建一个handler,用于写入日志文件
      handler = logging.FileHandler(log_file, encoding='utf-8')
      err_handler = logging.FileHandler(err_file, encoding='utf-8')
   
      @staticmethod
      def debug(log_meg):
          set_handler('debug')
          # 文件中输出模式
          logger.debug("[DEBUG " + get_current_time() + "]" + log_meg)
          remove_handler('debug')
   
      @staticmethod
      def info(log_meg):
          set_handler('info')
          logger.info("[INFO " + get_current_time() + "]" + log_meg)
          remove_handler('info')
   
      @staticmethod
      def warning(log_meg):
          set_handler('warning')
          logger.warning("[WARNING " + get_current_time() + "]" + log_meg)
          remove_handler('warning')
   
      @staticmethod
      def error(log_meg):
          set_handler('error')
          logger.error("[ERROR " + get_current_time() + "]" + log_meg)
          remove_handler('error')
   
      @staticmethod
      def critical(log_meg):
          set_handler('critical')
          logger.error("[CRITICAL " + get_current_time() + "]" + log_meg)
          remove_handler('critical')
   
      # 设置控制台输出格式
      formatter = logging.Formatter(
          '[%(asctime)s] [%(levelname)s] %(message)s', '%Y-%m-%d %H:%M:%S')
      # 再创建一个handler,用于输出到控制台
      console = logging.StreamHandler()
      console.setFormatter(formatter)
      logger.addHandler(console)
      console.setLevel(logging.NOTSET)
   
   
  if __name__ == "__main__":
      MyLog.debug("This is debug message")
      MyLog.info("This is info message")
      MyLog.warning("This is warning message")
      MyLog.error("This is error")
      MyLog.critical("This is critical message")

    
  运行结果:
  [2020-05-23 20:41:10] [DEBUG] [DEBUG 2020-05-23 20:41:10]This is debug message
  [2020-05-23 20:41:10] [INFO] [INFO 2020-05-23 20:41:10]This is info message
  [2020-05-23 20:41:10] [WARNING] [WARNING 2020-05-23 20:41:10]This is warning message
  [2020-05-23 20:41:10] [ERROR] [ERROR 2020-05-23 20:41:10]This is error
  [2020-05-23 20:41:10] [ERROR] [CRITICAL 2020-05-23 20:41:10]This is critical message
   
  Process finished with exit code 0

  文件结构:

  文件内容:

  总结
  以上就是python日志的简单封装,基本上满足日常的需求。

  本文内容不用于商业目的,如涉及知识产权问题,请权利人联系51Testing小编(021-64471599-8017),我们将立即处理
《2023软件测试行业现状调查报告》独家发布~

关注51Testing

联系我们

快捷面板 站点地图 联系我们 广告服务 关于我们 站长统计 发展历程

法律顾问:上海兰迪律师事务所 项棋律师
版权所有 上海博为峰软件技术股份有限公司 Copyright©51testing.com 2003-2024
投诉及意见反馈:webmaster@51testing.com; 业务联系:service@51testing.com 021-64471599-8017

沪ICP备05003035号

沪公网安备 31010102002173号