python - 在 Python3 的 tempfile.TemporaryDirectory() 中找不到文件

Question

我一般在使用 Python3 的临时文件库时遇到问题。

我需要在一个临时目录中写入一个文件，并确保它在那里。我使用的第三方软件工具有时会失败，所以我不能直接打开文件，我需要先使用“while 循环”或其他方法验证它是否存在，然后再打开它。所以我需要搜索 tmp_dir(使用 os.listdir() 或等价物)。

具体帮助/解决方案和一般帮助将在评论中表示赞赏。

谢谢。

小示例代码:

import os
import tempfile


with tempfile.TemporaryDirectory() as tmp_dir:

    print('tmp dir name', tmp_dir)

    # write file to tmp dir
    fout = open(tmp_dir + 'file.txt', 'w')
    fout.write('test write')
    fout.close()

    print('file.txt location', tmp_dir + 'lala.fasta')

    # working with the file is fine
    fin = open(tmp_dir + 'file.txt', 'U')
    for line in fin:
        print(line)

    # but I cannot find the file in the tmp dir like I normally use os.listdir()
    for file in os.listdir(tmp_dir):
        print('searching in directory')
        print(file)

Answer 1

这是意料之中的，因为临时目录名称不以路径分隔符结尾(在许多系统上为 os.sep、斜杠 或反斜杠)。所以文件是在错误的级别创建的。

tmp_dir = D:\Users\foo\AppData\Local\Temp\tmpm_x5z4tx
tmp_dir + "file.txt"
=> D:\Users\foo\AppData\Local\Temp\tmpm_x5z4txfile.txt

相反，join在临时目录中获取文件的两条路径:

fout = open(os.path.join(tmp_dir,'file.txt'), 'w')

请注意 fin = open(tmp_dir + 'file.txt', 'U')找到文件，这是预期的，但它在 tmp_dir 所在的同一目录中找到它已创建。