基于Python封装读取ini文件的工具类【接口自动化框架设计系列】【多测师】

#coding=utf-8
"""
===========================
Author:多测师_王sir
Time:2020/5/20 17:24
Company:上海多测师信息有限公司
===========================
"""

"""
配置文件类的封装
封装的目的：使用更简单
封装的需求：
1、简化创建配置文件解析器对象，加载配置文件的流程（需要封装），提示（重写init方法）
2、读取数据（不进行封装，使用原来的方法），通过继承父类即可
3、简化写入数据的操作（需要封装），提示：自定义一个write_data方法。

"""
import os
from configparser import ConfigParser
from common.handlepath import CONFDIR


class HandleConfig(ConfigParser):

    def __init__(self, filename):
        # 调用父类的init方法
        super().__init__()
        self.filename = filename
        self.read(filename,encoding="utf8")

    def write_data(self, section, options, value):
        """写入数据的方法"""
        self.set(section, options, value)
        self.write(fp=open(self.filename, "w"))


conf = HandleConfig(os.path.join(CONFDIR,"config.ini"))
print(conf.get("log","level"))   #拿到了log对应的字段DEBUG