• Python中with的用法


    在Python中,with语句的目的是简化try/finally的模式,使得代码可以保证运行完毕后执行某一项操作,哪怕是发生了异常、return等调用而终止,也一定会执行指定的操作。为了使代码可以使用with语句,with语句中的类必须实现__enter__和__exit__两个方法,分别在with语句开始运行前和执行后运行。
    比如我们平常用得最多的用法:

    with open('mirror.py') as fp:
        src = fp.read(60) 
    

    读取文件用的就是with语句,open返回一个文件类型TextIOWrapper,传值给fp。可以对fp执行读取文件操作,当出现异常时会执行TextIOWrapper的__exit__方法。
    下面看一个更复杂的例子,摘抄自《流畅的Python》,虽然这段代码我试过在Python2.7中无法运行(由于无法更改stdout.write)函数,但可以一窥其中的思想:

    class LookingGlass:
        def __enter__(self):
    	import sys
    	self.original_write = sys.stdout.write
    	sys.stdout.write = self.reverse_write
    	return "Jacy"
    	
        def reverse_write(self, text):
    	self.original_write(text[::-1])
    	
        def __exit__(self, exc_type, exc_value, traceback):
    	import sys
    	sys.stdout.write = self.original_write
    	if exc_type is ZeroDivisionError:
    	    print "zero division error"
    	    return True
    
    with LookingGlass() as what:
        print "El psy congroo"
        print what
    print what
    

    按照书中的解释,在LookingGlass对象__enter__时,将Python输出的函数予以替换,这样可以将输入的文字全部倒置;然后with语句中输出的所有文本就都是倒置的。在__exit__时再将输出流设定回初始值。其中__enter__的返回值就是what,因此两次输出what分别是“ycaJ”和“Jacy”。
    最后在__exit__函数中可以处理with块的异常情况,比如上面的示例代码就处理了ZeroDivisionError的情况。该函数返回True代表异常被正确地处理,否则会将异常继续抛出。

  • 相关阅读:
    pytest 失败重试
    mysql 增删改查
    jenkins分布式启动slave发现ip不对应实际ip
    测试架构师必读:编码规范、开发技巧、架构画图
    mysql limit和offset用法
    pytest 使用fixture给测试函数传参
    mysql union和union all
    pytest setup和teardown用法
    pytest的Hook函数详解
    pytest 用例依赖
  • 原文地址:https://www.cnblogs.com/wickedpriest/p/12804822.html
Copyright © 2020-2023  润新知