基于Python的接口自动化实战-基础篇之读写配置文件

引言

在编写接口自动化测试脚本时，有时我们需要在代码中定义变量并给变量固定的赋值。为了统一管理和操作这些固定的变量，咱们一般会将这些固定的变量以一定规则配置到指定的配置文件中，后续需要用到这些变量和变量值时通过代码读取或者写入数据到该配置文件即可，使用配置文件的好处就是不用在程序员写死，可以使程序更灵活。因而对于python语言就封装了configparser模块，用来处理指定格式的文件(文件名称一般为xxx.ini)，配置文件的格式跟windows下的ini配置文件相似，可以包含一个或多个节(section), 每个节可以有多个参数（键=值）。

注：目前python3版本操作配置文件模块名称变更为configparser，原有的python2版本配置文件模块名称变更为ConfigParser

configparser模块功能介绍

configparser模块主要封装提供了ConfigParser()类、RawConfigParser()类、SafeConfigParser()类等，如下图导入configparser模块，可以查看并使用模块下已经封装好的所有类

在configparser模块中主要还是使用封装的ConfigParser()类提供的方法来操作配置文件，对配置文件进行数据读取和写入等。

具体的该模块python官网对其有完整的介绍：https://docs.python.org/3/library/configparser.html

配置文件的格式如下：

• 中括号“[ ]”内包含的为section。

• section 下面为类似于key-value 的配置内容，一般是用赋值符“=”分隔。

示例：

 读取配置文件及其常用读取指定数据方法

ConfigParser()类提供了操作读取配置文件、读取指定配置文件中section、section下的所有key-value等方法，具体方法使用如下：

代码示例：

import configparser

cf = configparser.ConfigParser() # 实例化

# read(filename) 读取文件
filename = cf.read(r'G:\api_test\configfile.ini',encoding='utf-8')

# sections() 得到所有的section，以列表形式返回
sec = cf.sections()
print(sec)

# 得到section下的所有option
opt = cf.options("mysql")
print(opt)

# items 得到section的所有键值对
value = cf.items("mysql")
print(value)
print(dict(value)) # 转成字典类型

# get(section,option) 得到section中的option值，返回string/int类型的结果
mysql_host = cf.get("mysql","host")
mysql_password = cf.getint("mysql","port")
print(mysql_host,mysql_password)

　　运行以上代码输出如下：

 写入配置文件（增删改操作）及其提供的操作方法

ConfigParser()类提供了写入修改配置文件数据等方法，具体方法使用如下：

在配置文件编辑如下信息：

编辑如下代码：

import configparser

conf = configparser.ConfigParser()
conf.read(r'G:\api_test\configfile.ini',encoding='utf-8')

conf.set("code", "code", "6666")  # 修改指定section 的option
conf.set("code", "age", "123")  # 增加指定section 的option
conf.has_section("code")
conf.has_option("code","age")
#conf.remove_section("test")
#conf.remove_option("test","haha")
if 'test' not in conf.sections():
    conf.add_section("test")  # 增加section
    conf.set("test", "haha", "123")  # 给新增的section 写入option
file = open(r'G:\api_test\configfile.ini', 'w',encoding='utf-8')
conf.write(file)
file.close()

运行结束后如下：

编辑修改配置文件记得关闭配置文件。

通过以上使用ConfigParser()类提供读写配置文件的相关方法，就可以比较方便的操作配置文件。后续可以将上述读写配置文件需要用到的方法封装到自定义类中，便于程序其他模块引用。