如何在Tkinter GUI Python中添加PDF文件

本文将教我们如何在Tkinter GUI中显示PDF文件。我们将使用PyMuPDF库来读取PDF文件并将其转换为图像，然后使用Tkinter显示图像。

对于我们的任务，我们将执行以下步骤 –

• 读取PDF文件。

• 定义一个转换矩阵，将其应用于PDF的页面以获取它们的图像。

• 计算页面的总数以进行错误检查。

• 为GUI定义屏幕（画布）。

• 定义一个辅助函数，用于将PDF页面转换为PIL图像。

• 定义一个函数，在我们的GUI中显示页面的图像。

• 添加按钮，标签，滚动条等，并对它们进行配置。

要开始，请首先在终端中运行以下命令来安装PyMuPDF包。

python -m pip install --upgrade pip
python -m pip install --upgrade pymupdf

现在我们需要导入我们将使用的库。import fitz

import fitz
from tkinter import *
from PIL import Image, ImageTk

这里的fitz是PyMuPDF库附带的一个实用程序库，用于将PDF页面转换为图像。

我们首先使用fitz.open函数打开我们想要在GUI中显示的PDF文件。这将返回一个文档对象，用于访问页面。

# open pdf file
file_name = "sample.pdf"
doc = fitz.open(file_name)

现在，我们需要指定我们希望如何查看页面。通过定义以下矩阵函数，我们将保持1倍缩放。

# transformation matrix we can apply on pages
zoom = 1
mat = fitz.Matrix(zoom, zoom)

让我们也来计算页面的数量 –

# count number of pages
num_pages = 0
for p in doc:
num_pages += 1

现在我们将定义一个屏幕，将滚动条和画布添加到我们的代码中 –

# initialize and set screen size
root = Tk()
root.geometry('750x700')

# add scroll bar
scrollbar = Scrollbar(root)
scrollbar.pack(side = RIGHT, fill = Y)

# add canvas
canvas = Canvas(root, yscrollcommand = scrollbar.set)
canvas.pack(side = LEFT, fill = BOTH, expand = 1)

这里我们将屏幕表示为“root”，尺寸为750×700。我们还添加了一个滚动条，并指定它在右侧，并填充整个高度。同样，我们将画布添加到左侧，并填充整个屏幕的两侧！

为了获取用户想读的页码，我们需要将页码作为输入。我们可以按如下方式实现 –

# define entry point (field for taking inputs)
entry = Entry(root)

# add a label for the entry point
label = Label(root, text="Enter page number to display:")

参数’root’告诉我们这些对象将与屏幕的“根”关联。

现在我们将定义一个辅助函数，用于从页数获取PDF文件的图像。

def pdf_to_img(page_num):
page = doc.load_page(page_num)
pix = page.get_pixmap(matrix=mat)
return Image.frombytes("RGB", [pix.width, pix.height], pix.samples)

在这里，我们首先使用doc.load_page方法加载页面，并将页面编号作为参数传递。然后我们将PDF页面转换为像素表示，在第二行中进行操作。最后，我们将像素表示转换为PIL Image格式。

现在，我们定义一个名为show_image的函数，用于在我们的Tkinter GUI中显示图像。

def show_image():
try:
page_num = int(entry.get()) - 1
assert page_num >= 0 and page_num < num_pages
im = pdf_to_img(page_num)
img_tk = ImageTk.PhotoImage(im)
frame = Frame(canvas)
panel = Label(frame, image=img_tk)
panel.pack(side="bottom", fill="both", expand="yes")
frame.image = img_tk
canvas.create_window(0, 0, anchor='nw', window=frame)
frame.update_idletasks()
canvas.config(scrollregion=canvas.bbox("all"))
except:
pass

在这里，我们首先从之前定义的入口点读取页码，并将其转换为整数。如果它不是一个有效的整数，则会发生异常，并由异常块处理。然后我们检查页码是否在0到总页数之间，因为这是我们可以显示的唯一有效页码范围。然后使用之前定义的辅助函数，我们获得用户指定页面的PIL图像。然后我们设置帧、面板和画布以适当地显示这个图像。

接下来，我们添加一个按钮来显示页面。我们将show_image函数作为命令传递。因此，每当按下按钮时，将调用show_image函数。

# add button to display pages
button = Button(root, text="Show Page", command=show_image)

现在我们设置之前定义的各个组件的可视位置-

# set visual locations
label.pack(side=TOP, fill=None)
entry.pack(side=TOP, fill=BOTH)
button.pack(side=TOP, fill=None)

我们还可以将初始页面设置为PDF的第一页，如下所示−

entry.insert(0, '1')
show_image()

我们配置滚动条

scrollbar.config(command = canvas.yview)
root.mainloop()

最后我们关闭PDF，以免被我们的电脑破坏。

doc.close()

输出

运行程序后，我们看到以下输出！

结论

本文教会了我们如何使用PyMuPDF库以及Pillow和Tkinter来在Tkinter GUI中显示PDF文件。我们还学习了如何计算页面数量，将PDF页面转换为图像，制作交互按钮以及如何在GUI中可视化放置不同的元素。