python的with语句

上下文管理协议(Context Management Protocol):包含方法 enter()和__exit__(),支持该协议的对象要实现这两个方法。
上下文管理器(Context Manager):支持上下文管理协议的对象,这种对象实现了__enter__()和__exit__()方法。上下文管理器定义执行with语句时要建立的运行时上下文,负责执行with语句块上下文中的进入与退出操作。通常使用with语句调用上下文管理器,也可以通过直接调用其方法来使用。
with语句基本用法:

1 with EXPR as VAR:
2 
3     BLOCK

1、执行EXPR,生成上下文管理器context_manager;
2、获取上下文管理器的__exit()方法,并保存起来用于之后的调用;
3、调用上下文管理器的__enter
()方法;如果使用了as子句,则将__enter__()方法的返回值赋值给as子句中的VAR;
4、执行BLOCK中的表达式(即你要执行的语句);
5、不管是否执行过程中是否发生了异常,执行上下文管理器的__exit__()方法,exit()方法负责执行“清理”工作,如释放资源等。如果执行过程中没有出现异常,或者语句体中执行了语句break/continue/return,则以None作为参数调用__exit__(None, None, None);如果执行过程中出现异常,则使用sys.exc_info得到的异常信息为参数调用__exit__(exc_type, exc_value, exc_traceback);
出现异常时,如果__exit__(type, value, traceback)返回False,则会重新抛出异常,让with之外的语句逻辑来处理异常,这也是通用做法;如果返回True,则忽略异常,不再对异常进行处理。

class Sample:
    def __enter__(self):#重写上下文管理器对象context_manager的enter方法
        print ("In __enter__()")
        return "Foo"
 
    def __exit__(self, type, value, trace):#重写上下文管理器对象context_manager的__exit__方法
        print ("In __exit__()")
 
def get_sample():#
    return Sample() 
#在python中实现了__enter__和__exit__方法,即支持上下文管理器协议。上下文管理器就是支持上下文管理器协议的对象,
#它是为了with而生。当with语句在开始运行时,会在上下文管理器对象上调用 __enter__ 方法。
#with语句运行结束后,会在上下文管理器对象上调用 __exit__ 方法
with get_sample() as sample:
    print("sample:", sample) 

结果:
In enter()
sample: Foo
In exit()

例子:

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

type: <class ‘ZeroDivisionError’>
value: division by zero
trace: <traceback object at 0x0000027345C3AD88>
in do_something(self)
13
14 def do_something(self):
—> 15 bar = 1/0
16 return bar + 10
17

ZeroDivisionError: division by zero

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值