Python上下文管理器类和上下文管理器装饰器contextmanager用法实例分析
本文实例讲述了Python上下文管理器类和上下文管理器装饰器contextmanager用法。分享给大家供大家参考，具体如下：
一.什么是上下文管理器
上下文管理器是在Python2.5之后加入的功能，可以在方便的需要的时候比较精确地分配和释放资源,with便是上下文管理器的最广泛的应用,比如:
withopen("test/test.txt","w")asf:
f.write("hello")
这上会比使用try:...finally:f.close方便的多.
二.自定义一个上下文管理器类:
classMyResource:
#__enter__返回的对象会被with语句中as后的变量接受
def__enter__(self):
print('connecttoresource')
returnself
def__exit__(self,exc_type,exc_value,tb):
print('closeresourceconection')
defquery(self):
print('querydata')
类中有两个特殊的魔术方法:
- __enter__:with语句中的代码块执行前,会执行__enter__,返回的值将赋值给with句中as后的变量.
- __exit__:with语句中的代码块执行结束或出错,会执行_exit__
比如以下代码:
withMyresource()asr: r.query()
的打印结果为:
connecttoresource
querydata
closeresourceconection
那么有没有一个简化定义的方法呢,python提供了一个装饰器contextmanager
三.使用contextmanager
fromcontextlibimportcontextmanager
classMyResource:
defquery(self):
print('querydata')
@contextmanager
defmake_myresource():
print('starttoconnect')
yieldMyResource()
print('endconnect')
pass
被装饰器装饰的函数分为三部分:
- with语句中的代码块执行前执行函数中yield之前代码
- yield返回的内容复制给as之后的变量
- with代码块执行完毕后执行函数中yield之后的代码
比如下方代码:
withmake_myresource()asr: r.query()
的结果为:
starttoconnect
querydata
endconnect
四.一个例子,sqlalchemy:数据库的自动提交和回滚
在编程中如果频繁的修改数据库,一味的使用类似try:...except..:rollback()raisee其实是不太好的.
比如某一段的代码的是这样的:
try: gift=Gift() gift.isbn=isbn ... db.session.add(gift) db.session.commit() exceptExceptionase: db.session.rollback() raisee
为了达到使用with语句的目的,我们可以重写db所属的类:
fromflask_sqlalchemyimportSQLAlchemyas_SQLALchemy classSQLAlchemy(_SQLALchemy): @contextmanager defauto_commit(self): try: yield self.session.commit() exceptExceptionase: db.session.rollback() raisee
这时候,在执行数据的修改的时候便可以:
withdb.auto_commit(): gift=Gift() gift.isbn=isbndb.session.add(gift) db.session.add(gift) withdb.auto_commit(): user=User() user.set_attrs(form.data) db.session.add(user)
关于Python相关内容感兴趣的读者可查看本站专题：《Python函数使用技巧总结》、《Python面向对象程序设计入门与进阶教程》、《Python数据结构与算法教程》、《Python字符串操作技巧汇总》、《Python编码操作技巧总结》及《Python入门与进阶经典教程》
希望本文所述对大家Python程序设计有所帮助。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。