python 上下文管理器、 else 块、@contextmanager

文章目录

1. else
2. with上下文管理器
3. contextlib模块实用工具
4. @contextmanager 装饰器

learn from 《流畅的python》

1. else

for/else、while/else 和 try/else
前两者只有在 没有被break 时，才会运行 else
try 块中没有异常抛出时 才运行 else

for i in range(3):print(i)
else:print("finish, no break")# finish, no breakfor i in range(3):if i == 2:breakprint(i)
else:print("break")# 无输出i = 0
while i < 3:print(i)i += 1
else:print("no break") # no breaki = 0
while i < 3:if i==2:breakprint(i)i += 1
else:print("break") # 无输出arr = [0]
try:arr[0] = 1
except:print("error") # 无输出
else:print("no error, run else")# no error, run elsearr = []
try:arr[0] = 0
except:print("error") # error
else:print("error, will not run else") # 无输出
finally:print("finish") # finish
# finally 子句中的代 码通常用于释放重要的资源，或者还原临时变更的状态

else 这么用的比较少

2. with上下文管理器

上下文管理器协议包含 __enter__ 和 __exit__ 两个方法

with 语句运行时，会在上下文管理器对象上调用 __enter__ 方法
with 语句结束后，会在上下文管理器对象上调用 __exit__ 方法，以此扮演 finally 子句的角色（释放重要的资源，或者还原临时变更的状态），如关闭文件等

class LookingGlass:def __enter__(self): # 没有其它的参数了import sysself.original_write = sys.stdout.writesys.stdout.write = self.reverse_writereturn 'ABCD'def reverse_write(self, text): # 反转字符串，实现输出self.original_write(text[::-1])def __exit__(self, exc_type, exc_value, traceback):# exc_type 异常类# exc_value 异常实例# traceback 对象# 在 try/finally 语句的 finally 块中调用 sys.exc_info()# 得到的就是 __exit__ 接收的这三个参数import syssys.stdout.write = self.original_writeif exc_type is ZeroDivisionError:print('Please DO NOT divide by zero!')return Truewith LookingGlass() as what:# 进入 __enter__ 函数时返回的 ABCD 字符串存入 whatprint("Michael learning python")# nohtyp gninrael leahciMprint(what)# DCBA
print("Michael learning python")
# Michael learning python
print(what)
# ABCD

manager = LookingGlass()
print(manager)
# <__main__.LookingGlass object at 0x00000231226F2190>
string = manager.__enter__()
print(string)  # DCBA
print(string == "ABCD")  # eurT, 所有的输出经过管理器的 反向输出处理了
print(string == "DCBA")  # eslaF, 所有的输出经过管理器的 反向输出处理了
print(manager)
# >0912F62213200000x0 ta tcejbo ssalGgnikooL.__niam__<
manager.__exit__(None, None, None) # 还原了正常的输出
print(string)  # ABCD

3. contextlib模块实用工具

closing 如果对象提供了 close() 方法，但没有实现 __enter__/__exit__ 协议，那么可以使用这个函数构建上下文管理器
suppress, 构建临时 忽略指定异常 的上下文管理器
@contextmanager，装饰器把简单的生成器函数变成上下文管理器，这样就不用创建类去实现管理器协议了
ContextDecorator，这是个基类，用于定义基于类的上下文管理器。这种上下文管理器也能用于装饰函数，在受管理的上下文中运行整个函数
ExitStack，这个上下文管理器 能进入多个 上下文管理器。with 块结束时，ExitStack 按照后进先出的顺序调用栈中各个上下文管理器的 __exit__ 方法。
如果事先不知道 with 块要进入多少个上下文管理器，可以使用这个类。例如，同时打开任意一个文件列表中的所有文件

使用最广泛的是 @contextmanager 装饰器，因此要格外留心。
这个装饰器也有迷惑人的一面，因为它与迭代无关，却要使用 yield 语句

4. @contextmanager 装饰器

@contextmanager 装饰器能减少代码量，因为 不用编写一个完整的类，定义 __enter__ 和 __exit__ 方法，而只需实现 有一个 yield 语句的生成器，生成想让 __enter__ 方法返回的值
在使用 @contextmanager 装饰的生成器中，yield 语句的作用是把函数的定义体分成两部分：
yield 语句前面的所有代码在 with 块开始时（即解释器调用 __enter__ 方法时）执行
yield 语句后面的代码在 with 块结束时（即调用 __exit__ 方法时）执行

import contextlib@contextlib.contextmanager
def looking_glass1():import sysoriginal_write = sys.stdout.writedef reverse_write(text):original_write(text[::-1])sys.stdout.write = reverse_writeyield 'ABCD'  # 产出的值绑定到 as 目标上# 执行 with 块代码时，函数会在这里暂停sys.stdout.write = original_write#  控制权一旦跳出 with 块，继续执行 yield 语句之后的代码with looking_glass1() as what:print("Michael learning python")print(what)
# nohtyp gninrael leahciM
# DCBA
print("Michael learning python")
print(what)
# Michael learning python
# ABCD

其实，contextlib.contextmanager 装饰器会把函数包装成实现 __enter__ 和 __exit__ 方法的类

用于原地重写文件的上下文管理器

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)

inplace 函数是个上下文管理器，为同一个文件提供了两个句柄（例中的 infh 和 outfh），以便同时读写同一个文件。比标准库中的 fileinput.input 函数易用

python 上下文管理器、 else 块、@contextmanager相关推荐

Python 上下文管理器和with块二
标准库中还有个contextlib 模块,提供一些实用工具, closing 如果对象提供了 close() 方法,但没有实现 __enter__/__exit__ 协议,那么可以使用这个函数构建上下 ...
Python 上下文管理器和with块一
上下文管理器对象存在的目的是管理 with 语句,就像迭代器的存在是为了管理 for 语句一样上下文管理器协议包含 __enter__ 和 __exit__ 两个方法把值绑定到目标变量上(as 子 ...
Python上下文管理器的魔力
点击关注我哦一篇文章带你了解Python上下文管理器的魔力小编将为您准备一份很棒的Python上下文管理器使用指南,这将使您的代码更具可读性和可靠性,降低您的错误发生率. 资源管理器是我们在任何编 ...
此上下文中不支持函数定义。请在代码文件中创建函数。_深入解析Python上下文管理器，让你不再迷茫！...
1. 上下文管理器一个类只要实现了 __enter__() 和 __exit__() 这个两个方法,通过该类创建的对象我们就称之为上下文管理器. 上下文管理器可以使用 with 语句,with语句之 ...
爱上python系列------python上下文管理器(二):对suppress进行装饰器重新实现
python上下文管理器可以做的事情简直不能太多这不,官方的文档实现了一个方法suppress,用于处理异常 from contextlib import suppresswith suppress ...
Python 上下文管理器和 with 语句
1. 上下文管理器概念什么是 Python 的上下文管理器(Context Managers)呢? 含有 __enter__ 和 __exit__ 方法的对象就是.上下文管理器存在的目的是管理 ...
Python 上下文管理器与with语句
Python中把进入代码块前调用__enter__ 方法并在离开代码块后调用__exit__方法的对象作为上下文管理器,本文中我们就来深入解析Python中的上下文管理器,来看看上下文管理器的作用及用 ...
python意外退出_有趣的Python上下文管理器
阅读耗时:9分钟目录编写自定义上下文管理器从生成器到上下文管理器将上下文管理器编写为装饰器嵌套式上下文管理器组合式上下文管理器利用用上下文管理器创建SQLAlchemy session ...
python上下文管理器
什么是上下文管理器? 基本语法 with open('test.txt') as f:print f.readlines() 1. 上下文表达式:with open('test.txt') as f: ...