python讀取ini配置的類封裝

此爲基礎封裝，未考慮過多異常處理

 

類

# coding:utf-8
import configparser
import os

class IniCfg():
    def __init__(self):
        self.conf = configparser.ConfigParser()
        self.cfgpath = ''

    def checkSection(self, section):
        try:
            self.conf.items(section)
        except Exception:
            print(">> 無此section，請覈對[%s]" % section)
            return None
        return True

    # 讀取ini，並獲取全部的section名
    def readSectionItems(self, cfgpath):
        if not os.path.isfile(cfgpath):
            print(">> 無此文件，請覈對路徑[%s]" % cfgpath)
            return None
        self.cfgpath = cfgpath
        self.conf.read(cfgpath, encoding="utf-8")
        return self.conf.sections()

    # 讀取一個section，list裏面對象是元祖
    def readOneSection(self, section):
        try:
            item = self.conf.items(section)
        except Exception:
            print(">> 無此section，請覈對[%s]" % section)
            return None
        return item

    # 讀取一個section到字典中
    def prettySecToDic(self, section):
        if not self.checkSection(section):
            return None
        res = {}
        for key, val in self.conf.items(section):
            res[key] = val
        return res

    # 讀取全部section到字典中
    def prettySecsToDic(self):
        res_1 = {}
        res_2 = {}
        sections = self.conf.sections()
        for sec in sections:
            for key, val in self.conf.items(sec):
                res_2[key] = val
            res_1[sec] = res_2.copy()
            res_2.clear()
        return res_1

    # 刪除一個 section中的一個item（以鍵值KEY爲標識）
    def removeItem(self, section, key):
        if not self.checkSection(section):
            return
        self.conf.remove_option(section, key)

    # 刪除整個section這一項
    def removeSection(self, section):
        if not self.checkSection(section):
            return
        self.conf.remove_section(section)

    # 添加一個section
    def addSection(self, section):
        self.conf.add_section(section)

    # 往section添加key和value
    def addItem(self, section, key, value):
        if not self.checkSection(section):
            return
        self.conf.set(section, key, value)

    # 執行write寫入, remove和set方法並無真正的修改ini文件內容，只有當執行conf.write()方法的時候，纔會修改ini文件內容
    def actionOperate(self, mode):
        if mode == 'r+':
            conf.write(open(self.cfgpath, "r+", encoding="utf-8"))   # 修改模式
        elif mode == 'w':
            conf.write(open(self.cfgpath, "w"))                      # 刪除原文件從新寫入
        elif mode == 'a':
            conf.write(open(self.cfgpath, "a"))                      # 追加模式寫入




cfgpath = r'C:\Users\SXF\Desktop\config.ini'

inicfg = IniCfg()
sections = inicfg.readSectionItems(cfgpath)
print(sections)
content = inicfg.readOneSection('chaoji')
print(content)
dic = inicfg.prettySecToDic('chaoji')
print(dic)
dic = inicfg.prettySecsToDic()
print(dic)
inicfg.addSection('chaoji22')

content = inicfg.readOneSection('chaoji')
print(content)

 

 

測試ini

[chaoji]

chaoji_username = 123

chaoji_password = 456


[my]

soft_id         = 789

sleeptime         = asd

cnt_count         = zxc