Python标准库tempfile的使用总结

Posted 2021-06-17 Wan-deuk

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Python标准库tempfile的使用总结相关的知识，希望对你有一定的参考价值。

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实现了多种抽取式摘要算法，是自动文摘研究的利器。

参考材料

以上是关于Python标准库tempfile的使用总结的主要内容，如果未能解决你的问题，请参考以下文章

tempfile库：Python临时文件系统对象

python 模块初识

Python怎么如何使用临时文件？

2023最新整理！全网最全Python标准库总结（学习宝典，建议收藏）