Python写自动化之ini文件的读写

在我们写一些测试工具时，经常会需要使用到配置文件，此时一般选用ini文件是比较合适的

标准并且可以通过参数命名知道参数的含义，那么使用Python如何进行ini文件的读写呢？

首先看下，读取ini文件，我们直接使用Python自带的模块ConfigParser来进行配置文件的读写

看下代码吧

def ReadConfig(configPath):
    configDict = {}
    cf = ConfigParser.ConfigParser()
    cf.read(configPath)
    sessions = cf.sections()
    for session in sessions:
        options = cf.options(session)
        for option in options:
            key = session + "." + option
            value = cf.get(session, option)
            configDict[key.lower()] = value
    return configDict

这样通过上面的接口，我们就可以把配置文件读取成session.option=value的一个字典，想要使用时，直接传入相应的Key即可

再看下ini文件的写入吧，传入参数：要写入的配置文件路径，session节点，option节点，value值，为了与读取接口一致，我们采用key，value的方式来写入

代码如下：

def WriteConfig(configPath, key, value):
    if key.find(".") == -1:
        return
    wSession = key.split(".")[0]
    wOption = key.split(".")[1]

    cf = ConfigParser.ConfigParser()
    cf.read(configPath)
    cf.set(wSession, wOption, value)
    cf.write(open(configPath, "w"))

这样就可以在Python中修改ini文件了~

转载请注明：http://blog.youkuaiyun.com/sogouauto

探索测试奥义，交流测试技术，欢迎关注“搜狗测试”