PyQt/wordhelper.py

import sys  # 操作系统模块
from PyQt5.QtGui import  QColor  # 导入PyQt5的QtGui模块
from PyQt5.QtWidgets import QApplication, QMainWindow, QFileDialog, QTableWidgetItem, QTableWidget, QMessageBox # 导入PyQt5的QtWidgets模块
import os  # 导入操作系统模块
import tools.common as common  # 导入工具模块并设置别名为common
import tools.wordtopdf as wordtopdf
import tools.mergepdf as mergepdf
from mainWindow import *   # 导入主窗体的UI类
from pageWindow import *  # 导入Word转PDF窗体的UI类
from listWindow import *  # 导入统计Word文档页码窗体的UI类
from transformWindow import *  # 导入提取总页码窗体的UI类
import _thread
# 主窗体初始化类
class MyMainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self):
        super(MyMainWindow,self).__init__()
        self.setupUi(self)
        self.setGeometry(100, 100, 1024, 600)
        self.setWindowTitle('Word助手')  # 设置窗体的标题
        # 设置窗体背景
        palette = QtGui.QPalette()  # 创建调色板类的对象
        # 设置窗体背景自适应
        palette.setBrush(self.backgroundRole(),QBrush(QPixmap("./image/bg.png").scaled(self.size(),QtCore.Qt.IgnoreAspectRatio,QtCore.Qt.SmoothTransformation)))
        self.setPalette(palette)
        self.setAutoFillBackground(True) # 设置自动填充背景
        self.setFixedSize(1024,600)  # 禁止显示最大化按钮及调整窗体大小


'''Word转PDF模块'''
class TransformWindow(QMainWindow, Ui_TransformWindow):
    filelist = []
    def __init__(self):
        super(TransformWindow,self).__init__()
        self.setupUi(self)
        self.showLoding.setText("")  # 设置显示转换进度标签不显示内容
        self.showLoding.setMinimumWidth(100)  # 设置Label标签的最小宽度

        self.multipleExecute.clicked.connect( self.multipleExecuteClick)  #批量转换按钮绑定槽函数
        self.singleExecute.clicked.connect(self.singleExecuteClick)  # 合为一个PDF按钮绑定槽函数
        self.sourcebrowseButton.clicked.connect(self.sourcebrowseClick)   # 选择源文件夹按钮绑定槽函数
        self.targetbrowseButton.clicked.connect(self.targetbrowseClick)   # 选择目标文件夹按钮绑定槽函数
        self.listpdf.itemDoubleClicked.connect(self.itemdoubleClick)  # 为列表项的双击事件绑定槽函数
    # 自定义打开子窗体的方法
    def open(self):
        self.__init__()
        self.show()  # 显示子窗体
    def sourcebrowseClick(self):  # 单击浏览源文件夹按钮所触发的方法
        # 打开选择文件夹的对话框
        dir_path = QFileDialog.getExistingDirectory(self, "请选择源文件目录", r"E:\learn\test\doc")
        if dir_path == "":  # 处理没有选择路径的情况，这里为直接返回
            return
        self.sourcepath.setText(dir_path)  # 将获取到的文件夹路径添加到文本框控件中
        self.listword.clear()   # 清空列表
        global filelist  # 定义全局变量
        filelist = common.getfilenames(dir_path,[],'.doc')  # 获取Word文档路径
        self.listword.addItems(filelist)  # 将获取到的Word文件路径添加到列表控件中

    def targetbrowseClick(self): # 单击浏览目标文件夹按钮所触发的方法
        dir_path = QFileDialog.getExistingDirectory(self, "请选择目标文件目录", r"E:\learn\test\pdf")
        self.targetpath.setText(dir_path)

    def itemdoubleClick(self,item):  # 处理双击列表项触发的方法
        if os.path.exists(item.text()):
            os.startfile(item.text())  # 打开文件
        else:
            QMessageBox.information(self, "温馨提示：", "不是有效的文件名！", QMessageBox.Yes)

    def multipleExecuteClick(self):  #批量转换按钮触发的方法
        # 判断是否选择了源文件，如果没有选择则弹出提示框告知
        if self.listword.count() == 0:
            QMessageBox.information(self, "温馨提示：", "没有要转换的Word文档！", QMessageBox.Yes)
            return
        targetpath = self.targetpath.text()  # 获取目标文件夹
        # 判断是否选择了目标文件，如果没有选择则弹出提示框告知
        if not os.path.exists(targetpath):
            QMessageBox.information(self, "温馨提示：", "请选择正确的目标路径！", QMessageBox.Yes)
            return
        self.listpdf.clear()  # 清空结果列表
        self.showLoding.setMovie(self.gif)  # 设置gif图片
        self.gif.start()  # 启动图片，实现等待gif图片的显示
        _thread.start_new_thread(self.mExecute, ())  # 开启新线程执行批量转PDF

    # 实现批量Word转PDF操作的方法
    def mExecute(self):
        targetpath = self.targetpath.text()  # 获取目标文件夹
        valueList = wordtopdf.wordtopdf(filelist,targetpath)  # 实现将Word文档批量转换为PDF
        if(valueList != -1):
            self.showLoding.clear()  # 清除进度条
            self.listpdf.addItems(valueList)  # 将转换后的PDF路径显示在目标列表中

    # 合为一个PDF按钮所触发的方法
    def singleExecuteClick(self):
        # 判断是否选择了源文件，如果没有选择则弹出提示框告知
        if self.listword.count() == 0:
            QMessageBox.information(self, "温馨提示：", "没有要转换的Word文档！", QMessageBox.Yes)
            return
        # 判断是否选择了目标文件夹，如果没有选择则弹出提示框告知
        if not os.path.exists(self.targetpath.text()):
            QMessageBox.information(self, "温馨提示：", "请选择正确的目标路径！", QMessageBox.Yes)
            return
        self.listpdf.clear()  # 清空结果列表
        self.showLoding.setMovie(self.gif)  # 设置gif图片
        self.gif.start()  # 启动图片，实现等待gif图片的显示
        _thread.start_new_thread(self.sExecute,())  # 开启新线程执行多个Word合为一个PDF

    # 实现合为一个PDF文件操作的方法
    def sExecute(self):
        targetpath = self.targetpath.text()  # 获取目标路径
        valueList = wordtopdf.wordtopdf(filelist, targetpath)  # 将多个Word文档转换为PDF文件
        if(valueList != -1):
            mergepdf.mergefiles(targetpath, 'merged.pdf', True) # 将多个PDF文件合并为一个PDF文件
            self.showLoding.clear()  # 清除进度条
            temp = [os.path.join(targetpath , 'merged.pdf')] # 组合PDF文件路径
            self.listpdf.addItems( temp)  # 将PDF文件路径显示到结果列表中
            for file in valueList: # 遍历临时生成的PDF文件列表
                os.remove(file)  # 删除PDF文件

'''统计Word文档页码模块'''
class PageWindow(QMainWindow, Ui_PageWindow):

    filelist = []    # Word文件路径列表
    def __init__(self):
        super(PageWindow,self).__init__()
        self.setupUi(self)
        self.pagetable.setColumnWidth(0,600)  # 设置第一列的宽度
        self.pagetable.setColumnWidth(1,100)  # 设置第二列的宽度
        self.pagetable.setStyleSheet("background-color: lightblue"
                                     "(spread:pad,stop:0.823 rgba(255, 255, 255, 204), stop:1 rgba(255, 255, 255, 204));"
                                     "selection-background-color:lightblue;")
        headItem = self.pagetable.horizontalHeaderItem(0)  # 获得水平方向表头的Item对象
        headItem.setBackground(QColor(0, 60, 10))  # 设置单元格背景颜色
        headItem.setForeground(QColor(200, 111, 30))  # 设置文字颜色
        headItem = self.pagetable.horizontalHeaderItem(1)  # 获得水平方向表头的Item对象
        headItem.setBackground(QColor(0, 60, 10))  # 设置单元格背景颜色
        headItem.setForeground(QColor(200, 111, 30))  # 设置文字颜色
        self.pagetable.setEditTriggers(QTableWidget.NoEditTriggers)
        self.pagetable.setSelectionBehavior(QTableWidget.SelectRows)
        self.pagetable.setSelectionMode(QTableWidget.SingleSelection)
        self.pagetable.setAlternatingRowColors(True)
        self.totalpage.setMinimumWidth(100)  # 设置Label标签的最小宽度

        self.browseButton.clicked.connect(self.sourcebrowseClick)  # 选择源路径
        self.executeButton.clicked.connect(self.executeClick)  # 开始统计按钮的事件绑定

    # 自定义打开子窗体的方法
    def open(self):
        self.__init__()
        self.show()  # 显示子窗体


    def sourcebrowseClick(self):
        dir_path = QFileDialog.getExistingDirectory(self, "请选择源文件目录", r"E:\learn\test\doc")
        if dir_path != "":  # 判断已经选择了源文件目录
            self.sourcepath.setText(dir_path)
            self.listword.clear()   # 清空列表
            global filelist
            filelist = common.getfilenames(dir_path,[],'.doc')  # 获取Word文档
            self.listword.addItems(filelist)

    def executeClick(self):  #开始统计按钮的自定义事件
        if self.listword.count() == 0:
            QMessageBox.information(self, "温馨提示：", "没有要统计页码的Word文档！", QMessageBox.Yes)
            return
        self.totalpage.setText("")
        self.totalpage.setMovie(self.gif)  # 设置gif图片
        self.label_2.setText("正在统计：")
        self.gif.start()  # 启动图片，实现等待gif图片的显示
        _thread.start_new_thread(self.execute,())  # 开启新线程执行统计页码
    # 统计页码
    def execute(self):
        valueList = []
        valueList = wordtopdf.wordtopdf1(filelist)
        # if valueList != []:
        #     self.totalpage.clear()  # 转换完毕就将等待gif图片清理掉
        totalPages = str(valueList[0]) # 总页数
        self.label_2.setText("合计页码：")
        self.totalpage.setText(totalPages)   # 显示统计出来的页码
        print("行数：",len(valueList[1]))
        self.pagetable.setRowCount(len(valueList[1]))  # 指定行数
        resultList = valueList[1]  # 获取统计结果
        for i in range(self.pagetable.rowCount()):
            for j in range(self.pagetable.columnCount()):
                content = resultList[i][j]  # 获取一个单元格的内容
                newItem = QTableWidgetItem(content)  # 转换为一个单元格对象
                self.pagetable.setItem(i, j, newItem)  # 显示在单元格中

'''提取总目录模块'''
class ListWindow(QMainWindow, Ui_ListWindow):
    def __init__(self):
        super(ListWindow,self).__init__()
        self.setupUi(self)
        self.browseButton.clicked.connect(self.sourcebrowseClick)   # 选择源路径
        self.executeButton.clicked.connect( self.getListClick)  #按钮事件绑定
        self.openButton.clicked.connect(self.openButtonClick)  # 为打开文件按钮绑定事件
    # 自定义打开子窗体的方法
    def open(self):
        self.__init__()
        self.show()  # 显示子窗体
    def sourcebrowseClick(self):
        dir_path = QFileDialog.getExistingDirectory(self, "请选择源文件目录", r"E:\learn\test\doc")
        if dir_path != "":  # 判断已经选择了源文件目录
            self.sourcepath.setText(dir_path)
            self.listword.clear()   # 清空列表
            global filelist
            filelist = common.getfilenames(dir_path,[],'.doc')  # 获取Word文档
            self.listword.addItems(filelist)

    def getListClick(self):  #子窗体自定义事件
        if self.listword.count() == 0:
            QMessageBox.information(self, "温馨提示：", "没有要提取目录的Word文档！", QMessageBox.Yes)
            return
        self.listfile.setText("")
        self.listfile.setMovie(self.gif)  # 设置gif图片
        self.gif.start()  # 启动图片，实现等待gif图片的显示
        _thread.start_new_thread(self.getList,())  # 开启新线程执行统计页码

     # 提取目录
    def getList(self):
        sourcepath = self. sourcepath.text()  # 获取源路径
        if not os.path.exists(sourcepath):  # 判断是否选择了源目录
            QMessageBox.information(self,"温馨提示：","请先选择Word文档所在的文件夹！",QMessageBox.Yes)
            return
        targetpath = os.path.join(sourcepath, "pdf")  # 根据源路径生成目标目录
        if not os.path.exists(targetpath):  # 判断目录是否存在，不存在则创建
            os.makedirs(targetpath) # 创建目录
        valueList = wordtopdf.wordtopdf(filelist, targetpath)
        if(valueList != -1):
            mergepdf.mergefiles(targetpath, 'merged.pdf', True)  # 合并PDF
            temp = [os.path.join(targetpath , 'merged.pdf')] # 生成合并后的PDF文件的路径
            for file in valueList: # 遍历临时生成的PDF文件列表
                os.remove(file)  # 删除PDF文件
            isList = self.checkBox.isChecked()    # 指定是否带目录
            resultvalue=wordtopdf.getPdfOutlines(temp[0],targetpath,isList)   # 提取目录
            os.remove(temp[0]) # 删除合并后的PDF文件
            if valueList != []:
                self.listfile.clear()  # 转换完毕就将等待gif图片清理掉
            self.listfile.setText(resultvalue)  # 将生成的目录文件路径显示到页面中


    # 自定义打开子窗体的方法
    def open(self):
        self.__init__()
        self.show()  # 显示子窗体

    # 打开文件按钮触发的事件函数
    def openButtonClick(self):
        if self.listfile.text() == "还未提取...":
            QMessageBox.information(self,"温馨提示：","还没有提取目录，请先单击【开始提取】按钮！",QMessageBox.Yes)
        else:
            os.startfile(self.listfile.text())  # 打开文件

if __name__ == '__main__':
    app = QApplication(sys.argv)   # 创建GUI对象
    main = MyMainWindow()   # 创建主窗体ui类对象
    qmovie = QtGui.QMovie('image/loding.gif')

    transformWindow = TransformWindow()  # 创建Word转PDF窗体对象
    transformWindow.gif = qmovie  # 加载gif图片
    main.actionWord_PDF.triggered.connect(transformWindow.open)  #为Toolbar上的Word转PDF按钮指定连接槽函数

    pagewindow = PageWindow()  # 创建统计Word文档页码窗体对象
    pagewindow.gif = qmovie  # 加载gif图片
    main.action_Word.triggered.connect(pagewindow.open)  #为Toolbar上的统计Word文档页码按钮指定连接槽函数

    listwindow = ListWindow()  # 创建提取总目录窗体对象
    listwindow.gif = qmovie  # 加载gif图片
    main.action_list.triggered.connect(listwindow.open)  #为Toolbar上的提取总目录按钮指定连接槽函数

    main.show()  # 显示主窗体
    sys.exit(app.exec_())   # 除非退出程序关闭窗体，否则一直运行