Python标准库tempfile的使用总结

2021-06-17 02:01:38 阅读：210 来源： 互联网

标签：总结 name Python tempfile PKUSUMSUM 临时文件 fin

Python标准库tempfile的使用总结

临时文件是计算机程序存储临时数据的文件，它的扩展名通常是“.temp”。本文用于记录使用Python提供的临时文件API解决实际问题的过程，主要内容包括问题描述、tempfile库简要介绍、解决方案等。

Python标准库tempfile的使用总结

问题描述

一些开源工具提供的接口有时不能满足特定场景下的需求，需要根据实际情况进行调整。比如基于Java的自动文摘工具PKUSUMSUM，它接收一篇文章作为输入，输出文章的摘要，要求输入输出都是磁盘上的文件:

java -jar PKUSUMSUM.jar -T 1 -input [输入文件] -output [输出文件] -L 2 -n 15 -m 5 -stop n

由于我的代码是用Python写的，需要进行摘要抽取的文章已经加载到内存了，并且通过工具生成摘要以后需要马上进行评测，自动文摘工具PKUSUMSUM提供的接口没办法直接满足我的需求。Python脚本怎样调用Jar包？内存中的文章怎样作为PKUSUMSUM的输入？PKUSUMSUM输出的摘要怎样读入到内存中？

tempfile库

通过查阅资料，我了解到Python标准库中有名字叫tempfile的模块，这个模块实现了创建临时文件和文件夹的功能，相关接口包括：

TemporaryFile：创建临时文件，一旦文件关闭，就立即被销毁；
NamedTemporaryFile：创建有名字的、在文件系统中可见的临时文件，可以通过指定delete参数设置文件关闭时的行为；
TemporaryDirectory：创建临时文件夹；
mkstemp()：创建临时文件，需要手动关闭和销毁；
mkdtemp()：创建临时文件夹，需要手动关闭和销毁。

其中1_{4实现了上下文管理器协议（实现了`__enter__`和`__exit__`两个特殊方法），配合`with`关键字使用，可以方便地自动关闭文件；5}6是低阶API，使用完临时文件或文件夹以后，需要手动关闭。

import tempfile  # # 导入模块
from pathlib import Path

fin = tempfile.TemporaryFile(mode='w', encoding='utf8')  # 创建临时文件
fin.name  # C:\\Users\\ADMINI~1\\AppData\\Local\\Temp\\tmpi1ozv3mj
Path(fin.name).exists()  # True，临时文件是存在的
fin.delete  # True，delete参数的默认值是True，表示文件关闭以后，立即销毁

with fin.file as fp:  # 上下文管理器自动关闭文件
	content = bytes('The quick brown fox jumps over the lazy dog', 'utf8')
	fp.write(content)

Path(fin.name).exists()  # False，临时文件已经被销毁

解决方案

回到我们的问题：

Python脚本怎样调用Jar包？

我们可以使用os模块的system方法执行Shell命令，调用Jar包：

os.system(f'java -jar PKUSUMSUM.jar -T 1 -input {fin.name} -output {fout.name} -L 2 -n 15 -m {algo_id} -stop n')

内存中的文章怎样作为PKUSUMSUM的输入？

我们可以首先把文章写入到临时文件中，然后通过引用临时文件的name属性，就可以把文章当成本地文件那样使用了：
```
# 把文章写入到临时文件中
fin = tempfile.NamedTemporaryFile(delete=False)
fin.close()
with open(Path(fin.name), mode='w', encoding='utf8') as fp:
    fp.write(text)
```
注意临时文件必须要先关闭，然后才能往文件里面中写内容，否则会触发Permission denied异常。
PKUSUMSUM输出的摘要怎样读入到内存中？

思路与第二个问题一样，创建临时文件并通过name属性进行引用，临时文件销毁前读出里面的内容（就是PKUSUMSUM抽取出的摘要）。

完整的代码如下：

def do_extractive_summarization(text: str, algo: str = 'TextRank') -> str:
    """基于PKUSUMSUM实现抽取式摘要算法"""
    algo_warehouse = {'Centroid': 2, 'ILP': 3, 'LexPageRank': 4, 'TextRank': 5, 'Submodular': 6}
    if algo not in algo_warehouse:
        raise ValueError(f'Invalid extractive summarization algorithm: {algo}')
    algo_id = algo_warehouse[algo]

    # 把文章写入到临时文件中
    fin = tempfile.NamedTemporaryFile(delete=False)
    fin.close()
    with open(Path(fin.name), mode='w', encoding='utf8') as fp:
        fp.write(text)

    # 运行PKUSUMSUM.jar，把生成的摘要写入到临时文件中
    fout = tempfile.NamedTemporaryFile(delete=False)
    fout.close()
    os.system(f'java -jar PKUSUMSUM.jar -T 1 -input {fin.name} -output {fout.name} -L 2 -n 15 -m {algo_id} -stop n')

    with open(Path(fout.name), mode='r', encoding='utf8') as fp:
        content = fp.read()
    summary = ' '.join(word_tokenize(content))
    Path(fin.name).unlink()
    Path(fout.name).unlink()

    return summary

总结

Python的tempfile模块提供了方便的创建临时文件和文件夹的接口，这些接口配合在Python脚本中执行Shell命令的方法一起使用，可以在开源工具和自己的代码之间创建一个“适配层”（我在写些什么？我怎么自己都看不懂？）；PKUSUMSUM实现了多种抽取式摘要算法，是自动文摘研究的利器。

参考材料

标签：总结,name,Python,tempfile,PKUSUMSUM,临时文件,fin
来源： https://www.cnblogs.com/wan-deuk/p/14891772.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

Python标准库tempfile的使用总结

Python标准库tempfile的使用总结

问题描述

tempfile库

解决方案

总结

参考材料