import sys
import os
from PyQt5.QtWidgets import QApplication, QMainWindow, QPushButton, QVBoxLayout, QWidget, QFileDialog, QListWidget, QMessageBox, QLineEdit, QHBoxLayout, QInputDialog
from PyQt5.QtCore import Qt, QThread, pyqtSignal
from PyPDF2 import PdfReader, PdfWriter, PdfMerger
from PyQt5 import QtGui

class CustomListWidget(QListWidget):
    def __init__(self, parent=None):
        super().__init__(parent)

class Worker(QThread):
    finished = pyqtSignal(str)
    error = pyqtSignal(str)
    def __init__(self, pdf_files, range_str=None, save_path=None, operation=None):
        super().__init__()
        self.pdf_files = pdf_files
        self.range_str = range_str
        self.save_path = save_path
        self.operation = operation
    def run(self):
        try:
            if self.operation == 'merge':
                merger = PdfMerger()
                for pdf in self.pdf_files:
                    merger.append(pdf)
                merger.write(self.save_path)
                merger.close()
                self.finished.emit('PDF文件已成功合并。')
            elif self.operation == 'split':
                start_page, end_page = self.parse_range(self.range_str)
                reader = PdfReader(self.pdf_files[zxsq-anti-bbcode-0])
                os.makedirs(self.save_path, exist_ok=True)
                for page in range(start_page, end_page + 1):
                    writer = PdfWriter()
                    writer.add_page(reader.pages)
                    split_save_path = os.path.join(self.save_path, f'Page_{page + 1}.pdf')
                    writer.write(split_save_path)
                self.finished.emit('PDF文件已成功拆分并保存。')
            elif self.operation == 'delete':
                start_page, end_page = self.parse_range(self.range_str)
                reader = PdfReader(self.pdf_files[zxsq-anti-bbcode-0])
                writer = PdfWriter()
                for page_num in range(len(reader.pages)):
                    if not (start_page <= page_num <= end_page):
                        writer.add_page(reader.pages[zxsq-anti-bbcode-page_num])
                writer.write(self.save_path)
                self.finished.emit('指定頁(yè)面已從PDF中刪除。')
        except Exception as e:
            self.error.emit(str(e))
    def parse_range(self, range_str):
        if '-' in range_str:
            start_page, end_page = map(int, range_str.split('-'))
        else:
            start_page = end_page = int(range_str)
        return start_page - 1, end_page - 1  # Convert to 0-based index

class PDFMergerApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.initUI()
        self.pdf_files = []
    def initUI(self):
        self.setWindowTitle('PDF 工具箱')
        self.setWindowIcon(QtGui.QIcon('111.ico'))
        self.setGeometry(100, 100, 800, 600)
        mainLayout = QVBoxLayout()
        self.addButton = QPushButton('添加 PDF', self)
        self.addButton.clicked.connect(self.addPDF)
        mainLayout.addWidget(self.addButton)
        self.listWidget = CustomListWidget(self)
        mainLayout.addWidget(self.listWidget)
        # 刪除按鈕的水平布局
        deleteLayout = QHBoxLayout()
        self.removeButton = QPushButton('刪除選定', self)
        self.removeButton.clicked.connect(self.removeSelected)
        deleteLayout.addWidget(self.removeButton)
        self.removeAllButton = QPushButton('刪除全部', self)
        self.removeAllButton.clicked.connect(self.removeAll)
        deleteLayout.addWidget(self.removeAllButton)
        mainLayout.addLayout(deleteLayout)
        self.mergeButton = QPushButton('合并 PDFs', self)
        self.mergeButton.clicked.connect(self.mergePDFs)
        mainLayout.addWidget(self.mergeButton)
        # 拆分和刪除頁(yè)碼的水平布局
        splitDeleteLayout = QHBoxLayout()
        self.splitInput = QLineEdit(self)
        self.splitInput.setPlaceholderText('輸入拆分范圍，如 1 或 1-4')
        splitDeleteLayout.addWidget(self.splitInput)
        self.splitButton = QPushButton('拆分 PDF', self)
        self.splitButton.clicked.connect(self.splitPDF)
        splitDeleteLayout.addWidget(self.splitButton)
        self.deleteInput = QLineEdit(self)
        self.deleteInput.setPlaceholderText('輸入刪除頁(yè)碼，如 1 或 1-4')
        splitDeleteLayout.addWidget(self.deleteInput)
        self.deleteButton = QPushButton('刪除頁(yè)面', self)
        self.deleteButton.clicked.connect(self.deletePages)
        splitDeleteLayout.addWidget(self.deleteButton)
        mainLayout.addLayout(splitDeleteLayout)
        container = QWidget()
        container.setLayout(mainLayout)
        self.setCentralWidget(container)
    def addPDF(self):
        files, _ = QFileDialog.getOpenFileNames(self, '打開文件', '', 'PDF files (*.pdf)')
        for file_path in files:
            self.addPDFFile(file_path)
    def addPDFFile(self, file_path):
        if file_path and file_path not in self.pdf_files:
            self.pdf_files.append(file_path)
            self.listWidget.addItem(file_path)
    def removeSelected(self):
        for item in self.listWidget.selectedItems():
            self.pdf_files.remove(item.text())
            self.listWidget.takeItem(self.listWidget.row(item))
    def removeAll(self):
        self.pdf_files.clear()
        self.listWidget.clear()
    def mergePDFs(self):
        save_path, _ = QFileDialog.getSaveFileName(self, '保存文件', '', 'PDF files (*.pdf)')
        if save_path:
            self.thread = Worker(self.pdf_files, save_path=save_path, operation='merge')
            self.thread.finished.connect(self.onFinished)
            self.thread.error.connect(self.onError)
            self.thread.start()
    def splitPDF(self):
        if len(self.pdf_files) != 1:
            QMessageBox.warning(self, "錯(cuò)誤", "請(qǐng)只選擇一個(gè)PDF文件進(jìn)行拆分。")
            return
        range_str = self.splitInput.text().strip()
        folder_path = self.getFolderName()
        if range_str and folder_path:
            self.thread = Worker(self.pdf_files, range_str=range_str, save_path=folder_path, operation='split')
            self.thread.finished.connect(self.onFinished)
            self.thread.error.connect(self.onError)
            self.thread.start()
    def getFolderName(self):
        folder_path = QFileDialog.getExistingDirectory(self, "選擇保存拆分文件的位置")
        if folder_path:
            folder_name, ok = QInputDialog.getText(self, "文件夾名稱", "輸入文件夾名稱:")
            if ok and folder_name:
                full_path = os.path.join(folder_path, folder_name)
                os.makedirs(full_path, exist_ok=True)
                return full_path
        return None
    def deletePages(self):
        if len(self.pdf_files) != 1:
            QMessageBox.warning(self, "錯(cuò)誤", "請(qǐng)只選擇一個(gè)PDF文件進(jìn)行刪除操作。")
            return
        range_str = self.deleteInput.text().strip()
        save_path = QFileDialog.getSaveFileName(self, '保存文件', '', 'PDF files (*.pdf)')[zxsq-anti-bbcode-0]
        if save_path and range_str:
            self.thread = Worker(self.pdf_files, range_str=range_str, save_path=save_path, operation='delete')
            self.thread.finished.connect(self.onFinished)
            self.thread.error.connect(self.onError)
            self.thread.start()
    def onFinished(self, message):
        self.show_message("操作完成", message)
        self.clear_pdf_list()
    def onError(self, error_message):
        self.show_message("操作失敗", error_message)
    def show_message(self, title, message):
        QMessageBox.information(self, title, message)
    def clear_pdf_list(self):
        self.pdf_files.clear()
        self.listWidget.clear()
def main():
    app = QApplication(sys.argv)
    ex = PDFMergerApp()
    ex.show()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()