脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Python with 语句

Python 的 with 语句详解

 更新时间：2014年06月13日 10:11:27   投稿：junjie  
这篇文章主要介绍了Python 的 with 语句,本文详细讲解了with语句、with语句的历史、with语句的使用例子等,需要的朋友可以参考下

一、简介

with是从Python 2.5 引入的一个新的语法，更准确的说，是一种上下文的管理协议，用于简化try…except…finally的处理流程。with通过__enter__方法初始化，然后在__exit__中做善后以及处理异常。对于一些需要预先设置，事后要清理的一些任务，with提供了一种非常方便的表达。

with的基本语法如下，EXPR是一个任意表达式，VAR是一个单一的变量（可以是tuple），”as VAR”是可选的。

复制代码 代码如下:

with EXPR as VAR:
    BLOCK

根据PEP 343的解释，with…as…会被翻译成以下语句：
复制代码 代码如下:

mgr = (EXPR)
exit = type(mgr).__exit__  # Not calling it yet
value = type(mgr).__enter__(mgr)
exc = True
try:
    try:
        VAR = value  # Only if "as VAR" is present
        BLOCK
    except:
        # The exceptional case is handled here
        exc = False
        if not exit(mgr, *sys.exc_info()):
            raise
        # The exception is swallowed if exit() returns true
finally:
    # The normal and non-local-goto cases are handled here
    if exc:
        exit(mgr, None, None, None)

为什么这么复杂呢？注意finally中的代码，需要BLOCK被执行后才会执行finally的清理工作，因为当EXPR执行时抛出异常，访问mgr.exit执行就会报AttributeError的错误。


二、实现方式

根据前面对with的翻译可以看到，被with求值的对象必须有一个__enter__方法和一个__exit__方法。稍微看一个文件读取的例子吧，注意在这里我们要解决2个问题：文件读取异常，读取完毕后关闭文件句柄。用try…except一般会这样写：

复制代码 代码如下:

f = open('/tmp/tmp.txt')
try:
    for line in f.readlines():
        print(line)
finally:
    f.close()

注意我们这里没有处理文件打开失败的IOError，上面的写法可以正常工作，但是对于每个打开的文件，我们都要手动关闭文件句柄。如果要使用with来实现上述功能，需要需要一个代理类：
复制代码 代码如下:

class opened(object):

    def __init__(self, name):
        self.handle = open(name)

    def __enter__(self):
        return self.handle

    def __exit__(self, type, value, trackback):
        self.handle.close()

with opened('/tmp/a.txt') as f:
    for line in f.readlines():
        print(line)


注意我们定了一个名字叫opened的辅助类，并实现了__enter__和__exit__方法，__enter__方法没有参数，__exit__方法的3个参数，分别代表异常的类型、值、以及堆栈信息，如果没有异常，3个入参的值都为None。

如果你不喜欢定义class，还可以用Python标准库提供的contextlib来实现：

复制代码 代码如下:

from contextlib import contextmanager

@contextmanager
def opened(name):
    f = open(name)
    try:
        yield f
    finally:
        f.close()

with opened('/tmp/a.txt') as f:
    for line in f.readlines():
        print(line)


使用contextmanager的函数，yield只能返回一个参数，而yield后面是处理清理工作的代码。在我们读取文件的例子中，就是关闭文件句柄。这里原理上和我们之前实现的类opened是相同的，有兴趣的可以参考一下contextmanager的源代码。

三、应用场景

废话了这么多，那么到底那些场景下该使用with，有没有一些优秀的例子？当然啦，不然这篇文章意义何在。以下摘自PEP 343。

一个确保代码执行前加锁，执行后释放锁的模板：

复制代码 代码如下:

@contextmanager
    def locked(lock):
        lock.acquire()
        try:
            yield
        finally:
            lock.release()

    with locked(myLock):
        # Code here executes with myLock held.  The lock is
        # guaranteed to be released when the block is left (even
        # if via return or by an uncaught exception).


数据库事务的提交和回滚：
复制代码 代码如下:

@contextmanager
        def transaction(db):
            db.begin()
            try:
                yield None
            except:
                db.rollback()
                raise
            else:
                db.commit()

重定向stdout：
复制代码 代码如下:

@contextmanager
def stdout_redirected(new_stdout):
    save_stdout = sys.stdout
    sys.stdout = new_stdout
    try:
        yield None
    finally:
        sys.stdout = save_stdout

with opened(filename, "w") as f:
    with stdout_redirected(f):
        print "Hello world"


注意上面的例子不是线程安全的，再多线程环境中要小心使用。


四、总结

with是对try…expect…finally语法的一种简化，并且提供了对于异常非常好的处理方式。在Python有2种方式来实现with语法：class-based和decorator-based，2种方式在原理上是等价的，可以根据具体场景自己选择。

with最初起源于一种block…as…的语法，但是这种语法被很多人所唾弃，最后诞生了with，关于这段历史依然可以去参考PEP-343和PEP-340

您可能感兴趣的文章:

相关文章

  • 深入解析Python中的集合类型操作符

    深入解析Python中的集合类型操作符

    这篇文章主要介绍了深入解析Python中的集合类型操作符,是Python入门学习中的基础知识,需要的朋友可以参考下
    2015-08-08
  • python httpx的具体使用

    python httpx的具体使用

    本文主要介绍了python httpx的具体使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-04-04
  • Python requests及aiohttp速度对比代码实例

    Python requests及aiohttp速度对比代码实例

    这篇文章主要介绍了Python requests及aiohttp速度对比代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-07-07
  • Python使用pyh生成HTML文档的方法示例

    Python使用pyh生成HTML文档的方法示例

    这篇文章主要介绍了Python使用pyh生成HTML文档的方法示例,小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2018-03-03
  • 对numpy中数组转置的求解以及向量内积计算方法

    对numpy中数组转置的求解以及向量内积计算方法

    今天小编就为大家分享一篇对numpy中数组转置的求解以及向量内积计算方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2018-10-10
  • Python常见字典内建函数用法示例

    Python常见字典内建函数用法示例

    这篇文章主要介绍了Python常见字典内建函数用法,结合具体实例形式分析了Python字典内建函数实现针对字典的长度、hash运算、键值对遍历、获取、计算等相关操作技巧,需要的朋友可以参考下
    2018-05-05
  • Python实现的单向循环链表功能示例

    Python实现的单向循环链表功能示例

    这篇文章主要介绍了Python实现的单向循环链表功能,简单描述了单向循环链表的概念、原理并结合实例形式分析了Python定义与使用单向循环链表的相关操作技巧,需要的朋友可以参考下
    2017-11-11
  • Python使用turtle库绘制科赫雪花详细代码

    Python使用turtle库绘制科赫雪花详细代码

    科赫曲线是一种分形,其形态非常像雪花,因此又被称作科赫雪花、雪花曲线,这篇文章主要给大家介绍了关于Python使用turtle库绘制科赫雪花的相关资料,需要的朋友可以参考下
    2024-05-05
  • 分析PyTorch Dataloader报错ValueError:num_samples的另一种可能原因

    分析PyTorch Dataloader报错ValueError:num_samples的另一种可能原因

    这篇文章主要介绍了分析PyTorch Dataloader报错ValueError:num_samples的另一种可能原因,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2024-02-02
  • Python的图像处理库Pillow安装与使用教程

    Python的图像处理库Pillow安装与使用教程

    Pillow库是Python中用于图像处理的开源库,提供了丰富的图像处理功能,如图像读取、保存、裁剪、调整大小、旋转、添加文字等,这篇文章主要给大家介绍了关于Python的图像处理库Pillow安装与使用的相关资料,需要的朋友可以参考下
    2024-04-04

最新评论