Python自动化测试ConfigParser模块读写配置文件

Python自动化测试涉及到很多配置文件，如何方便读写配置文件成为了自动化测试中必不可少的一部分。Python自带的ConfigParser模块是一个用于读写配置文件的工具。

安装ConfigParser模块

ConfigParser模块是Python2.x的内置模块，如果你使用的是Python3.x版本，需要先安装此模块。

在命令行中执行以下命令即可安装：

pip install configparser

读取配置文件

在Python脚本中读取配置文件非常简单，只需要先导入ConfigParser模块，然后调用它的方法即可。

import configparser

config = configparser.ConfigParser()
config.read('config.ini')

print(config.get('section1', 'key1'))
print(config.get('section1', 'key2'))
print(config.get('section2', 'key1'))

在本例子中，使用ConfigParser读取config.ini配置文件，打印出了section1和section2中各自的key。

写入配置文件

在Python脚本中写入配置文件也非常简单，只需要通过ConfigParser的方法增加或修改配置项后再写入到文件中即可。

import configparser

config = configparser.ConfigParser()

config.add_section('section1')
config.set('section1', 'key1', 'value1')
config.set('section1', 'key2', 'value2')

config.add_section('section2')
config.set('section2', 'key1', 'value3')

with open('config.ini', 'w') as f:
    config.write(f)

在本例子中，使用ConfigParser新增了section1和section2配置项，并设置了相应的key value，然后将其写入config.ini文件中。

示例1

我们以写入邮件发送人、密码、收件人配置为例，具体代码如下：

import configparser

config = configparser.ConfigParser()

config.add_section('email_config')
config.set('email_config', 'sender', '[e-mail]')
config.set('email_config', 'password', 'example_password')
config.set('email_config', 'recipient', '[e-mail]')

with open('email_config.ini', 'w') as f:
    config.write(f)

上述代码新建了email_config节，然后增加了三个子项，包括了发件人、密码和收件人，最后将其写入email_config.ini文件中。

示例2

我们以修改邮件发送人、密码示例为例，示例代码如下：

import configparser

config = configparser.ConfigParser()
config.read('email_config.ini')

config.set('email_config', 'sender', '[e-mail]')
config.set('email_config', 'password', 'new_password')

with open('email_config.ini', 'w') as f:
    config.write(f)

上述代码读取email_config.ini文件，并将发件人和密码修改成新的值，最后将配置文件写回到文件中。

以上就是使用ConfigParser模块对Python自动化测试中的配置文件进行读写的完整攻略。