【ZZ】python with...as...用法

with从Python 2.5就有,须要from __future__ import with_statement。自python 2.6开始, 成为默认关键字
在What's new in python2.6/3.0中,明确提到:
The ‘with‘ statement is a control-flow structure whose basic structure is:
with expression [as variable]:
          with-block
也就是说with是一个控制流语句,跟if/for/while/try之类的是一类的,with能够用来简化try finally代码,看起来能够比try finally更清晰。
这里新引入了一个 "上下文管理协议"context management protocol,实现方法是为一个类 定义__enter__和__exit__两个函数。
with expresion as variable的执行过程是,首先执行__enter__函数,它的返回值会赋给as后面的variable,想让它返回什么就返回什么,只要你知道怎么处理就能够了,若是不写as variable,返回值会被忽略。
而后,开始执行with-block中的语句,不论成功失败(好比发生异常、错误,设置sys.exit()),在with-block执行完成后,会执行__exit__函数。
这样的过程其实等价于:
try:
    执行 __enter__的内容
    执行 with_block.
finally:
    执行 __exit__内容
只不过,如今把一部分代码封装成了__enter__函数,清理代码封装成__exit__函数。

 

它的原理以下:

全部实现上下文协议的对象都包含如下三种方法:

__context__() 它返回一个自我管理的上下文对象,或者一个真正意义的上下文管理器
__enter()__ 进入上下文管理器,开始迭代
当with语句结束的时候,不管是正常结束仍是抛出异常,都会执行__exit__(),该方法用于关闭资源链接。 python

若是有一个类包含 __enter__ 方法和 __exit__ 方法,像这样:
class controlled_execution:
    def__enter__(self):
            set things up
            return thing
    def__exit__(self, type, value, traceback): sql

            tear things down
那么它就能够和with一块儿使用,像这样:
with controlled_execution() as thing:
        some code
 当with语句被执行的时候,python对表达式进行求值,对求值的结果(叫作“内容守护者”)调用__enter__方法,并把__enter__方法的返回值赋给as后面的变量。而后python会执行接下来的代码段,而且不管这段代码干了什么,都会执行“内容守护者”的__exit__方法。

做为额外的红利,__exit__方法还可以在有exception的时候看到exception,而且压制它或者对它作出必要的反应。要压制exception,只须要返回一个true。好比,下面的__exit__方法吞掉了任何的TypeError,可是让全部其余的exceptions经过:
def__exit__(self, type, value, traceback):
        return isinstance(value, TypeError)

在Python2.5中,file object拥有__enter__和__exit__方法,前者仅仅是返回object本身,然后者则关闭这个文件
>>> f = open("x.txt")
>>> f
<open file 'x.txt', mode 'r' at 0x00AE82F0>
>>> f.__enter__()
<open file 'x.txt', mode 'r' at 0x00AE82F0>
>>> f.read(1)
'X'
>>> f.__exit__(None, None, None)
>>> f.read(1)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ValueError: I/O operation on closed file
这样要打开一个文件,处理它的内容,而且保证关闭它,你就能够简简单单地这样作:
with open("x.txt") as f:
    data = f.read()
    do something with data

个人补充:
数据库的链接好像也能够和with一块儿使用,我在一本书上看到如下内容:
conn = sqlite.connect("somedb")
with conn:
    conn.execute("insert into sometable values (?,?)",("foo","bar"))
 在这个例子中,commit()是在全部with数据块中的语句执行完毕而且没有错误以后自动执行的,若是出现任何的异常,将执行rollback()操做,再次提示异常 数据库

相关文章
相关标签/搜索