PyQt5 从零开始制作 PDF 阅读器（二）

上一篇文章中，我们实现了 PDF 阅读器的初始界面。这一次，新增了阅读功能，可以实现基本的翻页以及缩放等操作。不过，暂时只可以同时阅读一本书。

下图为效果图：

在这里插入图片描述

下面我来介绍下，这次主要新增的功能：

选项卡

QTabWidget 可以允许我们在一个窗口显示多个页面。对于书库的这个选项卡,页面显示为 self.table ,即初始界面

self.table(QTableWidget) -> self.tabwidget(QTabWidge)。

# 初始化选项卡
self.tabwidget = QTabWidget()
# 添加书库选项卡
self.tabwidget.addTab(self.table, '书库')
self.setCentralWidget(self.tabwidget)
# 设置选项卡可以关闭
self.tabwidget.setTabsClosable(True)
# 点击选项卡叉号时，执行 removeTabab 操作
self.tabwidget.tabCloseRequested[int].connect(self.remove_tab)

新建选项卡：每次开始阅读时，新建一个选项卡，名称为文件名。

def read_book(self, fname):
    # self.close()
    # 内存有可能泄露
    self.doc = fitz.open(fname)
    # metadata = doc.metadata
    title = fname.split('/' or '\\')[-1].replace('.pdf', '')

    vbox = self.book_area(self.doc.loadPage(0))
    self.book_add_tab(title, vbox)

其中，我们要求主选项卡，即书库选项卡是不可以关闭的。

def remove_tab(self, index):
    if index:
        # 当前页数
        self.current_page = 0
        self.tabwidget.removeTab(index)
        # 正在阅读的书
        self.read_list.pop(index)

阅读界面的选项卡对应的页面区域为 QScrollArea ，QScrollArea 支持滚轮操作。也就是说，如果我们缩放 PDF 页面大小超过 QScrollArea 的大小，那么就会自动出现滚轮，以便我们浏览页面。其中，MyArea 类是对 QScrollArea 的重载，绑定了快捷键以支持翻页以及缩放等操作。

Pixmap -> label -> area(MyArea) -> vbox(QVBoxLayout) -> tab(QWidget) -> self.tabwidget(QTabWidge)。

def book_add_tab(self, title, vbox):
    tab = QWidget()
    tab.setLayout(vbox)
    # tab 为页面，title 为标签名称
    self.tabwidget.addTab(tab, title)
    
def book_area(self, page):
    label = self.page_pixmap(page)
    # area = QScrollArea()
    area = MyArea(self)
    area.init(self)
    area.setWidget(label)

    vbox = QVBoxLayout()
    vbox.addWidget(area)
    return vbox

下面我们来看看, MyArea 这个类该如何定义:

MyArea(QScrollArea)

这里,我们定义了 init 方法,用来接受 Reader 主类 的 self 参数, 即通过 self.widget 调用 Reader 类的实例方法。

在 init_action 函数中,我们新建了四个 QShortCut 实例，分别支持快捷键实现缩小、放大、下一页、上一页的操作。

class MyArea(QScrollArea):
    def init(self, widget):
        self.widget = widget
        self.init_action()

    def init_action(self):
        zoom_minus = QShortcut(QKeySequence("Ctrl+-"), self)
        zoom_minus.activated.connect(self.minus)
        zoom_plus = QShortcut(QKeySequence("Ctrl+="), self)
        zoom_plus.activated.connect(self.plus)

        switch_left = QShortcut(QKeySequence(Qt.Key_Left), self)
        switch_left.activated.connect(self.left)
        switch_right = QShortcut(QKeySequence(Qt.Key_Right), self)
        switch_right.activated.connect(self.right)


    def plus(self):
        self.widget.zoom_book(plus=True)

    def minus(self):
        self.widget.zoom_book(plus=False)

    def right(self):
        self.widget.switch_page(right=True)

    def left(self):
        self.widget.switch_page(right=False)

下面，我们来介绍缩放与翻页功能的具体实现：

缩放功能

self.size 用来存储页面大小，self.page 正是根据 self.size 来实现缩放功能。

def zoom_book(self, plus=True):
    a, b = self.size
    if plus:
        a += 0.4
        b += 0.4
        self.size = (a, b)
        self.set_page()
    elif not plus and a > 0:
        if a >= 1:
            a -= 0.4
            b -= 0.4
        self.size = (a, b)
        self.set_page()

Pixmap -> label -> area(MyArea) -> vbox(QVBoxLayout) -> tab(QWidget) -> self.tabwidget(QTabWidge)。

tab 获取 tab 对象，layout 获取 vbox 对象，widget 获取 area 对象，直接更改 area 上 label 控件。

def set_page(self):
    # 加载页面
    page = self.doc.loadPage(self.current_page)
    # 获取当前 Widget
    tab = self.tabwidget.currentWidget()
    # 获取当前的 Layout
    layout = tab.layout()
    # 获取 Layout 上的控件
    widget = layout.itemAt(0).widget()
    # 获取已经绘制好的 label 对象
    label = self.page_pixmap(page)
    # 将 widget 的内容更改为现在的 label 对象
    widget.setWidget(label)

接着我们介绍如何实现翻页功能

翻页功能

这次，我们实现的 PDF 阅读器只能同时阅读一本书，所以翻页功能只需由 self.current_page 控制就行。

self.doc.pageCount 为总页数，当前页数不能为负数或者大于总页数。更改完 self.current_page 之后，就可以执行 self.set_page 操作，直接更改 area 上的 label 控件。

def set_current_page(self, right):
    if right and self.current_page < self.doc.pageCount - 1:
        self.current_page += 1

    elif not right and self.current_page > 0:
        self.current_page -= 1
        
 

def switch_page(self, right=True):
    self.set_current_page(right)
    self.set_page()

关注微信公众号：Python高效编程，微信后台回复 2019330 获取源码。