【Python】用 dict 实现一个简单的 json 数据库

废话不看：
最近写了一个小项目，基本逻辑是通过定时任务轮询 API，检测 API 状态变化并执行对应的操作。该 API 一共有 3 种状态，假设每种状态的值分别为 1、2、3，在状态 2 的时候需要调用一次处理方法。
这里的问题是状态 2 可能持续很长时间，而通过定时任务调用的 Python 脚本默认是没有记录状态的，因此需要本地保存状态。可以用 SQLite 或者 MangoDB，但是用到这种小项目上有点多余的感觉，最轻量级的解决方法大概还是用 json 保存。

实现

取 dict 的时候，如果 key 不在 dict 中，会抛出异常，先重写 __getitem__ 避免 key 不在 dict 中时的报错，然后在 __setitem__ 的时候，将 dict 写入到本地，这样就完成了数据的无感写入。

class ReactiveDict(dict):
    def __getitem__(self, __key: any) -> any:
        if __key not in self:
            self[__key] = ReactiveDict()
            return self[__key]

        return super().__getitem__(__key)

    def __setitem__(self, __key: any, __value: any) -> None:
        data = super().__setitem__(__key, __value)

        with open(rootpath, 'w', encoding='utf-8') as f:
            json.dump(reactive_dict, f)

        return data

reactive_dict = ReactiveDict()
reactive_dict["user"]["profile"]["name"] = "abc"

当然这样写的一个问题是可能会导致非常频繁的 I/O 操作，后续可以优化成类似 JavaScript 的防抖，使只有最后一次 set 写入到本地。

突然想到用 json.loads 可以直接将 JSON字符串转为 dict，用来初始化嵌套较深的对象时可能有用。

代码007普通

打赏 收藏 海报 链接