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 相关文章推荐
wxpython 学习笔记 第一天
Feb 09 Python
本地文件上传到七牛云服务器示例(七牛云存储)
Jan 11 Python
浅谈Python中用datetime包进行对时间的一些操作
Jun 23 Python
python学习之matplotlib绘制散点图实例
Dec 09 Python
python3模块smtplib实现发送邮件功能
May 22 Python
Python实现将Excel转换成xml的方法示例
Aug 25 Python
python使用paramiko模块通过ssh2协议对交换机进行配置的方法
Jul 25 Python
Python3.8对可迭代解包的改进及用法详解
Oct 15 Python
Python连接SQLite数据库并进行增册改查操作方法详解
Feb 18 Python
Python如何使用神经网络进行简单文本分类
Feb 25 Python
使用Python webdriver图书馆抢座自动预约的正确方法
Mar 04 Python
Python OpenCV实现图形检测示例详解
Apr 08 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无限分类(树形类)
2013/09/28 PHP
php插入排序法实现数组排序实例
2015/02/16 PHP
html 锁定页面(js遮罩层弹出div效果)
2009/10/27 Javascript
javascript判断机器是否联网的2种方法
2013/08/09 Javascript
Javascript控制input输入时间格式的方法
2015/01/28 Javascript
jQuery中大家不太了解的几个方法
2015/03/04 Javascript
关于JS中setTimeout()无法调用带参函数问题的解决方法
2016/06/21 Javascript
原生js实现焦点轮播图效果
2017/01/12 Javascript
利用vue.js插入dom节点的方法
2017/03/15 Javascript
JavaScript运动框架 多物体任意值运动(三)
2017/05/17 Javascript
Textarea输入字数限制实例(兼容iOS&amp;安卓)
2017/07/06 Javascript
nodejs对express中next函数的一些理解
2017/09/08 NodeJs
Vue-cli@3.0 插件系统简析
2018/09/05 Javascript
localstorage实现带过期时间的缓存功能
2019/06/28 Javascript
[02:50]【扭转乾坤,只此一招】DOTA2永雾林渊版本开启新篇章
2020/12/22 DOTA
Python生成随机密码
2015/03/10 Python
利用Python实现网络测试的脚本分享
2017/05/26 Python
Python语言描述机器学习之Logistic回归算法
2017/12/21 Python
Python Web程序部署到Ubuntu服务器上的方法
2018/02/22 Python
Python用csv写入文件_消除空余行的方法
2018/07/06 Python
PyQt Qt Designer工具的布局管理详解
2019/08/07 Python
在Django下测试与调试REST API的方法详解
2019/08/29 Python
Python安装tar.gz格式文件方法详解
2020/01/19 Python
详细分析Python可变对象和不可变对象
2020/07/09 Python
中国领先的汽车保养服务平台:途虎养车
2019/10/18 全球购物
正宗的日本零食和糖果订阅盒:Bokksu
2019/11/21 全球购物
标记环网Toke Ring IEEE802.5
2014/05/26 面试题
abstract class和interface有什么区别
2013/08/04 面试题
初中生三年学习生活的自我评价
2013/11/03 职场文书
小区门卫岗位职责范本
2014/08/24 职场文书
公安派出所所长四风问题个人对照检查材料
2014/10/04 职场文书
司法局群众路线教育实践活动整改措施思想汇报
2014/10/13 职场文书
2015年银行员工工作总结
2015/04/24 职场文书
小学中队活动总结
2015/05/11 职场文书
立案决定书范文
2015/06/24 职场文书
MySQL数字类型自增的坑
2021/05/07 MySQL