【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,用来初始化嵌套较深的对象时可能有用。

你可能感兴趣的:(Python,python,json,数据库)