高级技巧!Python装饰器里面的懒加载,真香!

共 5495字,需浏览 11分钟

 ·

2021-03-25 10:17

摄影:产品经理
禁止钓鱼

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

import pymongo

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

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

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

import pymongo

class MongoUtil:
    def __init__(self):
        connect = pymongo.MongoClient()
        self.db = connect.tieba
        self.post = None
        self.user = None
        
    def write_post(self, post):
        # 处理post信息
        if not self.post:
            self.post = self.db.post
        self.post.insert_one(post)
    
    def read_user_info(self):
        if not self.user:
            self.user = self.db.user
        rows = self.user.find()
        # 读取user信息并处理
        # ...

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

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

import pymongo

class lazy:
    def __init__(self, func):
        self.func = func

    def __get__(self, instance, cls):
        if instance is None:
            return self
        else:
            value = self.func(instance)
            setattr(instance, self.func.__name__, value)
            return value

class MongoUtil:
    def __init__(self):
        connect = pymongo.MongoClient()
        self.db = connect.tieba

    @lazy
    def post(self):
        return self.db.post
    
    @lazy
    def user(self):
        return self.db.user
    
    def write_post(self, post):
        # 处理post信息
        self.post.insert_one(post)
    
    def read_user_info(self):
        rows = self.user.find()
        # 读取user信息并处理
        # ...

我们实现了一个装饰器类lazy来装饰两个类属性postuser。当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: 会做诗的机器人 | 给图片上色 | 预测收入 | 碟中谍这么火,我用机器学习做个迷你推荐系统电影


年度爆款文案


点这里,直达菜鸟学PythonB站!!

浏览 54
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报