我正在使用Python3,我想模拟写入文件,但实际上没有创建文件。
例如,我的具体情况如下:
merger = PdfFileMerger()
for pdf in files_to_merge:
merger.append(pdf)
merger.write('result.pdf') # This creates a file. I want to avoid this
merger.close()
# pdf -> binary
with open('result.pdf', mode='rb') as file: # Conversely. I don't want to read the data from an actual file
file_content = file.read()
我认为StringIO
是这种情况的不错的选择,但是在这种情况下,我不知道如何使用它,这将写入StringIO对象。看起来像这样:
output = StringIO()
output.write('This goes into the buffer. ')
# Retrieve the value written
print output.getvalue()
output.close() # discard buffer memory
# Initialize a read buffer
input = StringIO('Inital value for read buffer')
# Read from the buffer
print input.read()
答案 0 :(得分:2)
由于PdfFileMerger.write
方法支持写入类似文件的对象,因此您可以简单地使PdfFileMerger
对象写入BytesIO
对象:
from io import BytesIO
merger = PdfFileMerger()
for pdf in files_to_merge:
merger.append(pdf)
output = BytesIO()
merger.write(output)
merger.close()
file_content = output.getvalue()