Python with语句

2018-04-06  本文已影响0人  青铜搬砖工

本人学习地址:https://blog.csdn.net/bitcarmanlee/article/details/52745676

1.with语法

with 上下文表达式 as 对象名
  语句体

2.名词解释

3.小栗子

(1).file

常见的文件操作中,我们经常忘记关闭文件句柄,或者会写一串try...except去捕获异常,可以是用with去简化代码:

with open("/temp.txt") as file:
    data = file.read()
image.png
open方法返回一个file,file 是不是我们要找的上下文管理器呢?
image.png
file 类里实现了__enter()__,__exit()__方法,所以为上下文管理器,所以file可以使用with语句。

(2)自定义上下文管理器

class sample():
    def __enter__(self):
        print('enter')
    def __exit__(self, exc_type, exc_val, exc_tb):
        print ('exit')

with sample() as sample:
    print('with body')

运行结果


image.png

从中可以看出,执行顺序为: enter方法 ->withbody->exit方法

class CustomException(Exception):
    def __init__(self):
        print('custom exception')
        Exception.__init__(self, 'custom exception')

class sample():
    def __enter__(self):
        print('enter')
    def __exit__(self, exc_type, exc_val, exc_tb):
        if isinstance(exc_val,CustomException):
            print ('exit custom exception')
        print ('exit')

with sample() as sample:
    print('with body 1')
    raise CustomException()
    print('with body 2')
image.png

这次执行顺序为 enter->with body 1->custom exception->exit custom exception->exit
说明当抛出异常时,也是执行exit方法。

上一篇下一篇

猜你喜欢

热点阅读