Python with语句上下文管理器两种实现方法分析


Posted in Python onFebruary 09, 2018

本文实例讲述了Python with语句上下文管理器。分享给大家供大家参考,具体如下:

在编程中会经常碰到这种情况:有一个特殊的语句块,在执行这个语句块之前需要先执行一些准备动作;当语句块执行完成后,需要继续执行一些收尾动作。例如,文件读写后需要关闭,数据库读写完毕需要关闭连接,资源的加锁和解锁等情况。

对于这种情况python提供了上下文管理器(Context Manager)的概念,可以通过上下文管理器来定义/控制代码块执行前的准备动作,以及执行后的收尾动作。

一、为何使用上下文管理器

1、不使用上下文管理器的情况

通过try...finally语句执行异常处理和关闭句柄的动作。

logger = open("log.txt", "w")
try:
logger.write('Hello ')

logger.write('World')
finally:

logger.close()
print logger.closed

2、使用上下文管理器

默认文件Python的内置file类型是支持上下文管理协议的。
使用上下文管理器with使得依据精简了很多。

with open("log.txt", "w") as logger:
logger.write('Hello ')

logger.write('World')
print logger.closed

二、实现上下文管理器实现上下文管理器有两种方式实现。方法一:类实现__enter__和__exit__方法。方法二:contextlib模块装饰器和生成器实现。

下面我们通过两种方法分别实现一个自定义的上下文管理器。

1、方法一:通过类实现__enter__和__exit__方法

class File(object):
 def __init__(self, file_name, method):
  self.file_obj = open(file_name, method)
 def __enter__(self):
  return self.file_obj
 def __exit__(self, type, value, traceback):
  self.file_obj.close()
with File('demo.txt', 'w') as opened_file:
 opened_file.write('Hola!')

实现__enter__和__exit__方法后,就能通过with语句进行上下文管理。

a、底层都发生了什么?

1、with语句先暂存了File类的__exit__方法,然后它调用File类的__enter__方法。
2、__enter__方法打开文件并返回给with语句,打开的文件句柄被传递给opened_file参数。
3、with语句调用之前暂存的__exit__方法,__exit__方法关闭了文件。

b、异常处理

关于异常处理,with语句会采取哪些步骤。

1. 它把异常的type,value和traceback传递给__exit__方法
2. 它让__exit__方法来处理异常
3. 如果__exit__返回的是True,那么这个异常就被忽略。
4. 如果__exit__返回的是True以外的任何东西,那么这个异常将被with语句抛出。

异常抛出

#异常抛出,_exit__返回的是True以外的任何东西,那么这个异常将被with语句抛出
class File(object):
 def __init__(self, file_name, method):
  self.file_obj = open(file_name, method)
 def __enter__(self):
  return self.file_obj
 def __exit__(self, type, value, traceback):
  self.file_obj.close()
  print "type:",type
  print "value:",value
  print "traceback:",traceback
with File('demo.txt', 'w') as opened_file:
 opened_file.undefined_function('Hola!')
#output================================================
# type: <type 'exceptions.AttributeError'>
# value: 'file' object has no attribute 'undefined_function'
# traceback: <traceback object at 0x000000000262D9C8>
#  opened_file.undefined_function('Hola!')
# AttributeError: 'file' object has no attribute 'undefined_function'

异常忽略:

#异常忽略,__exit__返回的是True,那么这个异常就被忽略。
class File(object):
 def __init__(self, file_name, method):
  self.file_obj = open(file_name, method)
 def __enter__(self):
  return self.file_obj
 def __exit__(self, exception_type, exception_value, traceback):
  print("Exception has been handled")
  self.file_obj.close()
  return True
with File('demo.txt', 'w') as opened_file:
 opened_file.undefined_function('Hola!')
# output==================================
# Exception has been handled

2、方法二:contextlib模块装饰器和生成器实现

这种方式实现更优雅,我个人更喜欢这种方式。

yield之前的代码由__enter__方法执行,yield之后的代码由__exit__方法执行。本质上还是__enter____exit__方法。

# coding:utf-8
import contextlib
@contextlib.contextmanager
def myopen(filename, mode):
 f = open(filename, mode)
 try:
  yield f.readlines()
 except Exception as e:
  print e
 finally:
  f.close()
if __name__ == '__main__':
 with myopen(r'c:\ip2.txt', 'r') as f:
  for line in f:
   print line

3、with语句上多个下文关联

直接通过一个with语句打开多个上下文,即可同时使用多个上下文变量,而不必需嵌套使用with语句。

class File(object):
 def __init__(self, file_name, method):
  self.file_obj = open(file_name, method)
 def __enter__(self):
  return self.file_obj
 def __exit__(self, exception_type, exception_value, traceback):
  self.file_obj.close()
  return True
with File('demo.txt', 'w') as f1,File('demo.txt','w') as f2:
 print f1,f2
# Output============================# <open file 'demo.txt', mode 'w' at 0x000000000263D150> <open file 'demo.txt', mode 'w' at 0x000000000263D1E0>

更多关于Python相关内容可查看本站专题:《Python数据结构与算法教程》、《Python Socket编程技巧总结》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》及《Python入门与进阶经典教程》

希望本文所述对大家Python程序设计有所帮助。

Python 相关文章推荐
Python linecache.getline()读取文件中特定一行的脚本
Sep 06 Python
Python中的ceil()方法使用教程
May 14 Python
Python3计算三角形的面积代码
Dec 18 Python
浅谈使用Python内置函数getattr实现分发模式
Jan 22 Python
python3 图片referer防盗链的实现方法
Mar 12 Python
python3使用smtplib实现发送邮件功能
May 22 Python
Python爬虫之正则表达式的使用教程详解
Oct 25 Python
python模拟登陆,用session维持回话的实例
Dec 27 Python
django2.0扩展用户字段示例
Feb 13 Python
python实现整数的二进制循环移位
Mar 08 Python
python简单验证码识别的实现方法
May 10 Python
python爬虫之利用selenium模块自动登录CSDN
Apr 22 Python
Python遍历pandas数据方法总结
Feb 09 #Python
python中的闭包函数
Feb 09 #Python
基于Python socket的端口扫描程序实例代码
Feb 09 #Python
利用python 更新ssh 远程代码 操作远程服务器的实现代码
Feb 08 #Python
Tensorflow 利用tf.contrib.learn建立输入函数的方法
Feb 08 #Python
django数据库migrate失败的解决方法解析
Feb 08 #Python
python使用锁访问共享变量实例解析
Feb 08 #Python
You might like
thinkPHP5框架自定义验证器实现方法分析
2018/06/11 PHP
点击下载链接 弹出页面实现代码
2009/10/01 Javascript
JavaScript 对象成员的可见性说明
2009/10/16 Javascript
一个很简单的jquery+xml+ajax的无刷新树结构(无css,后台是c#)
2010/06/02 Javascript
google jQuery 引用文件,jQuery 引用地址集合(jquery 1.2.6至jquery1.5.2)
2011/04/24 Javascript
javascript 另一种图片滚动切换效果思路
2012/04/20 Javascript
关于JS控制代码暂停的实现方法分享
2012/10/11 Javascript
javascript常用函数归纳整理
2014/10/31 Javascript
node.js中的fs.lchown方法使用说明
2014/12/16 Javascript
node.js中的fs.rmdir方法使用说明
2014/12/16 Javascript
常用的jQuery前端技巧收集
2014/12/24 Javascript
Vue下拉框回显并默认选中随机问题
2018/09/06 Javascript
移动端吸顶fixbar的解决方案详解
2019/07/17 Javascript
vue 获取视频时长的实例代码
2019/08/20 Javascript
Layui Table js 模拟选中checkbox的例子
2019/09/03 Javascript
Vue+iview+webpack ie浏览器兼容简单处理
2019/09/20 Javascript
Angular+ionic实现折叠展开效果的示例代码
2020/07/29 Javascript
javascript实现多边形碰撞检测
2020/10/24 Javascript
Python的网络编程库Gevent的安装及使用技巧
2016/06/24 Python
使用Python-OpenCV向图片添加噪声的实现(高斯噪声、椒盐噪声)
2019/05/28 Python
Python代码使用 Pyftpdlib实现FTP服务器功能
2019/07/22 Python
Python K最近邻从原理到实现的方法
2019/08/15 Python
求职简历自荐信范文
2013/10/21 职场文书
电焊工岗位工作职责
2014/07/09 职场文书
党的群众路线教育实践活动党员个人剖析材料
2014/10/08 职场文书
2014年保卫部工作总结
2014/11/21 职场文书
优秀教师个人材料
2014/12/15 职场文书
学雷锋感言
2015/08/03 职场文书
公司行政管理制度范本
2015/08/05 职场文书
如何写好活动总结
2019/06/21 职场文书
大学生创业计划书常用模板
2019/08/07 职场文书
用Python将库打包发布到pypi
2021/04/13 Python
JVM入门之类加载与字节码技术(类加载与类的加载器)
2021/06/15 Java/Android
用Python selenium实现淘宝抢单机器人
2021/06/18 Python
Python装饰器的练习题
2021/11/23 Python
Golang并发工具Singleflight
2022/05/06 Golang