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

概述本文实例讲述了Pythonwith语句上下文管理器。分享给大家供大家参考，具体如下：

本文实例讲述了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,traceback):  self.file_obj.close()  print "type:",type  print "value:",value  print "traceback:",tracebackwith 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,exception_type,exception_value,traceback):  print("Exception has been handled")  self.file_obj.close()  return Truewith 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-8import contextlib@contextlib.contextmanagerdef 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,traceback):  self.file_obj.close()  return Truewith 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上下文管理器和with块详解深入学习Python中的上下文管理器与else块正确理解python中的关键字“with”与上下文管理器Python中的with语句与上下文管理器学习总结深入解析Python中的上下文管理器Python深入学习之上下文管理器Python中的上下文管理器和with语句的使用 总结

以上是内存溢出为你收集整理的Python with语句上下文管理器两种实现方法分析全部内容，希望文章能够帮你解决Python with语句上下文管理器两种实现方法分析所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。