我正在尝试编写一个简单的脚本来合并两个PDF,但在尝试将输出保存到磁盘时遇到了问题。我的代码是
from PyPDF2 import PdfFileWriter, PdfFileReader
import tkinter as tk
from tkinter import filedialog
### Prompt the user for the 2 files to use via GUI ###
root = tk.Tk()
root.update()
file_path1 = tk.filedialog.askopenfilename(
filetypes=[("PDF files", "*.pdf")],
)
file_path2 = tk.filedialog.askopenfilename(
filetypes=[("PDF files", "*.pdf")],
)
###Function to combine PDFs###
output = PdfFileWriter()
def append_pdf_2_output(file_handler):
for page in range(file_handler.numPages):
output.addPage(file_handler.getPage(page))
#Actually combine the 2 PDFs###
append_pdf_2_output(PdfFileReader(open(file_path1, "rb")))
append_pdf_2_output(PdfFileReader(open(file_path2, "rb")))
###Prompt the user for the file save###
output_name = tk.filedialog.asksaveasfile(
defaultextension='pdf')
###Write the output to disk###
output.write(output_name)
output.close
问题是我收到错误
UserWarning:要写入的文件不是二进制模式。它可能无法正确写入。 [pdf.py:453]回溯(最近一次调用最后一次):文件“Combine2Pdfs.py”,第44行,在output.write(output_name)文件“/Library/Frameworks/Python.framework/Versions/3.5/lib/pytho n3.5 / site-packages / P yPDF2 / pdf.py“,第487行,写入stream.write(self。 header + b (”\ n“))TypeError:write( )参数必须是str,而不是字节
我哪里出错?
答案 0 :(得分:1)
尝试使用tk.filedialog.asksaveasfilename
代替tk.filedialog.asksaveasfile
。你只需要文件名,而不是文件处理程序本身。
###Prompt the user for the file save###
output_name = tk.filedialog.asksaveasfilename(defaultextension='pdf')