Python with用法:自动关闭文件进程


Posted in Python onJuly 10, 2019

实际上,Python 提供了 with 语句来管理资源关闭。比如可以把打开的文件放在 with 语句中,这样 with 语句就会帮我们自动关闭文件。

with 语句的语法格式如下:

with context expression [as target(s)]:
  with 代码块

在上面的语法格式中,context_expression 用于创建可自动关闭的资源。

例如,程序使用 with 语句来读取文件:

import codecs
# 使用with语句打开文件,该语句会负责关闭文件
with codecs.open("readlines_test.py", 'r', 'utf-8', buffering=True) as f:
  for line in f:
    print(line, end='')

程序也可以使用 with 语句来处理通过 fileinput.input 合并的多个文件,例如如下程序:

import fileinput
# 使用with语句打开文件,该语句会负责关闭文件
with fileinput.input(files=('test.txt', 'info.txt')) as f:
  for line in f:
    print(line, end='')

上面两个程序都使用了 with 语句来管理资源,因此它们都不需要显式关闭文件。

那么,with 语句的实现原理是什么?其实很简单,使用 with 语句管理的资源必须是一个实现上下文管理协议(context manage protocol)的类,这个类的对象可被称为上下文管理器。要实现上下文管理协议,必须实现如下两个方法:

  • context_manager.__enter__():进入上下文管理器自动调用的方法。该方法会在 with 代码块执行之前执行。如果 with 语句有 as子句,那么该方法的返回值会被赋值给 as 子句后的变量;该方法可以返回多个值,因此,在 as 子句后面也可以指定多个变量(多个变量必须由“()”括起来组成元组)。
  • context_manager.__exit__(exc_type, exc_value, exc_traceback):退出上下文管理器自动调用的方法。该方法会在 with 代码块执行之后执行。如果 with 代码块成功执行结束,程序自动调用该方法,调用该方法的三个参数都为 None:如果 with 代码块因为异常而中止,程序也自动调用该方法,使用 sys.exc_info 得到的异常信息将作为调用该方法的参数。

通过上面的介绍不难发现,只要一个类实现了 __enter__() 和 __exit__(exc_type, exc_value, exc_traceback) 方法,程序就可以使用 with 语句来管理它;通过 __exit__() 方法的参数,即可判断出 with 代码块执行时是否遇到了异常。

换而言之,上面程序所用的文件对象、FileInput 对象,其实都实现了这两个方法,因此它们都可以接受 with 语句的管理。

下面我们自定义一个实现上下文管理协议的类,并使用 with 语句来管理它:

class FkResource:
  def __init__(self, tag):
    self.tag = tag
    print('构造器,初始化资源: %s' % tag)
  # 定义__enter__方法,with体之前的执行的方法
  def __enter__(self):
    print('[__enter__ %s]: ' % self.tag)
    # 该返回值将作为as子句中变量的值
    return 'fkit' # 可以返回任意类型的值
  # 定义__exit__方法,with体之后的执行的方法
  def __exit__(self, exc_type, exc_value, exc_traceback):
    print('[__exit__ %s]: ' % self.tag)
    # exc_traceback为None,代表没有异常
    if exc_traceback is None:
      print('没有异常时关闭资源')
    else:
      print('遇到异常时关闭资源')
      return False  # 可以省略,默认返回None也被看做是False
with FkResource('孙悟空') as dr:
  print(dr)
  print('[with代码块] 没有异常')
print('------------------------------')
with FkResource('白骨精'):
  print('[with代码块] 异常之前的代码')
  raise Exception
  print('[with代码块] ~~~~~~~~异常之后的代码')

上面程序定义了一个 FkResource 类,该类定义了 __enter__() 和 __exit__() 两个方法,因此该类的对象可以被 with 语句管理:

  • 程序在执行 with 代码块之前,会执行 __enter__() 方法,并将该方法的返回值赋值给 as 子句后的变量。
  • 程序在执行 with 代码块之后,会执行 __exit__() 方法,可以根据该方法的参数来判断 with 代码块是否有异常。

程序两次使用 with 语句管理 FkResource 对象。第一次,with 代码块没有出现异常。第二次,with 代码块出现了异常。大家可以看到,使用 with 语句两次对 FkResource 的管理略有差异(主要是在 __exit()__ 方法中略有差异)。

运行上面的程序,可以看到如下输出结果:

构造器,初始化资源: 孙悟空
[__enter__ 孙悟空]:
fkit
[with代码块] 没有异常
[__exit__ 孙悟空]:
没有异常时关闭资源
------------------------------
构造器,初始化资源: 白骨精
[__enter__ 白骨精]:
[with代码块] 异常之前的代码
[__exit__ 白骨精]:
遇到异常时关闭资源
Traceback (most recent call last):
 File "C:\Users\mengma\Desktop\1.py", line 26, in <module>
  raise Exception
Exception

从上面的输出结果来看,使用 with 语句管理资源,程序总可以在进入 with 代码块之前自动执行 __enter__() 方法,无论 with 代码块是否有异常,这个部分都是一样的,而且 __enter__() 方法的返回值被赋值给了 as 子句后的变量,如上面的 ① 号输出信息所示。

对于 with 代码块有异常和无异常这两种情况,此时主要通过 exit() 方法的参数进行判断,程序可针对 with 代码块是否有异常分别进行处理,如程序中代码所示。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
详解Python Socket网络编程
Jan 05 Python
python中异常捕获方法详解
Mar 03 Python
分享一个简单的python读写文件脚本
Nov 25 Python
Django中间件工作流程及写法实例代码
Feb 06 Python
Numpy中矩阵matrix读取一列的方法及数组和矩阵的相互转换实例
Jul 02 Python
selenium+python实现1688网站验证码图片的截取功能
Aug 14 Python
Python 打印中文字符的三种方法
Aug 14 Python
Python初学者需要注意的事项小结(python2与python3)
Sep 26 Python
python实现简易数码时钟
Feb 19 Python
Python通过递归获取目录下指定文件代码实例
Nov 07 Python
开启Django博客的RSS功能的实现方法
Feb 17 Python
Django自定义列表 models字段显示方式
Apr 03 Python
Python中的几种矩阵乘法(小结)
Jul 10 #Python
用Python实现最速下降法求极值的方法
Jul 10 #Python
python networkx 根据图的权重画图实现
Jul 10 #Python
python networkx 包绘制复杂网络关系图的实现
Jul 10 #Python
python卸载后再次安装遇到的问题解决
Jul 10 #Python
Python求离散序列导数的示例
Jul 10 #Python
Python Matplotlib 基于networkx画关系网络图
Jul 10 #Python
You might like
木翼下载系统中说明的PHP安全配置方法
2007/06/16 PHP
php与paypal整合方法
2010/11/28 PHP
php数组函数序列之krsort()- 对数组的元素键名进行降序排序,保持索引关系
2011/11/02 PHP
使用Apache的htaccess防止图片被盗链的解决方法
2013/04/27 PHP
PHP网页游戏学习之Xnova(ogame)源码解读(十六)
2014/06/30 PHP
php无法连接mysql数据库的正确解决方法
2016/07/01 PHP
Linux下快速搭建php开发环境
2017/03/13 PHP
laradock环境docker-compose操作详解
2019/07/29 PHP
ExtJS TabPanel beforeremove beforeclose使用说明
2010/03/31 Javascript
细说浏览器特性检测(2)-通用事件检测
2010/11/05 Javascript
autoIMG 基于jquery的图片自适应插件代码
2011/03/12 Javascript
dojo学习第一天 Tab选项卡 实现
2011/08/28 Javascript
javascript中的window.location.search方法简介
2013/09/02 Javascript
jquery ajax对特殊字符进行转义防止js注入使用示例
2013/11/21 Javascript
JavaScript onkeydown事件入门实例(键盘某个按键被按下)
2014/10/17 Javascript
浅谈JavaScript的Polymer框架中的事件绑定
2015/07/29 Javascript
SelecT下拉框选中和取值的解决方法
2016/11/22 Javascript
纯js实现倒计时功能
2017/01/06 Javascript
JavaScript高阶函数_动力节点Java学院整理
2017/06/28 Javascript
Ant Design Pro 下实现文件下载的实现代码
2019/12/03 Javascript
Vue组件化开发之通用型弹出框的实现
2020/02/28 Javascript
uniapp 仿微信的右边下拉选择弹出框的实现代码
2020/07/12 Javascript
适用于 Vue 的播放器组件Vue-Video-Player操作
2020/11/16 Javascript
基于jQuery拖拽事件的封装
2020/11/29 jQuery
Python的一些用法分享
2012/10/07 Python
提升Python程序运行效率的6个方法
2015/03/31 Python
python操作mongodb根据_id查询数据的实现方法
2015/05/20 Python
简单了解Django模板的使用
2017/12/20 Python
django admin 后台实现三级联动的示例代码
2018/06/22 Python
解决pycharm 安装numpy失败的问题
2019/12/05 Python
瑞士灯具购物网站:Lampenwelt.ch
2018/07/08 全球购物
Under Armour安德玛德国官网:美国高端运动科技品牌
2019/03/09 全球购物
结婚喜宴家长答谢词
2014/01/15 职场文书
护理专业求职信
2014/06/15 职场文书
乡村教师党员四风问题对照检查材料思想汇报
2014/10/08 职场文书
幼儿园百日安全活动总结
2015/05/07 职场文书