如何优雅地实现在Python退出时强制运行一段代码

设想这样一个场景，你要给一个项目开发测试程序，程序开始运行的时候，会创建初始环境，测试完成以后，会清理环境。

为阳谷等地区用户提供了全套网页设计制作服务，及阳谷网站建设行业解决方案。主营业务为网站制作、网站建设、阳谷网站设计，以传统方式定制建设网站，并提供域名空间备案等一条龙服务，秉承以专业、用心的态度为用户提供真诚的服务。我们深信只要达到每一位用户的要求，就会得到认可，从而选择与我们长期合作。这样，我们也可以走得更远！

这段逻辑本身非常简单：

 
 
 
 
  
  
  
  setup()   
  
  
  test()   
  
  
  clean()

但由于测试的代码比较复杂，你总是在调试的时候程序异常，导致每次clean()函数还没有来得及运行，程序就崩溃了。

你可能想到，如果这样写会怎么样呢：

 
 
 
 
  
  
  
  setup()   
  
  
  try:   
  
  
      text()   
  
  
  except Exception as e:   
  
  
      print('运行异常：', e)   
  
  
  clean()

似乎看起来，程序一定会运行到clean()函数，但是，如果你代码写的多，你就应该知道，滥用try...except...会让你非常痛苦。例如它突然给你打印一个运行异常: 1。你根本不知道是哪里出了问题，也不知道具体出了什么问题。为了找到问题，你必须让程序把错误爆出来。但这样一来，clean()又不能正常运行了。

有什么办法，既能让程序报错，又能在报错已经还能运行clean()呢?

这个时候，我们就可以使用Python自带的atexit这个模块了。它的使用方法非常简单：

 
 
 
 
  
  
  
  import atexit   
  
  
     
  
  
  @atexit.register   
  
  
  def clean():   
  
  
      print('清理环境相关的代码')   
  
  
     
  
  
  setup()   
  
  
  test()

这样一来，我们不需要显式调用clean函数了。无论程序正常结束，还是程序异常报错，clean函数里面的内容总会执行。

如下图所示：

atexit使用中有下面几个注意事项：

你可以注册多个退出函数，他们会按照注册时间从晚到早以此执行。例如：

 
 
 
 
  
  
  
  import atexit   
  
  
     
  
  
  @atexit.register   
  
  
  def clean_1():   
  
  
      ...   
  
  
     
  
  
  @atexit.register   
  
  
  def clean_2():   
  
  
      ...

会先运行clean_2()后运行clean_1()

如果clean()函数有参数，那么你可以不用装饰器，而是直接调用atexit.register(clean_1, 参数1, 参数2, 参数3='xxx')。
如果程序是被你没有处理过的系统信号杀死的，那么注册的函数无法正常执行。
如果发生了严重的Python内部错误，你注册的函数无法正常执行。
如果你手动调用了os._exit()，你注册的函数无法正常执行。

分享题目：如何优雅地实现在Python退出时强制运行一段代码
文章链接：http://www.mswzjz.cn/qtweb/news4/543554.html

攀枝花网站建设、攀枝花网站运维推广公司-贝锐智能，是专注品牌与效果的网络营销公司；服务项目有等

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源：贝锐智能

贝锐智能技术为您推荐以下文章