with/as环境管理器
基本使用
with expression [as var]:
with-block
这里expression要返回一个对象，从而支持环境管理协议。如果选用的as字句存在，此时对象也可返回一个值，赋值给var。var并非赋值为expression的结果。expression的结果是支持环境协议的对象，而var则是赋值为其他的东西。expression返回的对象可在with-block
with open('catcher.py') as myfile:
for line in myfile:
print(line)
这里open的调用，会返回一个文件对象，赋值给变量myfile。文件迭代器会在for循环内逐行读取。
with语句执行后，环境管理机制保证由myfile所应用的文件对象会自动关闭，即使处理该文件时，for循环引发了异常也是如此
with语句实际的工作方式
- 计算表达式，所得到的对象称为环境管理器，必须有__enter__和__exit__方法
- 环境管理器的__enter__方法会被调用。如果as字句存在，其返回值会赋值个as字句中的变量，否则直接丢弃
- 代码块中嵌套的代码会执行
- 如果with代码块引发异常,__exit__方法就会被调用，这些也是sys.exc_info返回的相同值。如果此方法返回值为假，则异常会重新引发
- 如果with代码块没有异常，__exit__依然会被调用，其type、value以及traceback参数都会以None传递
#-*-coding:UTF-8-*-
class TraceBlock:
def message(self,arg):
print('running',arg)
def __enter__(self):
print('start with block')
return self
def __exit__(self,exc_type,exc_value,exc_tb):
if exc_type is None:
print('exited normally')
else:
print('raise an exception',exc_type)
return False
if __name__ == '__main__':
with TraceBlock() as action:
action.message('tets')
raise TypeError
print('ded')
>>>
start with block
running tets
ded
exited normally
使用contextlib构造Context Manager
#coding:utf-8
from contextlib import contextmanager
@contextmanager #装饰器 才能使用with语句 @contextmanager这个decorator接受一个generator，用yield语句把with ... as var把变量输出出去，然后，with语句就可以正常地工作了
def file_open(path):
try:
f_obj = open(path,'r')
yield f_obj #一定要有这一句不然报错
except OSError:
print('We had an error')
finally:
print('Close file')
f_obj.close()
if __name__ == '__main__':
with file_open('dequetest.py') as fobj:
print(fobj.read())
contextlib.closing(thing)
可以用closing()来把该对象变为上下文对象
from contextlib import closing
from urllib.request import urlopen
if __name__ == '__main__':
with closing(urlopen('https://www.baidu.com')) as webpage:
for line in webpage:
print(line)
>>>
输出百度的网页
这里的closin其实就是
@contextmanager
def closing(db):
try:
yield db
except Exception:
pass
finally:
db.close()
pass
contextlib.suppress(*exceptions)
能够忽略我们想忽略的错误，下文中没有dd.txt文件，如果没有suppress语句则会报错。
from contextlib import suppress
if __name__ == '__main__':
with suppress(FileNotFoundError):
with open('dd.txt') as f:
for line in f:
print(line)
contextlib.redirect_stdout / redirect_stderr 输出流重定向
from contextlib import redirect_stdout
with open('text.txt','w') as fobj:
with redirect_stdout(fobj):
print('hehehhhe')
>>>在控制台中不会输出'hehehhhe'，会在同一目录下生成一个文件text.txt并把'hehehhe'输出到文件中
