# main.py
import sys
import os
from pathlib import Path
from PyQt5.QtWidgets import QApplication
from pdf_converter_gui import PDFConverterApp

def main():
    # 創(chuàng)建QApplication實(shí)例
    app = QApplication(sys.argv)
    app.setApplicationName("PDF轉(zhuǎn)圖片工具")
    
    # 設(shè)置應(yīng)用程序樣式（可選）
    app.setStyle('Fusion')
    
    # 創(chuàng)建并顯示主窗口
    window = PDFConverterApp()
    window.show()
    
    # 進(jìn)入主事件循環(huán)
    sys.exit(app.exec_())

if __name__ == "__main__":
    main()

# converter_thread.py
from PyQt5.QtCore import QThread, pyqtSignal
import fitz  # PyMuPDF
import os
class PDFConverterThread(QThread):
    """用于在后臺執(zhí)行PDF轉(zhuǎn)換的線程"""
    
    # 定義信號：進(jìn)度更新、轉(zhuǎn)換完成、錯誤發(fā)生
    progress_updated = pyqtSignal(int)
    conversion_finished = pyqtSignal(str)
    error_occurred = pyqtSignal(str)

    def __init__(self, pdf_path, output_dir, image_format, dpi, convert_all_pages):
        super().__init__()
        self.pdf_path = pdf_path
        self.output_dir = output_dir
        self.image_format = image_format
        self.dpi = dpi
        self.convert_all_pages = convert_all_pages

    def run(self):
        """線程執(zhí)行的主方法"""
        try:
            # 打開PDF文件
            pdf_document = fitz.open(self.pdf_path)
            total_pages = pdf_document.page_count
            
            # 如果只轉(zhuǎn)換第一頁，則只處理一頁
            if not self.convert_all_pages:
                total_pages = 1
            
            # 逐頁轉(zhuǎn)換
            for page_num in range(total_pages):
                page = pdf_document[page_num]
                
                # 設(shè)置轉(zhuǎn)換矩陣（DPI）
                # 矩陣變換公式：$scale = dpi/72$，因?yàn)镻DF默認(rèn)72DPI
                mat = fitz.Matrix(self.dpi/72, self.dpi/72)
                pix = page.get_pixmap(matrix=mat)
                
                # 構(gòu)建輸出文件名
                if self.convert_all_pages:
                    output_filename = f"page_{page_num+1}.{self.image_format.lower()}"
                else:
                    output_filename = f"converted.{self.image_format.lower()}"
                
                output_path = os.path.join(self.output_dir, output_filename)
                
                # 保存圖片
                if self.image_format.upper() == "JPEG":
                    pix.save(output_path, "JPEG")
                else:
                    pix.save(output_path)
                
                # 更新進(jìn)度
                progress = int((page_num + 1) / total_pages * 100)
                self.progress_updated.emit(progress)
            
            pdf_document.close()
            self.conversion_finished.emit(f"轉(zhuǎn)換完成！共轉(zhuǎn)換了 {total_pages} 頁。")
            
        except Exception as e:
            self.error_occurred.emit(f"轉(zhuǎn)換過程中出現(xiàn)錯誤: {str(e)}")

# pdf_converter_gui.py
from PyQt5.QtWidgets import (QMainWindow, QVBoxLayout, QHBoxLayout, 
                             QPushButton, QLabel, QLineEdit, QFileDialog, QMessageBox,
                             QComboBox, QProgressBar, QSpinBox, QWidget, QGroupBox,
                             QCheckBox)
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QIcon
from converter_thread import PDFConverterThread
import os
from pathlib import Path
class PDFConverterApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.init_ui()
        
    def init_ui(self):
        """初始化用戶界面"""
        self.setWindowTitle("PDF轉(zhuǎn)圖片工具")
        self.setGeometry(100, 100, 600, 400)
        
        # 創(chuàng)建中央部件和布局
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        layout = QVBoxLayout(central_widget)
        
        # PDF文件選擇部分
        pdf_group = self.create_pdf_group()
        layout.addWidget(pdf_group)
        
        # 輸出設(shè)置部分
        output_group = self.create_output_group()
        layout.addWidget(output_group)
        
        # 進(jìn)度條
        self.progress_bar = QProgressBar()
        self.progress_bar.setVisible(False)
        layout.addWidget(self.progress_bar)
        
        # 狀態(tài)標(biāo)簽
        self.status_label = QLabel("準(zhǔn)備就緒")
        layout.addWidget(self.status_label)
        
        # 轉(zhuǎn)換按鈕
        self.convert_btn = QPushButton("開始轉(zhuǎn)換")
        self.convert_btn.clicked.connect(self.start_conversion)
        layout.addWidget(self.convert_btn)
        
    def create_pdf_group(self):
        """創(chuàng)建PDF文件選擇組件"""
        pdf_group = QGroupBox("PDF文件")
        pdf_layout = QVBoxLayout(pdf_group)
        
        pdf_file_layout = QHBoxLayout()
        self.pdf_path_edit = QLineEdit()
        self.pdf_path_edit.setPlaceholderText("選擇PDF文件...")
        pdf_file_layout.addWidget(self.pdf_path_edit)
        
        self.browse_pdf_btn = QPushButton("瀏覽...")
        self.browse_pdf_btn.clicked.connect(self.browse_pdf)
        pdf_file_layout.addWidget(self.browse_pdf_btn)
        
        pdf_layout.addLayout(pdf_file_layout)
        return pdf_group
    
    def create_output_group(self):
        """創(chuàng)建輸出設(shè)置組件"""
        output_group = QGroupBox("輸出設(shè)置")
        output_layout = QVBoxLayout(output_group)
        
        # 輸出目錄選擇
        output_dir_layout = QHBoxLayout()
        self.output_dir_edit = QLineEdit()
        self.output_dir_edit.setText(str(Path.home() / "Pictures"))
        output_dir_layout.addWidget(self.output_dir_edit)
        
        self.browse_output_btn = QPushButton("瀏覽...")
        self.browse_output_btn.clicked.connect(self.browse_output_dir)
        output_dir_layout.addWidget(self.browse_output_btn)
        
        output_layout.addLayout(output_dir_layout)
        
        # 轉(zhuǎn)換選項(xiàng)
        options_layout = QHBoxLayout()
        
        # 圖片格式選擇
        format_layout = QVBoxLayout()
        format_layout.addWidget(QLabel("圖片格式:"))
        self.format_combo = QComboBox()
        self.format_combo.addItems(["PNG", "JPEG"])
        format_layout.addWidget(self.format_combo)
        options_layout.addLayout(format_layout)
        
        # DPI設(shè)置
        dpi_layout = QVBoxLayout()
        dpi_layout.addWidget(QLabel("DPI (分辨率):"))
        self.dpi_spin = QSpinBox()
        self.dpi_spin.setRange(72, 600)
        self.dpi_spin.setValue(150)
        dpi_layout.addWidget(self.dpi_spin)
        options_layout.addLayout(dpi_layout)
        
        # 頁面選項(xiàng)
        pages_layout = QVBoxLayout()
        self.all_pages_check = QCheckBox("轉(zhuǎn)換所有頁面")
        self.all_pages_check.setChecked(True)
        pages_layout.addWidget(self.all_pages_check)
        options_layout.addLayout(pages_layout)
        
        output_layout.addLayout(options_layout)
        return output_group
    
    def browse_pdf(self):
        """瀏覽并選擇PDF文件"""
        file_path, _ = QFileDialog.getOpenFileName(
            self, "選擇PDF文件", "", "PDF文件 (*.pdf)")
        if file_path:
            self.pdf_path_edit.setText(file_path)
    
    def browse_output_dir(self):
        """瀏覽并選擇輸出目錄"""
        dir_path = QFileDialog.getExistingDirectory(
            self, "選擇輸出目錄", self.output_dir_edit.text())
        if dir_path:
            self.output_dir_edit.setText(dir_path)
    
    def start_conversion(self):
        """開始轉(zhuǎn)換過程"""
        pdf_path = self.pdf_path_edit.text()
        output_dir = self.output_dir_edit.text()
        
        # 驗(yàn)證輸入
        if not self.validate_inputs(pdf_path, output_dir):
            return
        
        # 獲取轉(zhuǎn)換選項(xiàng)
        image_format = self.format_combo.currentText()
        dpi = self.dpi_spin.value()
        convert_all_pages = self.all_pages_check.isChecked()
        
        # 準(zhǔn)備UI進(jìn)行轉(zhuǎn)換
        self.prepare_ui_for_conversion()
        
        # 創(chuàng)建并啟動轉(zhuǎn)換線程
        self.converter_thread = PDFConverterThread(
            pdf_path, output_dir, image_format, dpi, convert_all_pages)
        self.converter_thread.progress_updated.connect(self.update_progress)
        self.converter_thread.conversion_finished.connect(self.conversion_finished)
        self.converter_thread.error_occurred.connect(self.conversion_error)
        self.converter_thread.start()
    
    def validate_inputs(self, pdf_path, output_dir):
        """驗(yàn)證用戶輸入"""
        if not pdf_path:
            QMessageBox.warning(self, "警告", "請選擇PDF文件")
            return False
        
        if not os.path.exists(pdf_path):
            QMessageBox.warning(self, "警告", "PDF文件不存在")
            return False
        
        if not output_dir:
            QMessageBox.warning(self, "警告", "請選擇輸出目錄")
            return False
        
        # 嘗試創(chuàng)建輸出目錄（如果不存在）
        try:
            os.makedirs(output_dir, exist_ok=True)
        except OSError:
            QMessageBox.warning(self, "警告", "無法創(chuàng)建或訪問輸出目錄")
            return False
            
        return True
    
    def prepare_ui_for_conversion(self):
        """準(zhǔn)備UI以進(jìn)行轉(zhuǎn)換"""
        self.convert_btn.setEnabled(False)
        self.progress_bar.setVisible(True)
        self.progress_bar.setValue(0)
        self.status_label.setText("正在轉(zhuǎn)換...")
    
    def update_progress(self, value):
        """更新進(jìn)度條"""
        self.progress_bar.setValue(value)
    
    def conversion_finished(self, message):
        """轉(zhuǎn)換完成處理"""
        self.convert_btn.setEnabled(True)
        self.status_label.setText(message)
        QMessageBox.information(self, "完成", message)
    
    def conversion_error(self, error_message):
        """轉(zhuǎn)換錯誤處理"""
        self.convert_btn.setEnabled(True)
        self.progress_bar.setVisible(False)
        self.status_label.setText("轉(zhuǎn)換失敗")
        QMessageBox.critical(self, "錯誤", error_message)

pip install PyQt5 PyMuPDF