def dragEnterEvent(self, event: QDragEnterEvent):
    """拖拽進(jìn)入事件處理"""
    if event.mimeData().hasUrls():  # 檢查是否為文件/文件夾
        event.acceptProposedAction()

def dropEvent(self, event: QDropEvent):
    """拖拽釋放事件處理"""
    urls = event.mimeData().urls()
    if urls:
        path = urls[0].toLocalFile()
        if os.path.isdir(path):  # 處理文件夾
            self.dir_edit.setText(path)
        else:  # 處理單個文件
            directory = os.path.dirname(path)
            self.dir_edit.setText(directory)

class EncryptionWorker(QThread):
    """加密工作線程"""
    progress_update = pyqtSignal(str)  # 日志信號
    progress_value = pyqtSignal(int)   # 進(jìn)度條信號
    
    def run(self):
        try:
            excel = win32.gencache.EnsureDispatch('Excel.Application')
            for file in excel_files:
                # 加密邏輯...
                self.progress_update.emit(f"? 已加密: {file}")
                self.progress_value.emit(progress)
        finally:
            excel.Quit()

QPushButton {
    background-color: #4CAF50;
    border-radius: 4px;
    padding: 8px 16px;
}
QPushButton:hover {
    background-color: #45a049; /* 懸停效果 */
}

self.setWindowTitle('?? Excel批量加密工具')
self.log_text.append("? 加密失敗: 文件被占用")

try:
    wb.SaveAs(new_path, Password=password)
except Exception as e:
    self.progress_update.emit(f"? 錯誤: {str(e)}")
    failed_files.append(file_path)

import os
import sys
import win32com.client as win32
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
                             QHBoxLayout, QLabel, QLineEdit, QPushButton,
                             QTextEdit, QFileDialog, QProgressBar, QMessageBox,
                             QCheckBox)
from PyQt5.QtCore import Qt, QThread, pyqtSignal
  
  
class EncryptionWorker(QThread):
    """加密工作線程"""
    progress_update = pyqtSignal(str)
    progress_value = pyqtSignal(int)
    finished_signal = pyqtSignal(int, int)
  
    def __init__(self, directory, password, suffix):
        super().__init__()
        self.directory = directory
        self.password = password
        self.suffix = suffix
  
    def run(self):
        """執(zhí)行加密操作"""
        # 支持的Excel文件擴(kuò)展名
        excel_extensions = ['.et', '.xls', '.xlsx']
  
        # 確保目錄路徑存在
        if not os.path.exists(self.directory):
            self.progress_update.emit(f"錯誤: 目錄 '{self.directory}' 不存在!")
            return
  
        # 計數(shù)器
        encrypted_files = []
        failed_files = []
  
        self.progress_update.emit(f"開始掃描目錄: {self.directory}")
  
        # 首先計算總文件數(shù)
        excel_files = []
        for root, _, files in os.walk(self.directory):
            for file in files:
                file_path = os.path.join(root, file)
                file_ext = os.path.splitext(file)[1].lower()
                if file_ext in excel_extensions:
                    excel_files.append(file_path)
  
        total_files = len(excel_files)
        self.progress_update.emit(f"找到 {total_files} 個Excel文件")
  
        # 初始化Excel應(yīng)用程序
        excel = None
        try:
            excel = win32.gencache.EnsureDispatch('Excel.Application')
            excel.DisplayAlerts = False
  
            # 遍歷處理所有文件
            for index, file_path in enumerate(excel_files):
                try:
                    self.progress_update.emit(f"正在處理: {file_path}")
                    self.progress_value.emit(int((index / total_files) * 100) if total_files > 0 else 0)
  
                    # 生成新文件名
                    file_dir = os.path.dirname(file_path)
                    file_name, file_ext = os.path.splitext(os.path.basename(file_path))
                    new_file_name = f"{file_name}{self.suffix}{file_ext}"
                    new_file_path = os.path.join(file_dir, new_file_name)
  
                    # 打開Excel文件
                    wb = excel.Workbooks.Open(os.path.abspath(file_path))
  
                    # 設(shè)置密碼并另存為新文件
                    wb.SaveAs(os.path.abspath(new_file_path), Password=self.password)
                    wb.Close()
  
                    encrypted_files.append(new_file_path)
                    self.progress_update.emit(f"已加密并保存為: {new_file_path}")
                except Exception as e:
                    failed_files.append((file_path, str(e)))
                    self.progress_update.emit(f"加密失敗: {file_path} - 錯誤: {str(e)}")
        except Exception as e:
            self.progress_update.emit(f"初始化Excel應(yīng)用程序失敗: {str(e)}")
        finally:
            # 確保Excel應(yīng)用程序被關(guān)閉
            if excel:
                try:
                    excel.Quit()
                except:
                    pass
  
        self.progress_value.emit(100)
        self.finished_signal.emit(len(encrypted_files), len(failed_files))
  
  
class ExcelEncryptorApp(QMainWindow):
    """Excel文件批量加密工具主窗口"""
  
    def __init__(self):
        super().__init__()
        self.init_ui()
  
    def init_ui(self):
        """初始化用戶界面"""
        # 設(shè)置窗口標(biāo)題和大小
        self.setWindowTitle('Excel文件批量加密工具')
        self.setGeometry(300, 300, 600, 500)
  
        # 創(chuàng)建中央部件
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
  
        # 創(chuàng)建主布局
        main_layout = QVBoxLayout(central_widget)
  
        # 目錄選擇部分
        dir_layout = QHBoxLayout()
        dir_label = QLabel('目錄路徑:')
        self.dir_edit = QLineEdit()
        browse_btn = QPushButton('瀏覽...')
        browse_btn.clicked.connect(self.browse_directory)
  
        dir_layout.addWidget(dir_label)
        dir_layout.addWidget(self.dir_edit)
        dir_layout.addWidget(browse_btn)
  
        # 密碼輸入部分
        pwd_layout = QHBoxLayout()
        pwd_label = QLabel('加密密碼:')
        self.pwd_edit = QLineEdit()
        self.pwd_edit.setEchoMode(QLineEdit.Password)
  
        pwd_layout.addWidget(pwd_label)
        pwd_layout.addWidget(self.pwd_edit)
  
        # 文件后綴輸入部分
        suffix_layout = QHBoxLayout()
        suffix_label = QLabel('文件后綴:')
        self.suffix_edit = QLineEdit()
        self.suffix_edit.setText("_加密")
  
        suffix_layout.addWidget(suffix_label)
        suffix_layout.addWidget(self.suffix_edit)
  
        # 操作按鈕
        btn_layout = QHBoxLayout()
        self.encrypt_btn = QPushButton('開始加密')
        self.encrypt_btn.clicked.connect(self.start_encryption)
  
        btn_layout.addStretch()
        btn_layout.addWidget(self.encrypt_btn)
  
        # 進(jìn)度條
        self.progress_bar = QProgressBar()
        self.progress_bar.setValue(0)
  
        # 日志輸出區(qū)域
        log_label = QLabel('處理日志:')
        self.log_text = QTextEdit()
        self.log_text.setReadOnly(True)
  
        # 添加所有部件到主布局
        main_layout.addLayout(dir_layout)
        main_layout.addLayout(pwd_layout)
        main_layout.addLayout(suffix_layout)
        main_layout.addLayout(btn_layout)
        main_layout.addWidget(self.progress_bar)
        main_layout.addWidget(log_label)
        main_layout.addWidget(self.log_text)
  
    def browse_directory(self):
        """打開目錄選擇對話框"""
        directory = QFileDialog.getExistingDirectory(self, '選擇目錄')
        if directory:
            self.dir_edit.setText(directory)
  
    def start_encryption(self):
        """開始加密操作"""
        directory = self.dir_edit.text().strip()
        password = self.pwd_edit.text().strip()
        suffix = self.suffix_edit.text().strip()
  
        # 驗證輸入
        if not directory:
            QMessageBox.warning(self, '輸入錯誤', '請選擇要處理的目錄')
            return
  
        if not password:
            QMessageBox.warning(self, '輸入錯誤', '請輸入加密密碼')
            return
  
        if not suffix:
            reply = QMessageBox.question(self, '確認(rèn)操作',
                                         '您沒有輸入文件后綴，加密后的文件將覆蓋原文件，是否繼續(xù)?',
                                         QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
            if reply == QMessageBox.No:
                return
  
        # 確認(rèn)操作
        reply = QMessageBox.question(self, '確認(rèn)操作',
                                     f'將對目錄 "{directory}" 中的所有Excel文件進(jìn)行加密，并保存為帶有后綴 "{suffix}" 的新文件，是否繼續(xù)?',
                                     QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
  
        if reply == QMessageBox.Yes:
            # 清空日志和進(jìn)度條
            self.log_text.clear()
            self.progress_bar.setValue(0)
  
            # 禁用按鈕，防止重復(fù)操作
            self.encrypt_btn.setEnabled(False)
  
            # 創(chuàng)建并啟動工作線程
            self.worker = EncryptionWorker(directory, password, suffix)
            self.worker.progress_update.connect(self.update_log)
            self.worker.progress_value.connect(self.progress_bar.setValue)
            self.worker.finished_signal.connect(self.encryption_finished)
            self.worker.start()
  
    def update_log(self, message):
        """更新日志輸出"""
        self.log_text.append(message)
        # 自動滾動到底部
        self.log_text.verticalScrollBar().setValue(
            self.log_text.verticalScrollBar().maximum()
        )
  
    def encryption_finished(self, encrypted, failed):
        """加密完成后的處理"""
        self.update_log(f"\n加密完成! 成功加密 {encrypted} 個文件，失敗 {failed} 個文件。")
        self.encrypt_btn.setEnabled(True)
  
        # 顯示完成消息
        QMessageBox.information(self, '操作完成',
                                f'加密完成!\n成功加密 {encrypted} 個文件\n失敗 {failed} 個文件')
  
  
def main():
    """程序入口點"""
    app = QApplication(sys.argv)
    window = ExcelEncryptorApp()
    window.show()
    sys.exit(app.exec_())
  
  
if __name__ == "__main__":
    main()