Python 实现 Redis ORM
议题
我们来为 Redis 写一个简单而优雅的 ORM。这篇文章的灵感来自于 Django ORM。
这篇文章假定你对 Redis 以及 Python 中的 redis 库 redis-py 有了基本的了解。
实体
假设我们正在开发一个轮询应用程序,这个应用包括 Question 和 Choice。每一个问题都有多个选项。
我们希望在我们的应用程序中具有以下能力:
存储问题列表并检索根据 ID 检索问题存储选项关联问题与选项列表将问题与选项取消关联检索一个问题的所有选项跟踪一个选项的投票数我们希望将 Redis 作为我们的数据库使用。
模型
我们的模型主要就是Question 和 Choice。由于这两个模型与其他模型都有一些相似的功能,因此我们可以创建一个基类 Model,Question 与 Choice 都继承自这个基类 Model。
基础模型
Model 应该是这样的:
类 Question 如下:
我们来为 Model 的不同方法添加实现。
latest_instance_ID_key:
List_key:
add_to_List:
add_to_List 使用了 Redis 中的列表将所有实例的 key 存入 redis。我么这里使用 redis-py 中的 lpush 方法,它会映射到 redis 中的 LPUSH *** 作。
稍后我们会添加生成 self.ID 的代码。
确保 connection 定义在 models.py 模块内是可访问的。
latest_instance_ID:
increment_latest_instance_ID:
cache_key
save:
注意我们这里在 save() 方法中用的是 self.repr() 方法。它可以调用子类中定义的 repr() 方法,返回一个我们要保存的属性的字典。这个稍后解释。
你需要关注的是 save() 中是怎样调用 increment_latest_instance_ID() 和 add_to_List() 的。
save() 方法还用到了 redis-py 中的 hmset 方法,其对应 redis 中的 HMSET 命令。通过这种方式保存实例中我们需要的属性。
子类模型
现在我们来实现类 Question。
由于子类的 save 方法使用到了 repr() 方法,所以其每一个子类都要实现 repr() 方法。repr() 应该返回的是我们希望在 redis 中持久化的字典形式的数据。比如我们希望一个问题的 ID 以及 question_text 保存在 redis 中,那么 repr() 返回的字典中应该包含这些属性。
当我们试着使用 save() 保存第一个实例的时候,其内部会调用 latest_instance_ID_key() 方法,期望能从 redis 取出已经存在的键 question-latest-ID。
我们来创建一个文件 migration 并向 redis 中保存一个键。
运行这个文件:
使用 ORM
通过 ipython 运行我们的代码。
首先确保你的 redis 服务器 redis-server 已经启动了。
现在启动一个 redis 客户端 redis-cli,然后校验一下 question-1 是否已经插入到 redis 中了。
这里同样还校验了 questions 的存在。再来看看 ID 有没有保存到 questions 中。
现在再来校验一下 ID 和 question_text 是否也被保存在 questions 中。
我们再来用 ipython 创建一个 question 并将其保存在 redis 中。
现在看看我们期望的键值是否都保存到 redis 中了。
提取问题
我们添加一个 Question 方法来根据给出的 ID 提取一个问题。
注意这里我们传入了 ID 作为 cache_key。相应地我们就需要修改 cache_key 的代码:
现在来使用之前添加的 questions 的 IDs 来提取它们。
不过需要确保 save 功能可用。
非常棒!
提取所有问题
我们再添加一个方法来提取所有问题。
现在使用这个方法提取问题。
get_question 和 get_questions 拥有相似的功能并且可以使用我们稍后添加的模型。所以,现在将它们移到基类中去。将 get_question 重命名为 get,将 get_questions 重命名为 List。
现在就类用一下 get() 和 List() 方法,并校验一下其功能是否符合预期。
其他子类模型
现在来添加一个 Choice 模型。
在保存 choice 模型之前,我们需要添加一个键 choice-latest-ID。还记得我们之前的 migrations.py 文件吗?
在 ipython 中执行。
现在我们通过 Choice 来演示 ORM *** 作。
这应该归功于我们的基类。通过 Choice 我们可以自由地使用 save(),get(),List() 等方法。
英文原文:https://www.agiliq.com/blog/2019/11/writing-an-orm-for-redis/
译者:居老师的龙尾巴
以上是内存溢出为你收集整理的Python 实现 Redis ORM全部内容,希望文章能够帮你解决Python 实现 Redis ORM所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)