命名的临时文件未被删除

Question

我正在 lambda 函数中创建一个命名的临时文件，我认为我编写的代码可能存在问题。

我创建了一个命名文件，用于创建一个 excel 文件，然后我将命名文件引用传递给另一个函数以上传到存储桶，似乎进一步调用该函数最终导致磁盘已满。

我不完全确定为什么调用 close() 不会从磁盘上的任何位置删除文件。

这是我创建 excel 文件的函数：

def get_excel_file(df_report): 
    tmp_file = tempfile.NamedTemporaryFile(suffix=".xlsx")
    

    with pd.ExcelWriter(tmp_file.name) as writer:       
        df_report.to_excel(
            writer, sheet_name="Report", index=False, startrow=4
        )
        writer.save()

    return tmp_file

然后将返回的文件tmp_file传递给以下函数进行上传，其参数名为report：

def upload_renewal_report(bucket_name, report, file_name):    
    s3_client.upload_file(report.name, bucket_name, file_name)

    report.close()

代码是否有任何明显错误，这意味着文件不会被清除？

报告，我关闭的对象打印为：

WHAT IS REPORT: <tempfile._TemporaryFileWrapper object at 0x7f9aeb9d42e0>

WHAT TYPE IS REPORT: <class 'tempfile._TemporaryFileWrapper'>

有一条评论现在已被删除，这似乎暗示我没有在临时文件上调用 close。我确实打电话给close() 似乎是临时文件，但似乎在上传到 S3 后文件由于某种原因已经关闭 也许有一些东西，因为我直到现在还没有在日志中看到这个：

/opt/python/xlsxwriter/workbook.py:338: UserWarning: Calling close() on already closed file.

Answer 1

NamedTemporaryFile 可能有些奇怪。我有一个测试可以在 linux 上运行，但不能在你描述的确切行为的 windows 上运行。我修复了它-

with NamedTemporaryFile(mode="w", delete=False) as f:
    f.write(profile)
    f.seek(0)
    yield f.name
    f.close()
    os.unlink(f.name)

您也可能实际上不需要使用磁盘（如果文件对于内存来说不是太大的话）

data = io.BinaryIO()
with pd.ExcelWriter(data) as writer:       
    df_report.to_excel(
        writer, sheet_name="Report", index=False, startrow=4
    )

s3_client.put_object(Body=data)