Python上下文管理器類和上下文管理器裝飾器contextmanager用法實(shí)例分析
本文實(shí)例講述了Python上下文管理器類和上下文管理器裝飾器contextmanager用法。分享給大家供大家參考,具體如下:
一. 什么是上下文管理器
上下文管理器是在Python2.5之后加入的功能,可以在方便的需要的時(shí)候比較精確地分配和釋放資源, with便是上下文管理器的最廣泛的應(yīng)用, 比如:
with open("test/test.txt","w") as f: f.write("hello")
這上會(huì)比使用try:...finally:f.close
方便的多.
二. 自定義一個(gè)上下文管理器類:
class MyResource: # __enter__ 返回的對(duì)象會(huì)被with語(yǔ)句中as后的變量接受 def __enter__(self): print('connect to resource') return self def __exit__(self, exc_type, exc_value, tb): print('close resource conection') def query(self): print('query data')
類中有兩個(gè)特殊的魔術(shù)方法:
- __enter__: with語(yǔ)句中的代碼塊執(zhí)行前, 會(huì)執(zhí)行__enter__, 返回的值將賦值給with句中as后的變量.
- __exit__: with語(yǔ)句中的代碼塊執(zhí)行結(jié)束或出錯(cuò), 會(huì)執(zhí)行_exit__
比如以下代碼:
with Myresource() as r: r.query()
的打印結(jié)果為:
connect to resource
query data
close resource conection
那么有沒有一個(gè)簡(jiǎn)化定義的方法呢, python提供了一個(gè)裝飾器contextmanager
三. 使用contextmanager
from contextlib import contextmanager class MyResource: def query(self): print('query data') @contextmanager def make_myresource(): print('start to connect') yield MyResource() print('end connect') pass
被裝飾器裝飾的函數(shù)分為三部分:
- with語(yǔ)句中的代碼塊執(zhí)行前執(zhí)行函數(shù)中yield之前代碼
- yield返回的內(nèi)容復(fù)制給as之后的變量
- with代碼塊執(zhí)行完畢后執(zhí)行函數(shù)中yield之后的代碼
比如下方代碼:
with make_myresource() as r: r.query()
的結(jié)果為:
start to connect
query data
end connect
四. 一個(gè)例子, sqlalchemy: 數(shù)據(jù)庫(kù)的自動(dòng)提交和回滾
在編程中如果頻繁的修改數(shù)據(jù)庫(kù), 一味的使用類似try:... except..: rollback() raise e
其實(shí)是不太好的.
比如某一段的代碼的是這樣的:
try: gift = Gift() gift.isbn = isbn ... db.session.add(gift) db.session.commit() except Exception as e: db.session.rollback() raise e
為了達(dá)到使用with語(yǔ)句的目的, 我們可以重寫db所屬的類:
from flask_sqlalchemy import SQLAlchemy as _SQLALchemy class SQLAlchemy(_SQLALchemy): @contextmanager def auto_commit(self): try: yield self.session.commit() except Exception as e: db.session.rollback() raise e
這時(shí)候, 在執(zhí)行數(shù)據(jù)的修改的時(shí)候便可以:
with db.auto_commit(): gift = Gift() gift.isbn = isbndb.session.add(gift) db.session.add(gift) with db.auto_commit(): user = User() user.set_attrs(form.data) db.session.add(user)
關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題:《Python函數(shù)使用技巧總結(jié)》、《Python面向?qū)ο蟪绦蛟O(shè)計(jì)入門與進(jìn)階教程》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python字符串操作技巧匯總》、《Python編碼操作技巧總結(jié)》及《Python入門與進(jìn)階經(jīng)典教程》
希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。
相關(guān)文章
教你用Python寫一個(gè)京東自動(dòng)下單搶購(gòu)腳本
很多朋友都有網(wǎng)購(gòu)搶購(gòu)限量商品的經(jīng)歷,有時(shí)候蹲點(diǎn)搶怎么也搶不到,今天小編帶你們學(xué)習(xí)怎么用Python寫一個(gè)京東自動(dòng)下單搶購(gòu)腳本,以后再也不用拼手速拼網(wǎng)速啦,快來(lái)一起看看吧2023-03-03用Python將PDF文件轉(zhuǎn)存為圖片的實(shí)現(xiàn)方法
在Python中,將PDF文件轉(zhuǎn)換為圖片格式使用專門的庫(kù)來(lái)處理PDF文檔,并將其每一頁(yè)導(dǎo)出為常見的圖像格式,這可以通過PyMuPDF庫(kù)中的fitz模塊或pdf2image庫(kù)實(shí)現(xiàn),本文給大家介紹了用Python將PDF文件轉(zhuǎn)存為圖片的方法,需要的朋友可以參考下2024-04-04[項(xiàng)目布局配置]Nosql與PythonWeb-Flask框架組合
本文主要描述了怎樣使用輕量級(jí)NoSQL數(shù)據(jù)庫(kù)與PythonWeb-Flask框架組合來(lái)進(jìn)行項(xiàng)目布局及應(yīng)用配置,需要的同學(xué)可以參考下,希望可以對(duì)大家有所進(jìn)益2021-08-08使用pytorch和torchtext進(jìn)行文本分類的實(shí)例
今天小編就為大家分享一篇使用pytorch和torchtext進(jìn)行文本分類的實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來(lái)看看吧2020-01-01python3 對(duì)list中每個(gè)元素進(jìn)行處理的方法
今天小編就為大家分享一篇python3 對(duì)list中每個(gè)元素進(jìn)行處理的方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來(lái)看看吧2018-06-06使用PDB簡(jiǎn)單調(diào)試Python程序簡(jiǎn)明指南
這篇文章主要介紹了使用PDB簡(jiǎn)單調(diào)試Python程序簡(jiǎn)明指南,本文講解了使用PDB調(diào)試程序的簡(jiǎn)單技巧,方便、簡(jiǎn)潔實(shí)用,需要的朋友可以參考下2015-04-04pandas 對(duì)group進(jìn)行聚合的例子
今天小編就為大家分享一篇pandas 對(duì)group進(jìn)行聚合的例子,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來(lái)看看吧2019-12-12