python - 如何覆盖 Python 当前正在读取的文件

Question

我不太确定最好的表达方式，但我想做的是读取 pdf 文件，进行各种修改，然后将修改后的 pdf 保存在原始文件上。截至目前，我可以将修改后的 pdf 保存到单独的文件中，但我希望替换原始文件，而不是创建新文件。

这是我当前的代码:

from pyPdf import PdfFileWriter, PdfFileReader

output = PdfFileWriter()
input = PdfFileReader(file('input.pdf', 'rb'))
blank = PdfFileReader(file('C:\\BLANK.pdf', 'rb'))

# Copy the input pdf to the output.
for page in range(int(input.getNumPages())):
    output.addPage(input.getPage(page))

# Add a blank page if needed.
if (input.getNumPages() % 2 != 0):
    output.addPage(blank.getPage(0))

# Write the output to pdf.
outputStream = file('input.pdf', 'wb')
output.write(outputStream)
outputStream.close()

如果我将 outputStream 更改为不同的文件名，它工作正常，我只是无法保存输入文件，因为它仍在使用中。我已尝试 .close() 流，但它也给我错误。

我感觉这有一个相当简单的解决方案，只是我没有运气找到它。

谢谢!

Answer 1

您始终可以将临时输出文件重命名为旧文件:

import os
f = open('input.pdf', 'rb')
# do stuff to temp.pdf
f.close()
os.rename('temp.pdf', 'input.pdf')