Python进阶系列 - 20讲 with (1)

进入上下文管理<<<
一些操作…
<<<退出上下文管理>>>

### 异常处理


如果发生异常，Python 会将类型、值和回溯传递给 **exit** 方法。  
 它可以在这里处理异常。  
 如果 `__exit__` 方法返回 `True` 以外的任何内容，则 `with` 语句会引发异常。


代码：

class ManagedFile:
def __init__(self, filename):
print(‘开始’, filename)
self.filename = filename
def __enter__(self):
print(‘>>>进入上下文管理<<<’)
self.file = open(self.filename, ‘w’)
return self.file
def __exit__(self, exc_type, exc_value, exc_traceback):
if self.file:
self.file.close()
print(‘执行：’, exc_type, exc_value)
print(‘<<<退出上下文管理>>>’)

#没有异常
with ManagedFile(‘notes.txt’) as f:
print(‘一些操作…’)
f.write(‘写入内容…’)
print(‘继续…’)

跑出异常，文件仍然关闭。

with ManagedFile(‘notes2.txt’) as f:
print(‘一些操作…’)
f.write(‘写入内容…’)
f.do_something() # ❌这个方法会抛出异常。
print(‘继续…’)

结果：

开始 notes.txt

进入上下文管理<<<
一些操作…
执行： None None
<<<退出上下文管理>>>
继续…
开始 notes2.txt
进入上下文管理<<<
一些操作…
执行： <class ‘AttributeError’> ‘_io.TextIOWrapper’ object has no attribute ‘do_something’
<<<退出上下文管理>>>

---

AttributeError Traceback (most recent call last)
/var/folders/p9/ny6y7zmj0qdblv697s3_vj040000gn/T/ipykernel_68956/2387064930.py in
23 print(‘一些操作…’)
24 f.write(‘写入内容…’)
—> 25 f.do_something() # ❌这个方法会抛出异常。
26 print(‘继续…’)

AttributeError: ‘_io.TextIOWrapper’ object has no attribute ‘do_something’

> 
> 我们可以在 `__exit__` 方法中处理异常并返回 `True`。
> 
> 
> 


代码：

class ManagedFile:
def __init__(self, filename):
print(‘开始’, filename)
self.filename = filename
def __enter__(self):
print(‘>>>进入上下文管理<<<’)
self.file = open(self.filename, ‘w’)
return self.file
def __exit__(self, exc_type, exc_value, exc_traceback):
if self.file:
self.file.close()
if exc_type is not None:
print(‘管理器内：处理异常’)
print(‘<<<退出上下文管理>>>’)
return True
with ManagedFile(‘notes2.txt’) as f:
print(‘一些操作…’)
f.write(‘写入内容…’)
f.do_something()
print(‘继续…’)

结果：

开始 notes2.txt

进入上下文管理<<<
一些操作…
管理器内：处理异常
<<<退出上下文管理>>>
继续…

### 实现上下文管理为生成器


除了编写一个类，我们还可以编写一个生成器函数并使用 `contextlib.contextmanager` 装饰器来装饰它。  
 然后我们也可以使用 `with` 语句调用该函数。  
 对于这种方法，函数必须在 `try` 语句中`yield` 资源，并且释放资源的 `__exit__` 方法的所有内容现在都放在相应的 `finally` 语句中。


示列代码：

from contextlib import contextmanager
@contextmanager
def open_managed_file(filename):
f = open(filename, ‘w’)
try:
print(‘>>>进入上下文管理<<<’)
yield f
finally:
print(‘<<<退出上下文管理>>>’)
f.close()

with open_managed_file(‘notes.txt’) as f:
print(“一些操作…”)
f.write(‘写入内容…’)

生成器首先获取资源。  
 然后它会暂时挂起自己的执行并 *yields* 资源，以便调用者可以使用它。  
 当调用者离开 `with` 上下文时，生成器继续执行并释放 `finally` 语句中的资源。




![在这里插入图片描述](https://img-blog.csdnimg.cn/20210511152217670.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3poaWd1aWd1,size_16,color_FFFFFF,t_70)

**感谢每一个认真阅读我文章的人，看着粉丝一路的上涨和关注，礼尚往来总是要有的：**



①　2000多本Python电子书（主流和经典的书籍应该都有了）

②　Python标准库资料（最全中文版）

③　项目源码（四五十个有趣且经典的练手项目及源码）

④　Python基础入门、爬虫、web开发、大数据分析方面的视频（适合小白学习）

⑤ Python学习路线图（告别不入流的学习）




**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化学习资料的朋友，可以戳这里无偿获取](https://bbs.csdn.net/topics/618317507)**

**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**