理解Python中的With语句


Posted in Python onFebruary 02, 2015

With语句是什么?

有一些任务,可能事先需要设置,事后做清理工作。对于这种场景,Python的with语句提供了一种非常方便的处理方式。一个很好的例子是文件处理,你需要获取一个文件句柄,从文件中读取数据,然后关闭文件句柄。

如果不用with语句,代码如下:

file = open("/tmp/foo.txt")

data = file.read()

file.close()

这里有两个问题。一是可能忘记关闭文件句柄;二是文件读取数据发生异常,没有进行任何处理。下面是处理异常的加强版本:

file = open("/tmp/foo.txt")

try:

    data = file.read()

finally:

    file.close()

虽然这段代码运行良好,但是太冗长了。这时候就是with一展身手的时候了。除了有更优雅的语法,with还可以很好的处理上下文环境产生的异常。下面是with版本的代码:

with open("/tmp /foo.txt") as file:

    data = file.read()

with如何工作?

这看起来充满魔法,但不仅仅是魔法,Python对with的处理还很聪明。基本思想是with所求值的对象必须有一个__enter__()方法,一个__exit__()方法。

紧跟with后面的语句被求值后,返回对象的__enter__()方法被调用,这个方法的返回值将被赋值给as后面的变量。当with后面的代码块全部被执行完之后,将调用前面返回对象的__exit__()方法。

下面例子可以具体说明with如何工作:

#!/usr/bin/env python

# with_example01.py
class Sample:

    def __enter__(self):

        print "In __enter__()"

        return "Foo"
    def __exit__(self, type, value, trace):

        print "In __exit__()"


def get_sample():

    return Sample()


with get_sample() as sample:

    print "sample:", sample

行代码,输出如下

bash-3.2$ ./with_example01.py

In __enter__()

sample: Foo

In __exit__()

正如你看到的,

__enter__()方法被执行

__enter__()方法返回的值 - 这个例子中是"Foo",赋值给变量'sample'
执行代码块,打印变量"sample"的值为 "Foo"

__exit__()方法被调用

with真正强大之处是它可以处理异常。可能你已经注意到Sample类的__exit__方法有三个参数- val, type 和 trace。 这些参数在异常处理中相当有用。我们来改一下代码,看看具体如何工作的。

#!/usr/bin/env python

# with_example02.py


class Sample:

    def __enter__(self):

        return self
    def __exit__(self, type, value, trace):

        print "type:", type

        print "value:", value

        print "trace:", trace
    def do_something(self):

        bar = 1/0

        return bar + 10
with Sample() as sample:

    sample.do_something()

这个例子中,with后面的get_sample()变成了Sample()。这没有任何关系,只要紧跟with后面的语句所返回的对象有 __enter__()和__exit__()方法即可。此例中,Sample()的__enter__()方法返回新创建的Sample对象,并赋值给变量sample。

代码执行后:

bash-3.2$ ./with_example02.py

type: <type 'exceptions.ZeroDivisionError'>

value: integer division or modulo by zero

trace: <traceback object at 0x1004a8128>

Traceback (most recent call last):

  File "./with_example02.py", line 19, in <module>

    sample.do_somet hing()

  File "./with_example02.py", line 15, in do_something

    bar = 1/0

ZeroDivisionError: integer division or modulo by zero

实际上,在with后面的代码块抛出任何异常时,__exit__()方法被执行。正如例子所示,异常抛出时,与之关联的type,value和stack trace传给__exit__()方法,因此抛出的ZeroDivisionError异常被打印出来了。开发库时,清理资源,关闭文件等等操作,都可以放在__exit__方法当中。

因此,Python的with语句是提供一个有效的机制,让代码更简练,同时在异常产生时,清理工作更简单。

示例代码可以在Github上面找到。

Python 相关文章推荐
python中pycurl库的用法实例
Sep 30 Python
python实现堆栈与队列的方法
Jan 15 Python
Python使用Supervisor来管理进程的方法
May 28 Python
Python正则获取、过滤或者替换HTML标签的方法
Jan 28 Python
Python+selenium实现截图图片并保存截取的图片
Jan 05 Python
对pandas数据判断是否为NaN值的方法详解
Nov 06 Python
不到20行代码用Python做一个智能聊天机器人
Apr 19 Python
从列表或字典创建Pandas的DataFrame对象的方法
Jul 06 Python
python腾讯语音合成实现过程解析
Aug 01 Python
pandas 对group进行聚合的例子
Dec 27 Python
python之MSE、MAE、RMSE的使用
Feb 24 Python
Django windows使用Apache实现部署流程解析
Oct 12 Python
Linux环境下MySQL-python安装过程分享
Feb 02 #Python
Python中用pycurl监控http响应时间脚本分享
Feb 02 #Python
Python列表(list)常用操作方法小结
Feb 02 #Python
Python Sleep休眠函数使用简单实例
Feb 02 #Python
Python中实现从目录中过滤出指定文件类型的文件
Feb 02 #Python
Python实现二分法算法实例
Feb 02 #Python
Python标准异常和异常处理详解
Feb 02 #Python
You might like
5种PHP创建数组的实例代码分享
2014/01/17 PHP
thinkphp模板用法和内容输出实例
2014/11/28 PHP
PHP date()格式MySQL中插入datetime方法
2019/01/29 PHP
php反序列化长度变化尾部字符串逃逸(0CTF-2016-piapiapia)
2020/02/15 PHP
javascript模版引擎-tmpl的bug修复与性能优化分析
2011/10/23 Javascript
使用JSON.parse将json字符串转换成json对象的时候会出错
2014/09/04 Javascript
jQuery子窗体取得父窗体元素的方法
2015/05/11 Javascript
jquery左右全屏大尺寸多图滑动效果代码分享
2015/08/28 Javascript
详解Javascript中prototype属性(推荐)
2016/09/03 Javascript
javascript 的变量、作用域和内存问题
2017/04/19 Javascript
JS实现的找零张数最小问题示例
2017/11/28 Javascript
bmob js-sdk 在vue中的使用教程
2018/01/21 Javascript
JS实现显示当前日期的实例代码
2018/07/03 Javascript
微信小程序实现红包雨功能
2018/07/11 Javascript
通过jquery的ajax请求本地的json文件方法
2018/08/08 jQuery
vue实现微信获取用户信息的方法
2019/03/21 Javascript
es6中使用map简化复杂条件判断操作实例详解
2020/02/19 Javascript
leaflet加载geojson叠加显示功能代码
2020/02/21 Javascript
vue使用nprogress加载路由进度条的方法
2020/06/04 Javascript
Postman环境变量全局变量使用方法详解
2020/08/13 Javascript
vue实现井字棋游戏
2020/09/29 Javascript
[02:05]2014DOTA2国际邀请赛 BBC外卡赛赛后总结
2014/07/09 DOTA
[01:06:26]全国守擂赛第二周 Team Coach vs DeMonsTer
2020/04/28 DOTA
Python类的基础入门知识
2008/11/24 Python
Python 实现数据库更新脚本的生成方法
2017/07/09 Python
python实现音乐下载的统计
2018/06/20 Python
python批量解压zip文件的方法
2019/08/20 Python
Django利用elasticsearch(搜索引擎)实现搜索功能
2020/11/26 Python
基于CSS3的CSS 多栏(Multi-column)实现瀑布流源码分享
2014/06/11 HTML / CSS
大学生咖啡店创业计划书
2014/01/21 职场文书
高中生期末评语
2014/01/28 职场文书
接待员岗位责任制
2014/02/10 职场文书
幽灵公主观后感
2015/06/09 职场文书
2015年物流客服工作总结
2015/07/27 职场文书
Vue图片裁剪组件实例代码
2021/07/02 Vue.js
生命的关键成分来自太空?陨石说是的
2022/04/29 数码科技