Python上下文管理器類(lèi)和上下文管理器裝飾器contextmanager用法實(shí)例分析

更新時(shí)間：2019年11月07日 09:22:02 作者：weixin_42359464

這篇文章主要介紹了Python上下文管理器類(lèi)和上下文管理器裝飾器contextmanager用法,結(jié)合實(shí)例形式分析了上下文管理器類(lèi)定義、使用、sqlalchemy實(shí)現(xiàn)數(shù)據(jù)庫(kù)的自動(dòng)提交和回滾相關(guān)操作技巧,需要的朋友可以參考下

一. 什么是上下文管理器

上下文管理器是在Python2.5之后加入的功能，可以在方便的需要的時(shí)候比較精確地分配和釋放資源, with便是上下文管理器的最廣泛的應(yīng)用, 比如:

with open("test/test.txt","w") as f:
 f.write("hello")

這上會(huì)比使用try:...finally:f.close方便的多.

二. 自定義一個(gè)上下文管理器類(lèi):

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')

類(lèi)中有兩個(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

那么有沒(méi)有一個(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ù), 一味的使用類(lèi)似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ǔ)句的目的, 我們可以重寫(xiě)db所屬的類(lèi):

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)容感興趣的讀者可查看本站專(zhuān)題：《Python函數(shù)使用技巧總結(jié)》、《Python面向?qū)ο蟪绦蛟O(shè)計(jì)入門(mén)與進(jìn)階教程》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python字符串操作技巧匯總》、《Python編碼操作技巧總結(jié)》及《Python入門(mén)與進(jìn)階經(jīng)典教程》

希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。

您可能感興趣的文章: