了解上下文管理器前,我们来看一个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个魔法函数enter和exit,enter 用户定义的方法执行前运行、exit是用户定义的方法后运行。
根据这个特性,如果要操作一个文件,我们可以在enter做打开,exit中做关闭,中途的方法可以自由的进行编写。
网友评论