|
发表于 2022-9-4 21:56:32
|
显示全部楼层
抽象一个类吧
- # 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()
复制代码 |
|