本文例項講述了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')

類中有兩個特殊的魔術方法:

• __enter__: with語句中的程式碼塊執行前,會執行__enter__,返回的值將賦值給with句中as後的變數.
• __exit__: with語句中的程式碼塊執行結束或出錯,會執行_exit__

比如以下程式碼:

with Myresource() as r:
  r.query()

的列印結果為:

connect to resource
query data
close resource conection

那麼有沒有一個簡化定義的方法呢,python提供了一個裝飾器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

被裝飾器裝飾的函式分為三部分:

1. with語句中的程式碼塊執行前執行函式中yield之前程式碼
2. yield返回的內容複製給as之後的變數
3. with程式碼塊執行完畢後執行函式中yield之後的程式碼

比如下方程式碼:

with make_myresource() as r:
   r.query()

的結果為:

start to connect
query data
end connect

四. 一個例子,sqlalchemy: 資料庫的自動提交和回滾

在程式設計中如果頻繁的修改資料庫,一味的使用類似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程式設計有所幫助。