溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本文實例講述了Python上下文管理器類和上下文管理器裝飾器contextmanager用法。分享給大家供大家參考，具體如下：
上下文管理器是在Python2.5之后加入的功能，可以在方便的需要的時候比較精確地分配和釋放資源, with便是上下文管理器的最廣泛的應用, 比如:
with open("test/test.txt","w") as f:
f.write("hello")
這上會比使用try:...finally:f.close方便的多.
class MyResource:
# __enter__ 返回的對象會被with語句中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')
類中有兩個特殊的魔術方法:
比如以下代碼:
with Myresource() as r: r.query()
的打印結果為:
connect to resource
query data
close resource conection
那么有沒有一個簡化定義的方法呢, python提供了一個裝飾器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
被裝飾器裝飾的函數分為三部分:
比如下方代碼:
with make_myresource() as r: r.query()
的結果為:
start to connect
query data
end connect
在編程中如果頻繁的修改數據庫, 一味的使用類似try:... except..: rollback() raise e其實是不太好的.
比如某一段的代碼的是這樣的:
try:
gift = Gift()
gift.isbn = isbn
...
db.session.add(gift)
db.session.commit()
except Exception as e:
db.session.rollback()
raise e
為了達到使用with語句的目的, 我們可以重寫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
這時候, 在執行數據的修改的時候便可以:
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)
關于Python相關內容感興趣的讀者可查看本站專題：《Python函數使用技巧總結》、《Python面向對象程序設計入門與進階教程》、《Python數據結構與算法教程》、《Python字符串操作技巧匯總》、《Python編碼操作技巧總結》及《Python入門與進階經典教程》
希望本文所述對大家Python程序設計有所幫助。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
