500 行代码实现 PDF 阅读器

2019-04-05 本文已影响10人 Python高效编程

500 行代码实现 PDF 阅读器

经过两周的迭代，现在我们的 PyReadon 已经具备一般的 PDF 阅读器的基本功能：

添加书籍
删除书籍
阅读书籍
保存记录
查看书籍信息

前几版相比较，这一版优化了书籍的数据结构，支持同时阅读多本书，支持鼠标左键翻页，并通过与 sqlite3 数据库的交互来保存书库以及书籍信息（地址、页数等）。

在这里插入图片描述

书籍

在这里插入图片描述

使用 Book 类来保存书籍信息，比如元数据、页数以及阅读与否的信息。通过 __eq__ 特殊方法，来判断两个 book 实例是否为同一对象。

class Book:
    def __init__(self, fname):
        # 文件名
        self.fname = fname
        # 是否被阅读
        self.flag = None
        self._info = None
        self._page = 0
        self.get_meta_data(self.fname)
        
    def __eq__(self, other):
        if hasattr(other, 'fname'):
            return self.fname == other.fname
        return False

同时阅读

在这里插入图片描述

通过内部维护一个 read_list 列表来实现同时阅读多本书。列表储存 book 对象，每个 book 对象都有一个 page 属性。这样，我们的程序就能记住每本书被翻到的页数了。

第二行代码，是对 read_list 进行初始化。book.flag 用来判断这本书上次关闭前是否处于阅读的状态。如果是，我们就把它放在阅读列表中。

self.read_list = [None]
self.read_list.extend(book for book in self.booklist if book.flag)

左键翻页

我们重写 MyArea 类的 mousePressEvent 方法。event.pos() 函数用来获取鼠标的坐标，x() 用来获取横坐标。

width 为 MyArea 区域的宽度，如果点击鼠标左键，且鼠标位置的横坐标小于 1/3 区域宽度，那么向前翻页；大于 2/3 区域宽度，那么向后翻页。

# 鼠标左键翻页
def mousePressEvent(self, event):
    pos = event.pos().x()
    width = self.size().width()
    if event.button() == Qt.LeftButton:
        if pos > width * 2 / 3:
            self.right()
        elif pos < width / 3:
            self.left()

sqlite3

sqlite3 是轻量型本地数据库，具有无服务器、零配置、速度快等特点。

PyReadon 启动时，会从数据库中读取图书信息。read_db 函数主要执行以下功能：

如果路径中不存在 PDF.db 数据库，那么就新建 PDF.db 数据库，并且创建一个 book_info 表格，该表格拥有三个属性 path, page, flag;

从 book_info 表格中读取数据，并创建 book 对象来接收这些数据，最后通过 yield 函数返回 book 对象。

book_db = 'PDF.db'
book_info = namedtuple('info',  'path page flag')


def read_db():
    # 将路径更改为该文件所处路径
    os.chdir(os.path.dirname(os.path.realpath(__file__)))
    if not os.path.exists(book_db):
        conn = sqlite3.connect(book_db)
        conn.execute("CREATE TABLE book_info(path, page, flag)")
        conn.close()
    conn = sqlite3.connect(book_db)
    for row in conn.execute('SELECT * FROM book_info'):
        info = book_info(*row)
        book = Book(info.path)
        book.page = info.page
        book.flag = info.flag
        yield book
        
    conn.close()

将数据存储到数据库中：

将书籍列表传给 save2db 函数，通过列表推导式创建 book 所在地址的列表。conn.executemany 函数将迭代生成器表达式，并获得书籍地址、阅读页数、是否在阅读列表中等信息，最后将这些信息存储在数据库中。

def save2db(booklist):
    conn = sqlite3.connect(book_db)
    conn.executemany("INSERT INTO book_info Values (?,?,?)",
                ((book.fname, book.page, book.flag) for book in booklist))
    conn.commit()
    conn.close()

在进行存储数据之前，我们首先要将 book_info 数据库中的内容清空。

def remove_db():
    conn = sqlite3.connect(book_db)
    conn.execute('DELETE FROM book_info')
    conn.commit()
    conn.close()

查看书籍信息

在这里插入图片描述

书籍支持查看右键菜单，我们使用 QMessageBox.about 函数来显示书籍信息。

elif action == item3:
    index = row_num * 8 + col_num
    # 之后改成 book
    book = self.booklist[index]
    info = book.info
    fmt = f'路径：{info.path}\n\n' \
        f'格式：{info.format}\n\n' \
        f'标题：{info.title}\n\n' \
        f'作者：{info.author}\n\n' \
        f'Creator：{info.creator}\n\n' \
        f'Producer：{info.producer}\n\n'
    
    QMessageBox.about(self, '文档信息', fmt)

弹窗

在这里插入图片描述

通过 Qt Designer 设计了一个弹窗，并与主程序绑定：

info 即为弹窗，点击工具栏中的信息栏时会弹出窗口。

info = Info()
reader.infobar.triggered.connect(info.show)

以上就是本文的全部内容了，觉得不错的朋友可以点赞、或者转发支持，源代码已经上传到到我的 Github 上。请关注微信公众号：Python高效编程，后台回复 201945 获取代码。

500 行代码实现 PDF 阅读器