python上下文管理器及with语句

with语句支持在一个叫上下文管理器的对象的控制下执行一系列语句,语法大概如下:

with context as var:
    statements

其中的context必须是个上下文管理器,它实现了两个方法__enter__,__exit__。

1.需求是怎么产生的

在正常的管理各种系统资源(文件、锁定和连接),在涉及到异常时通常是个棘手的问题。异常很可能导致控制流跳过负责释放关键资源的语句。

看一段简单的文件写入代码:

filename = ‘my_file.txt‘
f = open(filename,‘w‘)
f.write(‘Hello ‘)
f.write(‘World‘)
f.close()

如果发生了意外的情况,例如写入‘World‘时磁盘空间不足,就会抛出异常,那么close()语句将不会被执行。

一般的解决方案是使用try-finally语句:

try:
    filename = ‘my_file.txt‘
    f = open(filename,‘w‘)
    f.write(‘Hello ‘)
    f.write(‘World‘)
finally:
    f.close()

但随着语句的增多,try-finally显然不够简洁,用with-as语句可以很简洁的实现以上功能:

with open(‘my_file‘,‘w‘) as f:
    f.write(‘Hello ‘)
    f.write(‘World‘)

这样不仅能处理出现异常的情况,而且还避免了在open一个文件后忘记了写close()方法的情况。

线程中的锁其实也实现了上下文管理协议:

import threading
with threading.Lock():
    # 关键部分
    statements
    # 关键部分结束

再也不怕忘记将锁释放了。

2.上下文管理器的实现

上下文管理器要实现__enter__和__exit__的特殊方法。

__enter__(self): 进入上下文管理器时调用此方法,其返回值将被放入with-as语句中as说明符指定的变量中。

__exit__(self,type,value,tb):离开上下文管理器调用此方法。如果有异常出现,type、value、tb分别为异常的类型、值和追踪信息。如果没有异常,

3个参数均设为None。此方法返回值为True或者False,分别指示被引发的异常得到了还是没有得到处理。如果返回False,引发的异常会被传递出上下文。

文件上下文管理协议大概是如下实现的:

class OpenFile(object):
    def __init__(self,filename,mode):
        self.filename=filename
        self.mode=mode
    def __enter__(self):
        self.f=open(self.filename,self.mode)
        return self.f  #作为as说明符指定的变量的值
    def __exit__(self,type,value,tb):
        self.f.close()
        return False   #异常会被传递出上下文
with OpenFile(‘my_file.txt‘,‘w‘) as f:
    f.write(‘Hello ‘)
    f.write(‘World‘)

  相当于

try :
    执行__enter__的内容
finally:
    执行__exit__的内容

3.异常的处理 

__exit__函数就能够拿到关于异常的所有信息(异常类型,异常值以及异常追踪信息),这些信息将帮助异常处理操作。

class ListTrans(object):
    def __init__(self,alist):
        self.alist=alist
    def __enter__(self):
        self.thecopy=list(self.alist)
        return self.thecopy
    def __exit__(self,exc_type,value,tb):
        if exc_type is None:
            self.alist[:]=self.thecopy
        return False

没有异常发生时:

alist=[]
with ListTrans(alist) as working:
    working.append(1)
    working.append(2)
print alist

生成:

[1, 2]

有异常发生时:

alist=[]
with ListTrans(alist) as working:
    working.append(1)
    working.append(2)
    raise RuntimeError(‘we are hosed‘)
print alist

生成:

Traceback (most recent call last):
  File "C:/Python27/contextmanager.py", line 15, in <module>
    raise RuntimeError(‘we are hosed‘)
RuntimeError: we are hosed

alist无变化。

可以捕捉异常:

alist=[]
try:
    with ListTrans(alist) as working:
        working.append(1)
        working.append(2)
        raise RuntimeError(‘we are hosed‘)
except RuntimeError as e:
    print e
print alist

生成:

we are hosed
[]

当然,也可以简单的将__exit__的返回值设为True来忽略异常。

4.contextmanager装饰器

contextlib模块的contextmanager装饰器可以更方便的实现上下文管理器。

任何能够被yield关键词分割成两部分的函数,都能够通过装饰器装饰的上下文管理器来实现。任何在yield之前的内容都可以看做在代码块执行前的操作,

而任何yield之后的操作都可以放在exit函数中。

from contextlib import contextmanager
@contextmanager
def listTrans(alist):
    thecopy=list(alist)
    yield thecopy
    alist[:]=thecopy
alist=[]
with listTrans(alist) as working:
    working.append(1)
    working.append(2)
print alist

yield返回的值相当于__enter__的返回值。

要注意的是,这不是异常安全的写法,也就是说,当出现异常时,yield后的语句是不会执行的,想要异常安全,可用try捕捉异常:

from contextlib import contextmanager
@contextmanager
def listTrans(alist):
    thecopy=list(alist)
    try:
        yield thecopy
    except RuntimeError:
        pass
    alist[:]=thecopy
alist=[]
with listTrans(alist) as working:
    working.append(1)
    working.append(2)
    raise RuntimeError

contextlib模块还有两个好玩的方法:nested,closing。

nested:用来更方便的减少嵌套写法:

开始是这么样的:

with open(‘toReadFile‘, ‘r‘) as reader:
    with open(‘toWriteFile‘, ‘w‘) as writer:
        writer.writer(reader.read())

用nested:

with contextlib.nested(open(‘fileToRead.txt‘, ‘r‘),
                       open(‘fileToWrite.txt‘, ‘w‘)) as (reader, writer):
    writer.write(reader.read())

python2.7后可以:

with open(‘fileToRead.txt‘, ‘r‘) as reader,open(‘fileToWrite.txt‘, ‘w‘) as writer:
        writer.write(reader.read())

closing(object):创建上下文管理器,在执行过程离开with语句时自动执行object.close():

class Door(object) :
    def open(self) :
        print ‘Door is opened‘
    def close(self) :
        print ‘Door is closed‘
with contextlib.closing(Door()) as door :
    door.open()

python上下文管理器及with语句

时间: 2024-12-14 08:25:31

python上下文管理器及with语句的相关文章

Python上下文管理器与with语句

什么是上下文管理器 上下文管理器顾名思义是管理上下文的,也就是负责冲锋和垫后,而让主人专心完成自己的事情.我们在编写程序的时候,通常会将一系列操作放到一个语句块中,当某一条件为真时执行该语句快.有时候,我们需要再执行一个语句块时保持某种状态,并且在离开语句块后结束这种状态.例如对文件的操作,我们在打开一个文件进行读写操作时需要保持文件处于打开状态,而等操作完成之后要将文件关闭.所以,上下文管理器的任务是:代码块执行前准备,代码块执行后收拾.上下文管理器是在Python2.5加入的功能,它能够让你

python上下文管理器ContextLib及with语句

http://blog.csdn.net/pipisorry/article/details/50444736 with语句 with语句是从 Python 2.5 开始引入的一种与异常处理相关的功能(2.5 版本中要通过 from __future__ import with_statement 导入后才可以使用),从 2.6 版本开始缺省可用(参考 What's new in Python 2.6? 中 with 语句相关部分介绍).with 语句适用于对资源进行访问的场合,确保不管使用过程

Python核心技术与实战——二一|巧用上下文管理器和with语句精简代码

我们在Python中对于with的语句应该是不陌生的,特别是在文件的输入输出操作中,那在具体的使用过程中,是有什么引伸的含义呢?与之密切相关的上下文管理器(context manager)又是什么呢? 什么是上下文管理器 在任何一种编程语言里,文件的输入输出.数据库的建立连接和断开等操作,都是很常见的资源管理操作.但是资源是有限的,在写程序的时候,我们必须保证这些资源在使用后得到释放,不然就容易造成资源泄漏,轻者系统处理缓慢,重则系统崩溃. 我们看一个例子: for i in range(100

Python上下文管理器

上下文管理器 在使用Python编程中,可以会经常碰到这种情况:有一个特殊的语句块,在执行这个语句块之前需要先执行一些准备动作:当语句块执行完成后,需要继续执行一些收尾动作. 例如:当需要操作文件或数据库的时候,首先需要获取文件句柄或者数据库连接对象,当执行完相应的操作后,需要执行释放文件句柄或者关闭数据库连接的动作. 又如,当多线程程序需要访问临界资源的时候,线程首先需要获取互斥锁,当执行完成并准备退出临界区的时候,需要释放互斥锁. 对于这些情况,Python中提供了上下文管理器(Contex

Python 上下文管理器和else块

最终,上下文管理器可能几乎与子程序(subroutine)本身一样重要.目前,我们只了解了上下文管理器的皮毛--Basic 语言有with 语句,而且很多语言都有.但是,在各种语言中 with 语句的作用不同,而且做的都是简单的事,虽然可以避免不断使用点号查找属性,但是不会做事前准备和事后清理.不要觉得名字一样,就意味着作用也一样.with 语句是非常了不起的特性.  --Raymond Hettinger 雄辩的 Python 布道者 先做这个,再做那个:if语句之外的else块 这个语言特性

python --上下文管理器

转自:http://www.cnblogs.com/vamei/archive/2012/11/23/2772445.html 上下文管理器(context manager)是Python2.5开始支持的一种语法,用于规定某个对象的使用范围.一旦进入或者离开该使用范围,会有特殊操作被调用 (比如为对象分配或者释放内存).它的语法形式是with...as... 关闭文件 我们会进行这样的操作:打开文件,读写,关闭文件.程序员经常会忘记关闭文件.上下文管理器可以在不需要文件的时候,自动关闭文件. 下

Python上下文管理器(Context managers)

上下文管理器(Context managers) 上下文管理器允许你在有需要的时候,精确地分配和释放资源. 使用上下文管理器最广泛的案例就是with语句了.想象下你有两个需要结对执行的相关操作,然后还要在它们中间放置一段代码.上下文管理器就是专门让你做这种事情的.举个例子: with open('some_file', 'w') as opened_file: opened_file.write('Hola!') 上面这段代码打开了一个文件,往里面写入了一些数据,然后关闭该文件.如果在往文件写数

Python 上下文管理器模块--contextlib

在 Python 处理文件的时候我们使用 with 关键词来进行文件的资源关闭,但是并不是只有文件操作才能使用 with 语句.今天就让我们一起学习 Python 中的上下文管理 contextlib. 上下文管理器 上下文,简而言之,就是程式所执行的环境状态,或者说程式运行的情景.既然提及上下文,就不可避免的涉及 Python 中关于上下文的魔法.上下文管理器(context manager)是 Python2.5 开始支持的一种语法,用于规定某个对象的使用范围.一旦进入或者离开该使用范围,会

吃透Python上下文管理器

什么是上下文管理器? 我们常见的with open操作文件,就是一个上下文管理器.如: with open(file, 'rb') as f: text = f.read() 那上下文管理器具体的定义是什么呢? 上下文管理器:是指在一段代码执行之前执行一段代码,用于一些预处理工作:执行之后再执行一段代码,用于一些清理工作. 比如刚提到的文件操作,打开文件进行读写,读写完之后需要将文件关闭.很明显用到了上下文管理器.主要依靠__enter__.__exit__这两个"魔术方法"实现. _