抽象一个类吧# coding: utf-8
import json
import os
class ConfigController:
"""配置控制器"""
def __init__(self, path):
"""path 为配置文件的路径,不存在则创建,存在则载入"""
self.__path = path
self.__config = dict()
self.__loadconfig()
def adddata(self, zh, mac, bz):
"""通过此方法添加数据"""
if zh in self.__config:
raise Exception("zh 数据已存在") # 也可将报错改为不添加数据
for zhres in self.__config.values():
if mac in zhres:
raise Exception("mac 数据已存在") # 也可将报错改为不添加数据
self.__config[zh] = [mac, bz]
# 如果想添加成功一次数据保存一次,此处可执行 saveconfig 方法
# self.saveconfig()
def deldata(self, zh):
"""通过此方法删除数据"""
del self.__config[zh]
def cleardata(self):
"""通过此方法清空数据"""
self.__config.clear()
def getdata(self, zh, default = None):
"""通过此方法访问数据"""
if default is None:
default = list() # 访问不存在的 zh 返回空列表
return self.__config.get(zh, default)
# 也可改为访问不存在的 zh 直接报错
def saveconfig(self):
"""通过此方法将数据储存到磁盘"""
with open(self.__path, "wt", encoding="utf8") as f:
json.dump(self.__config, f, ensure_ascii=False)
def __loadconfig(self):
"""通过此方法从磁盘载入数据"""
if not os.path.exists(self.__path):
with open(self.__path, "wt", encoding="utf8") as f:
json.dump(self.__config, f, ensure_ascii=False)
return
if self.__config: # 字典已有数据
return
if os.path.isfile(self.__path):
with open(self.__path, "rt", encoding="utf8") as f:
self.__config = json.load(f)
return
else:
raise Exception("配置文件路径是个文件夹")
def delete(self):
"""通过此方法删除配置文件"""
os.remove(self.__path)
# 配置控制器 ConfigController 的使用
config = ConfigController("./config.json")
# 添加数据
config.adddata("zh1", "a0:b6:82:79:63", "bz1")
# 添加重复 zh1:报错
# config.adddata("zh1", "00:00:00:00:00", "bz1")
# 添加重复 mac:报错
# config.adddata("zh2", "a0:b6:82:79:63", "bz1")
# 读取数据
print("zh1 的内容:",config.getdata("zh1"))
# 保存到配置文件
config.saveconfig()
|