python 装饰器的妙用,实现类属性的懒加载
生活随笔
收集整理的這篇文章主要介紹了
python 装饰器的妙用,实现类属性的懒加载
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
假設(shè)我們有一個工具類MongoUtil,它的作用是封裝一些數(shù)據(jù)庫操作。例如:
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信息并處理#?...我們發(fā)現(xiàn)這樣寫有一個問題——類在初始化的時候,就會創(chuàng)建數(shù)據(jù)庫的鏈接。但我們并不是在類剛剛初始化時就讀寫數(shù)據(jù)庫。
為了讓數(shù)據(jù)庫在第一次使用時再創(chuàng)建連接,我們就要實(shí)現(xiàn)懶加載機(jī)制:
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信息并處理#?...這樣寫確實(shí)實(shí)現(xiàn)了懶加載,但每一個操作都需要判斷當(dāng)前是否聯(lián)系到了對應(yīng)的集合中。這樣就會出現(xiàn)大量的重復(fù)代碼。
為了解決這個問題,我們可以使用裝飾器實(shí)現(xiàn)一個懶加載機(jī)制:
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信息并處理#?...我們實(shí)現(xiàn)了一個裝飾器類lazy來裝飾兩個類屬性post和user。當(dāng)self.post第一次被調(diào)用時,它會正常連接結(jié)合,當(dāng)?shù)诙位蛞陨显L問self.post時,就會直接使用第一次返回的對象,不會再次連接MongoDB的集合。self.user同理。
?
總結(jié)
以上是生活随笔為你收集整理的python 装饰器的妙用,实现类属性的懒加载的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 简单、易用的 MySQL 官方压测工具
- 下一篇: Python 工具链让你写的代码更规范