#《流流畅的Python》第15章 上下文管理器和else块
#15.1 先做这个，再做那个：if语句之外的else块
#else子句不仅能在if语句中使用，还能在for、while和try语句中使用。
#for/else、while/else和try/else的语义关系紧密，不过与if/else差别很大。
#可是，在循环中，else的语义恰好相反：“运行这个循环，然后做那件事。
#在Python中，try/except不仅用于处理错误，还常用于控制流程。
#EAFP取得原谅比获得许可容易（easier to ask for forgiveness than permission）。
#15.2 上下文管理器和with块
#上下文管理器协议包含 __enter__ 和 __exit__ 两个方法。
#with语句开始运行时，会在上下文管理器对象上调用 __enter__ 方法。
#with语句运行结束后，会在上下文管理器对象上调用 __exit__ 方法，以此扮演finally子句的角色。
with open('mirror.py')as fp: #fp绑定到打开的文件上，因为文件的__enter__方法返回self。
src=fp.read(60) #从fp中读取一些数据。
print(len(src))
print(fp) #fp变量仍然可用。
print(fp.closed,fp.encoding) #可以读取fp对象的属性。
fp.read()#但是不能在fp上执行I/O操作，因为在with块的末尾，调用TextIOWrapper.__exit__方法把文件关闭了。
#示例15-2 测试LookingGlass上下文管理器类
class LookingClass:
def __enter__(self):
import sys
self.original_write=sys.stdout.write
sys.stdout.write=self.reverse_write
return 'JABBERWOCKY'
def reverse_write(self,text):
self.original_write(text[::-1])
def __exit__(self, exc_type, exc_value, exc_traceback):
# 如果一切正常Python调用__exit__方法时传入的参数是None,None, None；如果抛出了异常，这三个参数是异常数据，
import sys
sys.stdout.write=self.original_write
if exc_traceback is ZeroDivisionError:
print('Please DO NOT divide by zero')
return True
with LookingClass() as what:
print("Alice,kitty and Snowdrop") #pordwonS dna yttik,ecilA
print(what) #YKCOWREBBAJ
print(what) #JABBERWOCKY
print('Back to normal') #Back to normal
#示例15-4 在with块之外使用LookingGlass类
manager=LookingClass()
print(manager) #<__main__.LookingClass object at 0x004B5AB0>
monster=manager.__enter__()
print(monster=='JABBERWOCKY') #eurT
print(monster) #YKCOWREBBAJ
print(manager) #>0BA50400x0 ta tcejbo ssalCgnikooL.__niam__<
print(monster) #JABBERWOCKY'
#15.3 contextlib模块中的实用工具
#closing如果对象提供了close()方法，但没有实现__enter__/__exit__协议，那么可以使用这个函数构建上下文管理器。
#suppress构建临时忽略指定异常的上下文管理器。
#@contextmanager这个装饰器把简单的生成器函数变成上下文管理器，这样就不用创建类去实现管理器协议了。
#ContextDecorator这是个基类，用于定义基于类的上下文管理器。这种上下文管理器也能用于装饰函数，在受管理的上下文中运行整个函数。
#ExitStack这个上下文管理器能进入多个上下文管理器。with 块结束时，ExitStack 按照后进先出的顺序调用栈中各个上下文管理器的__exit__ 方法。
#如果事先不知道 with 块要进入多少个上下文管理器，可以使用这个类。例如，同时打开任意一个文件列表中的所有文件。
#15.4 使用@contextmanager
#在使用 @contextmanager 装饰的生成器中，yield语句的作用是把函数的定义体分成两部分：
#yield语句前面的所有代码在with块开始时（即解释器调用 __enter__ 方法时）执行，yield语句后面的代码在with 块结束时（即调用 __exit__ 方法时）执行。
import contextlib
@contextlib.contextmanager
def looking_glass():
import sys
original_write=sys.stdout.write
def reverse_write(text):
original_write(text[::-1])
sys.stdout.write=reverse_write
yield 'JABBERWOCKY'
sys.stdout.write=original_write
with looking_glass() as what:
print('Alice,kitty and Snowdrop') #pordwonS dna yttik,ecilA
print(what) #YKCOWREBBAJ
print(what) #JABBERWOCKY
#示例 15-8用于原地重写文件的上下文管理器
import csv
with inplace(csvfilename,'r',newline='') as(infh,outfh):
reader=csv.reader(infh)
writer=csv.writer(outfh)
for row in reader:
row += ['new','columns']
writer.writerow(row)
#15.5 本章小结
#除了自动关闭打开的文件之外，with语句还有很多用途。
#@contextmanager装饰器优雅且实用，把三个不同的Python特性结合到了一起：函数装饰器、生成器和with语句。
