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"))