基於PyQt5完成的PDF拆分功能

Posted on by

PyQt是Qt框架的Python語言實現，是最強大的GUI庫之一。PyQt提供瞭一個設計良好的窗口控件集合，具有更方便的操作性。學過VB的同學會知道，相比與VB的使用，在界面設計上元素更豐富。

本文就是基於PyQt5完成的PDF拆分功能，先看界面：

 一、界面PdfC.ui可參考上文如何實現，程序代碼如下：

import os
from PyQt5 import QtCore, QtGui, QtWidgets
import  base64
from return_png import img as retu  # 引入img變量，賦別名為pdf_cf
 
class Ui_PdfC(object):
    def setupUi(self, PdfC):
        tmp = open('return.png', 'wb')  # 創建臨時的文件
        tmp.write(base64.b64decode(retu))  # 把這個one圖片解碼出來，寫入文件中去。
        tmp.close()
        PdfC.setObjectName("PdfC")
        PdfC.setWindowModality(QtCore.Qt.NonModal)
        PdfC.resize(1400, 814)
        PdfC.setMinimumSize(QtCore.QSize(1400, 814))
        PdfC.setMaximumSize(QtCore.QSize(1400, 814))
        palette = QtGui.QPalette()
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Window, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Window, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Window, brush)
        PdfC.setPalette(palette)
        self.Reback = QtWidgets.QToolButton(PdfC)
        self.Reback.setGeometry(QtCore.QRect(10, 10, 51, 41))
        palette = QtGui.QPalette()
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.ToolTipBase, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.ToolTipBase, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.ToolTipBase, brush)
        self.Reback.setPalette(palette)
        icon = QtGui.QIcon()
        icon.addPixmap(QtGui.QPixmap("return.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)
        self.Reback.setIcon(icon)
        self.Reback.setIconSize(QtCore.QSize(100, 100))
        self.Reback.setAutoRaise(True)
        self.Reback.setObjectName("Reback")
        self.label = QtWidgets.QLabel(PdfC)
        self.label.setGeometry(QtCore.QRect(88, 8, 171, 41))
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        font.setPointSize(16)
        font.setBold(True)
        font.setWeight(75)
        self.label.setFont(font)
        self.label.setObjectName("label")
        self.frame = QtWidgets.QFrame(PdfC)
        self.frame.setGeometry(QtCore.QRect(20, 60, 1351, 161))
        palette = QtGui.QPalette()
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(245, 255, 250))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Window, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(245, 255, 250))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Window, brush)
        brush = QtGui.QBrush(QtGui.QColor(245, 255, 250))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(245, 255, 250))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Window, brush)
        self.frame.setPalette(palette)
        self.frame.setAutoFillBackground(True)
        self.frame.setFrameShape(QtWidgets.QFrame.Box)
        self.frame.setFrameShadow(QtWidgets.QFrame.Plain)
        self.frame.setLineWidth(1)
        self.frame.setMidLineWidth(0)
        self.frame.setObjectName("frame")
        self.scwd = QtWidgets.QPushButton(self.frame)
        self.scwd.setGeometry(QtCore.QRect(570, 30, 191, 61))
        palette = QtGui.QPalette()
        brush = QtGui.QBrush(QtGui.QColor(0, 217, 159))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Text, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.ButtonText, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255, 128))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.PlaceholderText, brush)
        brush = QtGui.QBrush(QtGui.QColor(0, 217, 159))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Text, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.ButtonText, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255, 128))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.PlaceholderText, brush)
        brush = QtGui.QBrush(QtGui.QColor(0, 217, 159))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(120, 120, 120))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Text, brush)
        brush = QtGui.QBrush(QtGui.QColor(120, 120, 120))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.ButtonText, brush)
        brush = QtGui.QBrush(QtGui.QColor(0, 0, 0, 128))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.PlaceholderText, brush)
        self.scwd.setPalette(palette)
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        font.setPointSize(16)
        font.setBold(True)
        font.setWeight(75)
        self.scwd.setFont(font)
        self.scwd.setAutoFillBackground(True)
        self.scwd.setAutoDefault(False)
        self.scwd.setDefault(False)
        self.scwd.setFlat(True)
        self.scwd.setObjectName("scwd")
        self.label_2 = QtWidgets.QLabel(self.frame)
        self.label_2.setGeometry(QtCore.QRect(620, 100, 131, 21))
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        self.label_2.setFont(font)
        self.label_2.setObjectName("label_2")
        self.frame_2 = QtWidgets.QFrame(PdfC)
        self.frame_2.setGeometry(QtCore.QRect(1050, 280, 321, 301))
        palette = QtGui.QPalette()
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(245, 255, 250))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Window, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(245, 255, 250))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Window, brush)
        brush = QtGui.QBrush(QtGui.QColor(245, 255, 250))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Base, brush)
        brush = QtGui.QBrush(QtGui.QColor(245, 255, 250))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Window, brush)
        self.frame_2.setPalette(palette)
        self.frame_2.setAutoFillBackground(True)
        self.frame_2.setFrameShape(QtWidgets.QFrame.NoFrame)
        self.frame_2.setFrameShadow(QtWidgets.QFrame.Plain)
        self.frame_2.setLineWidth(1)
        self.frame_2.setMidLineWidth(0)
        self.frame_2.setObjectName("frame_2")
        self.label_3 = QtWidgets.QLabel(self.frame_2)
        self.label_3.setGeometry(QtCore.QRect(20, 20, 191, 31))
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        font.setPointSize(14)
        font.setBold(False)
        font.setWeight(50)
        self.label_3.setFont(font)
        self.label_3.setObjectName("label_3")
        self.xzwd = QtWidgets.QPushButton(self.frame_2)
        self.xzwd.setGeometry(QtCore.QRect(30, 190, 271, 61))
        palette = QtGui.QPalette()
        brush = QtGui.QBrush(QtGui.QColor(0, 217, 159))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Text, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.ButtonText, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255, 128))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.PlaceholderText, brush)
        brush = QtGui.QBrush(QtGui.QColor(0, 217, 159))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Text, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.ButtonText, brush)
        brush = QtGui.QBrush(QtGui.QColor(255, 255, 255, 128))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.PlaceholderText, brush)
        brush = QtGui.QBrush(QtGui.QColor(0, 217, 159))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Button, brush)
        brush = QtGui.QBrush(QtGui.QColor(120, 120, 120))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.Text, brush)
        brush = QtGui.QBrush(QtGui.QColor(120, 120, 120))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.ButtonText, brush)
        brush = QtGui.QBrush(QtGui.QColor(0, 0, 0, 128))
        brush.setStyle(QtCore.Qt.SolidPattern)
        palette.setBrush(QtGui.QPalette.Disabled, QtGui.QPalette.PlaceholderText, brush)
        self.xzwd.setPalette(palette)
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        font.setPointSize(14)
        font.setBold(True)
        font.setWeight(75)
        self.xzwd.setFont(font)
        self.xzwd.setAutoFillBackground(True)
        self.xzwd.setAutoDefault(False)
        self.xzwd.setDefault(False)
        self.xzwd.setFlat(True)
        self.xzwd.setObjectName("xzwd")
        self.label_5 = QtWidgets.QLabel(self.frame_2)
        self.label_5.setGeometry(QtCore.QRect(30, 60, 131, 21))
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        self.label_5.setFont(font)
        self.label_5.setObjectName("label_5")
        self.textEdit = QtWidgets.QTextEdit(self.frame_2)
        self.textEdit.setGeometry(QtCore.QRect(50, 110, 61, 51))
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        font.setPointSize(12)
        font.setBold(True)
        font.setWeight(75)
        self.textEdit.setFont(font)
        self.textEdit.setObjectName("textEdit")
        self.label_6 = QtWidgets.QLabel(self.frame_2)
        self.label_6.setGeometry(QtCore.QRect(140, 130, 41, 21))
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        self.label_6.setFont(font)
        self.label_6.setObjectName("label_6")
        self.textEdit_2 = QtWidgets.QTextEdit(self.frame_2)
        self.textEdit_2.setGeometry(QtCore.QRect(200, 110, 61, 51))
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        font.setPointSize(12)
        font.setBold(True)
        font.setWeight(75)
        self.textEdit_2.setFont(font)
        self.textEdit_2.setObjectName("textEdit_2")
        self.label_7 = QtWidgets.QLabel(PdfC)
        self.label_7.setGeometry(QtCore.QRect(480, 250, 71, 21))
        font = QtGui.QFont()
        font.setFamily("微軟雅黑")
        self.label_7.setFont(font)
        self.label_7.setObjectName("label_7")
        self.label_4 = QtWidgets.QLabel(PdfC)
        self.label_4.setGeometry(QtCore.QRect(270, 290, 461, 511))
        self.label_4.setFrameShape(QtWidgets.QFrame.Box)
        self.label_4.setFrameShadow(QtWidgets.QFrame.Plain)
        self.label_4.setLineWidth(1)
        self.label_4.setText("")
        self.label_4.setObjectName("label_4")
 
        self.retranslateUi(PdfC)
        QtCore.QMetaObject.connectSlotsByName(PdfC)
 
        os.remove("return.png")
 
    def retranslateUi(self, PdfC):
        _translate = QtCore.QCoreApplication.translate
        PdfC.setWindowTitle(_translate("PdfC", "PDF拆分"))
        self.Reback.setText(_translate("PdfC", "..."))
        self.label.setText(_translate("PdfC", "PDF拆分"))
        self.scwd.setText(_translate("PdfC", "+ 上傳文檔"))
        self.label_2.setText(_translate("PdfC", "上傳文檔至此"))
        self.label_3.setText(_translate("PdfC", "操作"))
        self.xzwd.setText(_translate("PdfC", "開始拆分"))
        self.label_5.setText(_translate("PdfC", "輸入頁數(請勿輸0)"))
        self.label_6.setText(_translate("PdfC", "到"))
        self.label_7.setText(_translate("PdfC", "PDF預覽"))

二、main.py實現主要功能程序如下：

class pdcwindow(QtWidgets.QWidget,Ui_PdfC):
    def __init__(self):
        super(pdcwindow, self).__init__()
        self.setupUi(self)
    def re_back(self):
        self.close()
    def sctp(self):
         fileName, fileType = QtWidgets.QFileDialog.getOpenFileName(self, "選取文件", os.getcwd(),"自定義(*.pdf;*);;All Files(*)")
         self.f = str(fileName)
         self.p, self.s =self.f.split(".")
         try:
          if self.s == "pdf":
            # 打開文件
            doc = fitz.open(self.f)
            # 讀取一頁 0代表第1頁
            page_one = doc.loadPage(0)
            # 將第一頁轉換為Pixmap
            page_pixmap = page_one.getPixmap()
            # 將Pixmap轉換為QImage
            image_format = QImage.Format_RGBA8888 if page_pixmap.alpha else QImage.Format_RGB888
            page_image = QImage(page_pixmap.samples, page_pixmap.width, page_pixmap.height, page_pixmap.stride,image_format)
            #width = page_image.width()
            #height = page_image.height()
            # QImage 轉為QPixmap
            pix = QPixmap.fromImage(page_image)
            # 設置圖片大小自適應標簽
            self.label_4.setScaledContents(True)
            # 給標簽設置圖像
            self.label_4.setPixmap(pix)
          else:
            msg_box = QMessageBox(QMessageBox.Warning, "錯誤", "文件不是 pdf 文件，請重新選擇！")
            msg_box.exec_()
         except Exception as e:
          pass
    def xzwj(self):
        pdf_file = PdfFileReader(self.f)
        output = PdfFileWriter()
        str1 = self.textEdit.toPlainText()
        num1 = int(str1)
        str2 = self.textEdit_2.toPlainText()
        num2 = int(str2)
        if num1 == num2:
            num1 = num1 - 1
        for i in range(num1, num2):
            output.addPage(pdf_file.getPage(i))
        # 寫出pdf
        pdf_out = "{}_pdf拆分.{}".format(self.p,self.s)
        with open(pdf_out, 'ab') as out_pdf:
            output.write(out_pdf)
        root = tkinter.Tk()
        root.withdraw()
        tkinter.messagebox.showinfo("提示", "成功下載到源文件所在文件夾")

說明如下：

(1) def sctp(self): 其中的sctp為按鈕的objectName屬性名字；def xzwj(self):其中的xzwj為按鈕的objectName屬性名字；

(2) 可通過點擊按鈕實現本地文件上傳；

fileName, fileType = QtWidgets. QFileDialog.getOpenFileName(self, "選取文件", os.getcwd(), "自定義(*.jpg;*.png;*.jpeg;*);;All Files(*)")

(3) 增加瞭防錯功能，當上傳的文檔格式不是pdf時，將不執行程序，同時提示錯誤；

(4) 將pdf第一頁轉為圖片，並用label組件顯示；

(5) 在def xzwj(self):程序中可實現按頁數拆分pdf；

三、最後在主程序中加入以上子程序。

if __name__ =="__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    win = mywindow()
    pc = pdcwindow()
    win.show()
    win.pdf_cf.clicked.connect(win.pic_main)
    win.pdf_cf.clicked.connect(pc.show)
    pc.Reback.clicked.connect(pc.re_back)
    pc.Reback.clicked.connect(win.show)
    pc.scwd.clicked.connect(pc.sctp)
    pc.xzwd.clicked.connect(pc.xzwj)
    sys.exit(app.exec_())

四、總結

本文介紹的pdf拆分功能還有一些待完善地方，例如可增加預覽功能，實現每頁預覽，以及如何實現多條件拆分。當然我們可以在後續階段逐漸完善。

到此這篇關於基於PyQt5完成的PDF拆分功能的文章就介紹到這瞭,更多相關PyQt5 PDF拆分內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：