配置文件類的封裝

在作自動化測試的時候,無論是接口自動化仍是UI自動化,都須要把一些經常使用的數據存放到一個單獨的文件,其實就是所謂的配置文件。python

配置文件不只方便數據、參數的統一配置管理,同時也加強了代碼的可讀性、可維護性。測試

python中提供了ConfigParser類供咱們對配置文件進行操做blog

下面就針對配置文件的經常使用操做讀寫進行封裝接口

讀操做:根據提供的section和option讀取其中的值utf-8

寫操做:在section中寫入option的值get

from configparser import ConfigParser, NoSectionError, NoOptionError

class ConfigHandler:
    """封裝配置文件類
    一、讀取配置參數
    二、修改配置參數
    """
    def __init__(self, filename, encoding='utf-8'):
        self.filename = filename
        self.encoding = encoding
        self.config = ConfigParser()
        self.config.read(filename, encoding=encoding)

    def read(self, section, option):
        # 讀取配置文件的某一項
        try:
            return self.config.get(section, option)
        except NoSectionError:
            print('沒有這個section')
        except NoOptionError:
            print('沒有這個option')

    def write(self, section, option, value, mode='w'):
         # 往配置文件中寫操做
        if self.config.has_section(section):
            self.config.set(section, option, value)
            with open(self.filename, mode=mode, encoding=self.encoding) as f:
                self.config.write(f)

  根據自身的須要,該封裝類中還能夠增長其餘操做,好比獲取全部section,獲取section的全部option值等等。it

相關文章
相關標籤/搜索