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

Question

我不太确定最好的word方法，但我想做的是，阅读一个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')

Answer 2

你说你试图关闭流，但是出错了？您可以删除PdfFileReader对象，以确保没有人仍然可以访问流。然后关闭水流。在

from pyPdf import PdfFileWriter, PdfFileReader

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

...

del input # PdfFileReader won't mess with the stream anymore
inputStream.close()
del blank
blankStream.close()

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

Answer 3

如果PDF足够小（这取决于您的平台），您只需读入全部内容，关闭文件，修改数据，然后将整个内容写回同一个文件中。在