Python 高手编程系列九:上下文管理器 — with 语句

为了确保即使在出现错误的情况下也能运行某些清理代码,try...finally 语句是

很有用的。这一语句有许多使用场景,例如:

• 关闭一个文件。

• 释放一个锁。

• 创建一个临时的代码补丁。

• 在特殊环境中运行受保护的代码。

with 语句为这些使用场景下的代码块包装提供了一种简单方法。即使该代码块引发了异常,你也可以在其执行前后调用一些代码。例如,处理文件通常采用这种方式:

hosts = open('/etc/hosts')

try:

... for line in hosts:

... if line.startswith('#'):

... continue

... print(line.strip())

... finally:

... hosts.close()

...

127.0.0.1 localhost

255.255.255.255 broadcasthost

::1 localhost

利用 with 语句,上述代码可以重写为:

with open('/etc/hosts') as hosts:

... for line in hosts:

... if line.startswith('#'):

... continue

... print(line.strip())

...

127.0.0.1 localhost

255.255.255.255 broadcasthost

::1 localhost

在前面的示例中,open 的作用是上下文管理器,确保即使出现异常也要在执行完 for

循环之后关闭文件。

与这条语句兼容的其他项目是来自 threading 模块的类:

• threading.Lock

• threading.RLock

• threading.Condition

• threading.Semaphore

• threading.BoundedSemaphore

一般语法和可能的实现

with 语句的一般语法的最简单形式如下:

with context_manager:

代码块

...

此外,如果上下文管理器提供了上下文变量,可以用 as 子句保存为局部变量:

with context_manager as context:

代码块

...

注意,多个上下文管理器可以同时使用,如下所示:

with A() as a, B() as b:

...

这种写法等价于嵌套使用,如下所示:

with A() as a:

with B() as b:

...

作为一个类

任何实现了上下文管理器协议(context manager protocol)的对象都可以用作上下文管

理器。该协议包含两个特殊方法。

enter (self):更多内容请访问https://docs.python.org/3.3/reference/datamodel.html

#object.enter

exit (self, exc_type, exc_value, traceback):更多内容请访问

https://docs.python.org/3.3/reference/datamodel.html#object.**exit** 。

简而言之,执行 with 语句的过程如下:

• 调用__enter__方法。任何返回值都会绑定到指定的 as 子句。

• 执行内部代码块。

• 调用__exit__方法。

__exit__接受代码块中出现错误时填入的 3 个参数。如果没有出现错误,那么这 3 个

参数都被设为 None。出现错误时,__exit__不应该重新引发这个错误,因为这是调用者

(caller)的责任。但它可以通过返回 True 来避免引发异常。这可用于实现一些特殊的使用

场景,例如下一节将会看到的 contextmanager 装饰器。但在大多数使用场景中,这一

方法的正确行为是执行类似于 finally 子句的一些清理工作,无论代码块中发生了什么,

它都不会返回任何内容。

下面是某个实现了这一协议的上下文管理器示例,以更好地说明其工作原理:

class ContextIllustration:

def enter (self):

print('entering context')

def exit (self, exc_type, exc_value, traceback):

print('leaving context')

if exc_type is None:

print('with no error')

else:

print('with an error (%s)' % exc_value)

没有引发异常时的运行结果如下:

with ContextIllustration():

... print("inside")

...

entering context

inside

leaving context

with no error

引发异常时的输出如下:

with ContextIllustration():

... raise RuntimeError("raised within 'with'")

...

entering context

leaving context

with an error (raised within 'with')

Traceback (most recent call last):

File "", line 2, in

RuntimeError: raised within 'with'

相关推荐
evans在进步10 分钟前
LeetCode 33:搜索旋转排序数组——Java 两阶段二分查找详解
java·python·leetcode
倒流时光三十年18 分钟前
PostgreSQL Semi-Join(半连接)通俗讲解
数据库·postgresql
言乐642 分钟前
Python游戏水平测试辅助系统2
开发语言·windows·python·游戏·django
NiceCloud喜云1 小时前
腾讯云国际版云数据库选型:MySQL、Redis 怎么按业务架构来判断
数据库·mysql·腾讯云
ew452181 小时前
MySQL JOIN 关联语法、差异、标准用法、经典坑点汇总
数据库·mysql
Rain的Java大神之路2 小时前
线上接口负载满了如何解决
java·数据库·redis·后端·缓存·面试·架构
Nturmoils2 小时前
给运营导个 CSV,在 ksql 里用 \copy 就够了
数据库
jerryinwuhan2 小时前
《Python数据预处理》第四章
开发语言·python
ciqingloveless2 小时前
Oracle打开数据库加密
数据库·oracle
雨晨源码(同名B站)2 小时前
基于Python的网易云音乐评论数据情感化分析系统 音乐爬虫信息可视化 |SnowNLP评论情感分析
开发语言·hadoop·爬虫·python·信息可视化·毕业设计