上下文管理器和else快
类似于 then 的 else
for...else... 仅在 for 循环运行完毕后运行 else, 不能被 break
while...else... 仅在 while 条件为 false 而退出后运行 else, 不能被 break
try...else... 仅在 try 没有抛出异常后, 运行 else
注: 若存在 break, continue, return 把控制权跳到了复合语句的主块外, else 也会被跳过
for item in ('A', 'B', 'C'):
if item == 'D':
break
else:
raise ValueError('No D')
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
Cell In [4], line 5
3 break
4 else:
----> 5 raise ValueError('No D')
ValueError: No D
try:
print("RUA")
except OSError:
print("WARMING")
else:
print("after rua")
RUA
after rua
上下文管理器和with块
with 化简 try...finally... 模式
上下文管理器协议包含 __enter__ 和 __exit__
with 后跟 上下文管理器对象
as xxx 可选
__exit__ 中返回 True 以外的值, with 中的任何异常会向上冒泡, 默认返回 None, 即会向上冒泡异常
with open(r'C:\Users\qiany\OneDrive\Document\Code\python\study\FluentPython\data\rua.txt') as fp:
src = fp.read(60)
print(src)
print(fp.encoding) # fp 依旧可以使用(仅限读取属性)
print(fp.read(60)) # 不能在 fp 上执行 I/O 操作, 因为 with 末尾调用 __exit__将文件关闭了
ruaruaruaruaruaruaruaruaruaruaruaruaruaruaruaruaruaruaruarua
cp936
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
Cell In [12], line 6
3 print(src)
5 print(fp.encoding) # fp 依旧可以使用(仅限读取属性)
----> 6 print(fp.read(60)) # 不能在 fp 上执行 I/O 操作, 因为 with 末尾调用 __exit__将文件关闭了
ValueError: I/O operation on closed file.
class LookingGlass:
def __enter__(self): # Python 调用 enter 不会传入任何参数
import sys
self.original_write = sys.stdout.write
sys.stdout.write = self.reverse_write # 猴子补丁
return 'RUAAAAA'
def reverse_write(self, text):
self.original_write(text[::-1])
def __exit__(self, exc_type, exc_value, traceback):
import sys # python 会缓存导入的模块, 重复导入 sys 不会浪费资源
sys.stdout.write = self.original_write
if exc_type == ZeroDivisionError:
print('Dont divide zero')
return True # __exit__中返回True以外的值, with中的任何异常会向上冒泡
print('--- 使用 with 方法 ---')
with LookingGlass() as what:
print("QAQ233")
print(what)
print("what", what)
print('--- 不使用 with 方法 ---')
manager = LookingGlass()
monster = manager.__enter__()
print(monster)
manager.__exit__(None, None, None)
--- 使用 with 方法 ---
332QAQ
AAAAAUR
what RUAAAAA
--- 不使用 with 方法 ---
AAAAAUR
exc_type 异常类 eg.ZeroDivisionError
exc_value 异常实例 exc_value.args 可捕获 Error(value) 的 value
traceback traceback对象
contextlib 模块中的实用工具
| 方法 | 释义 |
|---|---|
closing |
如果对象提供了 close() 方法, 但没有实现 __enter__/__exit__ 协议, 那么可通过这个函数构造上下文 |
suppress |
构建临时忽略指定异常的上下文管理器 |
@contextmanager |
把 生成器函数 变成 上下文管理器 |
ContextDecorator |
基类, 用于定义 基于类 的上下文管理器 |
ExitStack |
这个 上下文管理器 能进入 多个 上下文管理器。with 结束时, ExitStack 按 后进先出 的顺序调用栈中各个上下文管理器的 __exit__。eg.同时打开多个文件 |
使用 @contextmanager
在使用 @contextmanager 装饰的生成器中, yield语句的作用是把函数的定义体分成两部分,
-
yield之前的代码在with块开始时(解释器调用__enter__方法时) 执行 -
yield之后的代码在with块结束时(解释器调用__exit__方法时) 执行 -
yield的值绑定到as的目标变量上去
注意: yield 语句最好放在 try...finally... 或在 with 语句中, 因为我们永远不知道用户会在 with 中干什么
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 'RUAAAAAA'
sys.stdout.write = original_write
with looking_glass() as what:
print("QAQ233")
print(what)
332QAQ
AAAAAAUR
# 加入异常处理
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
msg = ''
try:
yield 'RUAAAAAA'
except ZeroDivisionError:
msg = 'Dont divie by zero'
finally:
sys.stdout.write = original_write
if msg:
print(msg)
with looking_glass() as what:
print("QAQ233")
print(what)