编程语言
首页 > 编程语言> > Python标准库tempfile的使用总结

Python标准库tempfile的使用总结

作者:互联网

Python标准库tempfile的使用总结

临时文件是计算机程序存储临时数据的文件,它的扩展名通常是“.temp”。本文用于记录使用Python提供的临时文件API解决实际问题的过程,主要内容包括问题描述、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的模块,这个模块实现了创建临时文件和文件夹的功能,相关接口包括:

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

其中14实现了上下文管理器协议(实现了`__enter__`和`__exit__`两个特殊方法),配合`with`关键字使用,可以方便地自动关闭文件;56是低阶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,临时文件已经被销毁

解决方案

回到我们的问题:

  1. 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')
    
  2. 内存中的文章怎样作为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异常。

  3. 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