0x03 接口自动化测试-操作配置文件的工具类

自动化测试过程中，为了方便使用，会配置一些环境信息、数据库信息、账号信息等，故此封装了一个操作配置文件的工具类。

类中实现了对配置信息的读取和写入。

以下是V1版本，后续完善。

# -*- coding:UTF-8 -*-
# @author  : Joker
# @Time    : 2019/12/24
# @IDE     : PyCharm
# @Version : Python 3.7

"""操作配置文件"""

import os
from configparser import ConfigParser
from common.handle_path import CONF_DIR


class Conf:

    def __init__(self, filename, encoding="utf8"):
        self.filename = filename
        self.encoding = encoding
        # 创建文件解析器对象，并使用解释器对象加载配置文件内容
        self.conf = ConfigParser()
        self.conf.read(filename, encoding)

    def get_str(self, section, option):
        """获取字符串类型数据"""
        return self.conf.get(section, option)

    def get_int(self, section, option):
        """获取整型数值类型数据"""
        return self.conf.getint(section, option)

    def get_float(self, section, option):
        """获取浮点型数值数据"""
        return self.conf.getfloat(section, option)

    def get_bool(self, section, option):
        """获取布尔类型数据"""
        return self.conf.getboolean(section, option)

    def write_data(self, section, option, value):
        """写入数据到配置文件"""
        self.conf.set(section, option, value)
        self.conf.write(open(self.filename, "w", encoding=self.encoding))


# 创建实例对象方便其他模块直接使用
conf_path = os.path.join(CONF_DIR, "conf.ini")
conf = Conf(conf_path)


if __name__ == "__main__":
    print(conf.get_str("log", "level"))