Python脚本日志系统

Python通过logging模块提供日志功能，关于logging模块的使用网络上已经有很多详细的资料，这里要分享的是怎样在实际工程中使用日志功能。

假设要开发一个自动化脚本工具，工程结构如下，Common这个package是框架功能的实现，Scripts目录是我们编写的测试用例脚本（请忽略其他不相关的目录）。

我们对日志功能的需求如下：

1 为了便于日志的查看，每个脚本对应一个日志文件，日志文件以脚本的名字命名

2 日志路径以及每个脚本保存的日志容量可以设置，比如设置为5MB，则超过后最老日志被自动覆盖

3 日志功能要使用方便，减少与框架业务功能的耦合

现在来逐一分析上述需求。

1 要实现每个脚本一个日志文件，则需要在日志模块中，根据用例脚本的名字来生成日志文件，这里的关键问题就是怎样在日志模块中获取用例脚本的名字。

获取文件名的常用方法有：os.getcwd(), sys.argv[0], file，来看下各种的作用：

先在一个文件(假设为test.py）中编写如下代码：

然后在另一个文件中（假设为script1.py）中import test，然后调用func方法：

运行script1.py，结果为：

可见，os.getcwd()获取的是执行脚本的目录， sys.argv[0]是执行脚本的绝对路径名， __file__是被执行代码所在文件的绝对路径名。

现在就清楚了，我们应该用sys.argv[0]来获取执行脚本的名字，由于获取到的是绝对路径，需要做一点处理：sys.argv[
0].split(
'/')[-
1].split(
'.')[
0]

2 日志容量问题，要实现超过容量后自动覆盖最老日志，采用logging中的
RotatingFileHandler类即可，可以设置日志文件的大小，以及备份个数。

那么日志路径和容量配置放在哪里呢？让用户直接修改RotatingFileHandler的参数显然不好，最好不要让用户修改框架文件，用户只需调用接口写自己的脚本即可。

这里采用的方案是将配置信息写入一个文件，XML文件比较适合用来作为配置文件，用户通过修改XML文件来制定配置，日志模块从XML文件读取参数。

这里为了方便将XML文件放入Common下面，命名为config.xml，内容为：

1
2
3
4
5
6
7
8
9
10
11
12
13
1<?xml version="1.0" encoding="utf-8"?>

2

3<config>

4    <!--  日志保存路径  -->

5    <logpath>E:\PythonLog</logpath>

6

7    <!-- 每个脚本对应的日志文件大小，单位MB -->

8    <logsize>8</logsize>

9

10    <!-- 每个脚本保存的日志文件个数 -->

11    <lognum>3</lognum>

12</config>

13

读取XML文件内容，使用lxml库非常简单，后面再给出代码。

3 日志功能要使用方便，减少与框架业务功能的耦合，最好就是对日志功能进行封装，只提供记录日志的接口即可。

日志接口采用类方法的形式就可以满足上述要求，用户只需要通过类调用日志记录接口，随处调用，使用方便，并且无需定义类实例，与框架业务没有耦合。

有了上述分析，我们来实现日志模块。

由于日志功能也是框架基础的一部分，我们将日志模块也放在Common这个package中，在Common下新建log.py文件，代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
11 # coding: utf-8

2 2 

3 3 from lxml import etree

4 4 import logging.handlers

5 5 import logging

6 6 import os

7 7 import sys

8 8 

9 9 # 提供日志功能

1010 class logger:

1111     # 先读取XML文件中的配置数据

1212     # 由于config.xml放置在与当前文件相同的目录下，因此通过 __file__ 来获取XML文件的目录，然后再拼接成绝对路径

1313     # 这里利用了lxml库来解析XML

1414     root = etree.parse(os.path.join(os.path.dirname(__file__), 'config.xml')).getroot()

1515     # 读取日志文件保存路径

1616     logpath = root.find('logpath').text

1717     # 读取日志文件容量，转换为字节

1818     logsize = 1024*1024*int(root.find('logsize').text)

1919     # 读取日志文件保存个数

2020     lognum = int(root.find('lognum').text)

2121 

2222     # 日志文件名：由用例脚本的名称，结合日志保存路径，得到日志文件的绝对路径

2323     logname = os.path.join(logpath, sys.argv[0].split('/')[-1].split('.')[0])

2424 

2525     # 初始化logger

2626     log = logging.getLogger()

2727     # 日志格式，可以根据需要设置

2828     fmt = logging.Formatter('[%(asctime)s][%(filename)s][line:%(lineno)d][%(levelname)s] %(message)s', '%Y-%m-%d %H:%M:%S')

2929 

3030     # 日志输出到文件，这里用到了上面获取的日志名称，大小，保存个数

3131     handle1 = logging.handlers.RotatingFileHandler(logname, maxBytes=logsize, backupCount=lognum)

3232     handle1.setFormatter(fmt)

3333     # 同时输出到屏幕，便于实施观察

3434     handle2 = logging.StreamHandler(stream=sys.stdout)

3535     handle2.setFormatter(fmt)

3636     log.addHandler(handle1)

3737     log.addHandler(handle2)

3838 

3939     # 设置日志基本，这里设置为INFO，表示只有INFO级别及以上的会打印

4040     log.setLevel(logging.INFO)

4141 

4242     # 日志接口，用户只需调用这里的接口即可，这里只定位了INFO, WARNING, ERROR三个级别的日志，可根据需要定义更多接口

4343     @classmethod

4444     def info(cls, msg):

4545         cls.log.info(msg)

4646         return

4747 

4848     @classmethod

4949     def warning(cls, msg):

5050         cls.log.warning(msg)

5151         return

5252 

5353     @classmethod

5454     def error(cls, msg):

5555         cls.log.error(msg)

5656         return

57

来测试一下，在脚本script1和script2中分别编写下面代码：

1
2
3
4
5
6
1from Common.log import *

2

3logger.info('This is info')

4logger.warning('This is warning')

5logger.error('This is error')

6

分别运行两个脚本，控制台输出为：

产生的日志文件：

文件内容：

好了，现在不管是在框架的其他文件中，或是在用户脚本中，都可以方便的通过logger类的日志接口记录日志。