python with上下文管理器

2021-02-07  本文已影响0人  鸟它鸟

了解上下文管理器前,我们来看一个try方法使用的案例

In [11]: def exe_try():
    ...:     try:
    ...:         print ("code started")
    ...:         raise KeyError    # 主动抛出异常
    ...:         return 1
    ...:     except KeyError as e:   # 捕获KeyError异常
    ...:         print ("key error")
    ...:         return 2                        # 捕获后执行return返回
    ...:     else:                                   # 未捕获到异常时执行else中内容
    ...:         print ("other error")
    ...:         return 3
    ...:     finally:                              # 不管try语句中执行是否正常,都执行
    ...:         print ("finally")
    ...:         return 4
    ...:

In [12]: exe_try()
code started
key error
finally
Out[12]: 4

在这个执行过程中,我们发现一个问题,最后的return返回的是4 而不是2
这个是因为,在KeyError捕获异常后,执行了相关的代码块,但是由于定义了finally语句,所以KeyError中返回的2被压如栈中,finally执行完成后,返回的4也被压如栈中,根据栈先入后出的特性,最终返回了4而不是2.
当我们把finally中的return 注释掉后,返回的就是2了,可以试试。

我们讲上下文管理器,为什么要说这个try的用法呢? 想一个场景你要打开一个文件,执行完逻辑后要关闭掉该文件。但是打开后,执行过程中代码抛异常了,如果捕获没有做好,那么这个文件可能处于一直被打开的状态。
此时,可以使用finally方法,保证文件无论什么情况都可以被关闭掉。

在开发场景中,你可以使用这种模式去做,但是有更简单的方案,我们一起看下with

class Sample:
    def __enter__(self):
        print ("enter")
        #获取资源
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        #释放资源
        print ("exit")
    def do_something(self):
        print ("doing something")

In [15]: with Sample() as sample:
    ...:     sample.do_something()
    ...:
enter
doing something
exit

我们定义一个类 Sample,类中定义2个魔法函数enterexitenter 用户定义的方法执行前运行、exit是用户定义的方法后运行。
根据这个特性,如果要操作一个文件,我们可以在enter做打开,exit中做关闭,中途的方法可以自由的进行编写。

上一篇下一篇

猜你喜欢

热点阅读