美文网首页
python with上下文管理器

python with上下文管理器

作者: 鸟它鸟 | 来源:发表于2021-02-07 13:54 被阅读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中做关闭,中途的方法可以自由的进行编写。

    相关文章

      网友评论

          本文标题:python with上下文管理器

          本文链接:https://www.haomeiwen.com/subject/kotitltx.html