<?xml version="1.0" encoding="utf-8"?>

<config>
  <!-- 日志保存路徑 -->
  <logpath>E:\PythonLog</logpath>

  <!-- 每個腳本對應的日志文件大小，單位MB -->
  <logsize>8</logsize>

  <!-- 每個腳本保存的日志文件個數(shù) -->
  <lognum>3</lognum>
</config>

# coding: utf-8

from lxml import etree
import logging.handlers
import logging
import os
import sys

# 提供日志功能
class logger:
  # 先讀取XML文件中的配置數(shù)據(jù)
  # 由于config.xml放置在與當前文件相同的目錄下，因此通過 __file__ 來獲取XML文件的目錄，然后再拼接成絕對路徑
  # 這里利用了lxml庫來解析XML
  root = etree.parse(os.path.join(os.path.dirname(__file__), 'config.xml')).getroot()
  # 讀取日志文件保存路徑
  logpath = root.find('logpath').text
  # 讀取日志文件容量，轉(zhuǎn)換為字節(jié)
  logsize = 1024*1024*int(root.find('logsize').text)
  # 讀取日志文件保存?zhèn)€數(shù)
  lognum = int(root.find('lognum').text)

  # 日志文件名：由用例腳本的名稱，結(jié)合日志保存路徑，得到日志文件的絕對路徑
  logname = os.path.join(logpath, sys.argv[0].split('/')[-1].split('.')[0])

  # 初始化logger
  log = logging.getLogger()
  # 日志格式，可以根據(jù)需要設置
  fmt = logging.Formatter('[%(asctime)s][%(filename)s][line:%(lineno)d][%(levelname)s] %(message)s', '%Y-%m-%d %H:%M:%S')

  # 日志輸出到文件，這里用到了上面獲取的日志名稱，大小，保存?zhèn)€數(shù)
  handle1 = logging.handlers.RotatingFileHandler(logname, maxBytes=logsize, backupCount=lognum)
  handle1.setFormatter(fmt)
  # 同時輸出到屏幕，便于實施觀察
  handle2 = logging.StreamHandler(stream=sys.stdout)
  handle2.setFormatter(fmt)
  log.addHandler(handle1)
  log.addHandler(handle2)

  # 設置日志基本，這里設置為INFO，表示只有INFO級別及以上的會打印
  log.setLevel(logging.INFO)

  # 日志接口，用戶只需調(diào)用這里的接口即可，這里只定位了INFO, WARNING, ERROR三個級別的日志，可根據(jù)需要定義更多接口
  @classmethod
  def info(cls, msg):
    cls.log.info(msg)
    return

  @classmethod
  def warning(cls, msg):
    cls.log.warning(msg)
    return

  @classmethod
  def error(cls, msg):
    cls.log.error(msg)
    return

from Common.log import *

logger.info('This is info')
logger.warning('This is warning')
logger.error('This is error')