摄影：产品经理

禁止钓鱼

假设我们有一个工具类MongoUtil，它的作用是封装一些数据库操作。例如：

import pymongoclass MongoUtil:def __init__(self):connect = pymongo.MongoClient()db = connect.tiebaself.post = db.postself.user = db.userdef write_post(self, post):# 处理post信息self.post.insert_one(post)def read_user_info(self):rows = self.user.find()# 读取user信息并处理# ...

我们发现这样写有一个问题——类在初始化的时候，就会创建数据库的链接。但我们并不是在类刚刚初始化时就读写数据库。

为了让数据库在第一次使用时再创建连接，我们就要实现懒加载机制：

import pymongoclass MongoUtil:def __init__(self):connect = pymongo.MongoClient()self.db = connect.tiebaself.post = Noneself.user = Nonedef write_post(self, post):# 处理post信息if not self.post:self.post = self.db.postself.post.insert_one(post)def read_user_info(self):if not self.user:self.user = self.db.userrows = self.user.find()# 读取user信息并处理# ...

这样写确实实现了懒加载，但每一个操作都需要判断当前是否联系到了对应的集合中。这样就会出现大量的重复代码。

为了解决这个问题，我们可以使用装饰器实现一个懒加载机制：

import pymongoclass lazy:def __init__(self, func):self.func = funcdef __get__(self, instance, cls):if instance is None:return selfelse:value = self.func(instance)setattr(instance, self.func.__name__, value)return valueclass MongoUtil:def __init__(self):connect = pymongo.MongoClient()self.db = connect.tieba@lazydef post(self):return self.db.post@lazydef user(self):return self.db.userdef write_post(self, post):# 处理post信息self.post.insert_one(post)def read_user_info(self):rows = self.user.find()# 读取user信息并处理# ...

我们实现了一个装饰器类lazy来装饰两个类属性post和user。当self.post第一次被调用时，它会正常连接结合，当第二次或以上访问self.post时，就会直接使用第一次返回的对象，不会再次连接MongoDB的集合。self.user同理。

我们来测试一下，如下图所示。

可以看到，第二次调用self.post时，并没有打印出第一次访问self.post，因为第二次会直接使用之前的缓存。

最后，特别说明：本文使用MongoDB举例只是为了说明基于装饰器的类属性懒加载的代码写法。而实际上，pymongo已经自动实现了懒加载机制，当我们直接db.tieba.post时，它并不会真的去连接MongoDB，只有当我们要增删改查集合里面的数据时，pymongo才会创建连接。

推荐阅读:
入门: 最全的零基础学Python的问题  | 零基础学了8个月的Python  | 实战项目 |学Python就是这条捷径
干货:爬取豆瓣短评，电影《后来的我们》 | 38年NBA最佳球员分析 |   从万众期待到口碑扑街！唐探3令人失望  | 笑看新倚天屠龙记 | 灯谜答题王 ｜用Python做个海量小姐姐素描图 ｜
趣味:弹球游戏  | 九宫格  | 漂亮的花 | 两百行Python《天天酷跑》游戏!
AI: 会做诗的机器人 | 给图片上色 | 预测收入 | 碟中谍这么火，我用机器学习做个迷你推荐系统电影
年度爆款文案

• 1).卧槽！Pdf转Word用Python轻松搞定！

• 2).学Python真香！我用100行代码做了个网站，帮人PS旅行图片，赚个鸡腿吃

• 3).首播过亿，火爆全网，我分析了《乘风破浪的姐姐》，发现了这些秘密

• 4).80行代码！用Python做一个哆来A梦分身

• 5).你必须掌握的20个python代码，短小精悍，用处无穷

• 6).30个Python奇淫技巧集

• 7).我总结的80页《菜鸟学Python精选干货.pdf》,都是干货

• 8).再见Python！我要学Go了！2500字深度分析！

• 9).发现一个舔狗福利！这个Python爬虫神器太爽了，自动下载妹子图片

点这里，直达菜鸟学PythonB站！！