挨T砖家技术教程 2019-12-02

Python利用ConfigParser读写配置文件

在自动化测试过程中，为了提高脚本的可读性和降低维护成本，没必要一直去修改源代码。

将需要重复使用的方法写成公共模块进行封装，需要使用时候直接调用即可。

在Python中可以利用ConfigParser模块来读写配置文件，在程序中使用配置文件来灵活的配置一些参数。

ConfigParser模块在Python3修改为configparser，这个模块定义了一个ConfigeParser类，该类的作用是让配置文件生效。

ConfigParser简介

我们新建一个.ini文件

格式如下：中括号“[ ]”内包含的为section。section 下面为类似于key-value 的配置内容。

[db]
db_host = 127.0.0.1
db_port = 66
db_user = root
db_pass = root
host_port = 66

[login]
user = 123
pass = 456

ConfigParser 初始化

使用ConfigParser 首选需要初始化实例，并读取配置文件：

import configparser
config = configparser.ConfigParser()
config.read("xxx.ini")

ConfigParser 读文件的方法：

read(filename)：读取ini文件中的内容
sections()：获取所有section，返回列表形式
options(section)：得到给定section的所有option
items(section):得到指定section的所有key-value
get(section,option)：得到section中的option值，返回str类型
getint(section,option)：得到section中的option值，返回int类型
getfloat(section,option)：得到section中的option值，返回float类型
getboolean(section,option)：得到section中的option值，返回boolean类型
get函数fullback传值，如果option不存在返回fullback的值，否则抛出异常

ConfigParser 写文件的方法：

add_section(sectionname)：添加一个名为sectionname的新section
set(sectionname,option,value):设置sectionname的option和value的值

演示代码：
1、获取所有的section节点

# 获取所有的section节点
import configparser
config = configparser.ConfigParser()
config.read("xxx.ini")
print(config.sections())
#运行结果
# ['db', 'login']

2、获取指定section 的options 就是把指定section中的key读取到列表中：

import configparser
config = configparser.ConfigParser()
config.read("xxx.ini")
r = config.options("db")
print(r)
#运行结果
# ['db_host', 'db_port', 'db_user', 'db_pass', 'host_port']

3、获取指点section下指点option的值

import configparser
config = configparser.ConfigParser()
config.read("xxx.ini")
r = config.get("db", "db_host")  #将获取到值转换为str型
# r = config.getint("db", "db_host") #将获取到值转换为int型
# r = config.getboolean("db", "db_host") #将获取到值转换为bool型
# r = config.getfloat("db", "db_host") #将获取到值转换为浮点型
print(r)
#运行结果
# 127.0.0.1

4、获取指点section的所有配置信息

import configparser
config = configparser.ConfigParser()
config.read("xxx.ini")
r = config.items("db")
print(r)
#运行结果
#[('db_host', '127.0.0.1'), ('db_port', '66'), ('db_user', 'root'), ('db_pass', 'root'), ('host_port', '66')]

5、设置某个option的key，value值，如果不存在则会创建一个

# 修改某个option的值，如果不存在该option 则会创建
import configparser
config = configparser.ConfigParser()
config.read("xxx.ini")
config.set("db", "db_port", "69") #修改db_port的值为69
config.write(open("xxx.ini", "w")) # 写入文件xxx.ini

6、添加section 和 option

import configparser
config = configparser.ConfigParser()
config.read("xxx.ini")
if not config.has_section("default"): # 检查是否存在section
   config.add_section("default")
if not config.has_option("default", "db_host"): # 检查是否存在该option
   config.set("default", "db_host", "1.1.1.1")
config.write(open("xxx.ini", "w"))