基于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