第一次遇到with
是在文件那一章,with
的作用是就是会自动关掉文件管道。
with open('path','读写模式‘) as f:
do something
这一部分就等价于
f = open('path','读写模式')
do something
f.close()
第二次是在数据库连接,连接池那里。使用的基本思想大致是with
所求值的对象必须有一个enter()
方法和一个exit()
方法。下面给一个简单的例子去说明使用with
的时候做了哪些操作
class Sample:
def __enter__(self):
print "In __enter__()"
return "Foo"
def __exit__(self, type,value, trace):
print "In__exit__()"
def get_sample():
return Sample()
with get_sample() as sample:
print(sample)
-
with
开始,enter()
方法被执行 -
enter()
方法返回的值 - 这个例子中是Foo
,赋值给变量sample
- 执行代码块,打印变量
sample
的值为Foo
-
exit()
方法被调用with
真正强大之处是它可以处理异常。注意到Sample
类的exit
方法有三个参数-val
,type
和trace
。 这些参数在异常处理中相当有用。
class Sample:
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print(exc_type)
print(exc_val)
print(exc_tb)
def doSomething(self):
a = 1/0
return a
def getSample():
return Sample()
if __name__ == '__main__':
with getSample() as sample:
sample.doSomething()
这段代码的执行结果是
/usr/bin/python3.5 /home/zionhuang/data/awesome-python3-webapp/test/test8.py
<class 'ZeroDivisionError'>
division by zero
<traceback object at 0x7fbb5c091d48>
Traceback (most recent call last):
File "/home/zionhuang/data/awesome-python3-webapp/test/test8.py", line 23, in <module>
sample.doSomething()
File "/home/zionhuang/data/awesome-python3-webapp/test/test8.py", line 14, in doSomething
a = 1/0
ZeroDivisionError: division by zero
Process finished with exit code 1
结果头三句加黑的分别是val
, type
和 trace
这三个参数在出现异常时的值
常抛出时,与之关联的type
,value
和stack trace
传给exit()
方法,因此抛出的ZeroDivisionError
异常被打印出来了。开发库时,清理资源,关闭文件等等操作,都可以放在exit
方法当中。
因此,Python
的with
语句是提供一个有效的机制,让代码更简练,同时在异常产生时,清理工作更简单。