import sys
import os
import time
import subprocess
from PyQt5.QtWidgets import (
    QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, 
    QLineEdit, QPushButton, QListWidget, QLabel, QProgressBar, 
    QMessageBox, QCheckBox, QMenu, QAction, QFileDialog, QSplitter
)
from PyQt5.QtCore import Qt, QThread, pyqtSignal
from PyQt5.QtGui import QFont

class SearchWorker(QThread):
    """后臺(tái)搜索線程，負(fù)責(zé)文件系統(tǒng)遍歷"""
    update_progress = pyqtSignal(int)          # 已掃描文件數(shù)
    found_file = pyqtSignal(str)               # 找到的文件路徑
    search_complete = pyqtSignal()             # 搜索完成信號(hào)
    error_occurred = pyqtSignal(str)           # 錯(cuò)誤信號(hào)
    current_dir = pyqtSignal(str)              # 當(dāng)前搜索目錄

    def __init__(self, search_term, search_paths, include_hidden):
        super().__init__()
        self.search_term = search_term.lower()
        self.search_paths = search_paths
        self.include_hidden = include_hidden
        self.cancel_search = False
        self.file_count = 0
        self.found_count = 0

    def run(self):
        """執(zhí)行搜索操作"""
        try:
            self.file_count = 0
            self.found_count = 0
            
            # 遍歷所有指定路徑
            for path in self.search_paths:
                if self.cancel_search:
                    break
                self._search_directory(path)
            
            self.search_complete.emit()
        except Exception as e:
            self.error_occurred.emit(str(e))

    def _search_directory(self, path):
        """遞歸搜索目錄"""
        if self.cancel_search:
            return

        # 通知UI當(dāng)前搜索目錄
        self.current_dir.emit(path)
        
        try:
            # 使用高效的文件系統(tǒng)遍歷
            with os.scandir(path) as entries:
                for entry in entries:
                    if self.cancel_search:
                        return

                    try:
                        # 跳過隱藏文件/目錄（根據(jù)設(shè)置）
                        if not self.include_hidden and entry.name.startswith('.'):
                            continue

                        # 處理目錄
                        if entry.is_dir(follow_symlinks=False):
                            # 跳過系統(tǒng)目錄提升速度
                            if entry.name.lower() in [
                                '$recycle.bin', 
                                'system volume information', 
                                'windows', 
                                'program files', 
                                'program files (x86)'
                            ]:
                                continue
                            self._search_directory(entry.path)
                        # 處理文件
                        else:
                            self.file_count += 1
                            # 每100個(gè)文件更新一次進(jìn)度
                            if self.file_count % 100 == 0:
                                self.update_progress.emit(self.file_count)
                            
                            # 檢查文件名是否匹配
                            if self.search_term in entry.name.lower():
                                self.found_file.emit(entry.path)
                                self.found_count += 1
                    except PermissionError:
                        continue  # 跳過權(quán)限錯(cuò)誤
                    except Exception:
                        continue  # 跳過其他錯(cuò)誤
        except PermissionError:
            return  # 跳過無(wú)權(quán)限目錄
        except Exception:
            return  # 跳過其他錯(cuò)誤

    def cancel(self):
        """取消搜索"""
        self.cancel_search = True


class FileSearchApp(QMainWindow):
    """文件搜索應(yīng)用程序主窗口"""
    def __init__(self):
        super().__init__()
        self.setWindowTitle("Windows文件快速檢索工具")
        self.setGeometry(100, 100, 1000, 700)
        self.init_ui()
        self.search_thread = None
        self.current_selected_file = ""
        self.last_search_time = 0

    def init_ui(self):
        """初始化用戶界面"""
        # 主窗口設(shè)置
        main_widget = QWidget()
        self.setCentralWidget(main_widget)
        main_layout = QVBoxLayout(main_widget)
        
        # 使用分割器布局
        splitter = QSplitter(Qt.Vertical)
        main_layout.addWidget(splitter)
        
        # 控制面板
        control_panel = self.create_control_panel()
        splitter.addWidget(control_panel)
        
        # 結(jié)果面板
        result_panel = self.create_result_panel()
        splitter.addWidget(result_panel)
        
        # 設(shè)置分割比例
        splitter.setSizes([200, 500])
        
        # 狀態(tài)欄
        self.status_bar = self.statusBar()
        self.selected_file_label = QLabel("未選擇文件")
        self.status_bar.addPermanentWidget(self.selected_file_label)
        self.status_bar.showMessage("就緒")
        
        # 應(yīng)用樣式
        self.apply_styles()

    def create_control_panel(self):
        """創(chuàng)建控制面板"""
        panel = QWidget()
        layout = QVBoxLayout(panel)
        
        # 搜索輸入?yún)^(qū)域
        search_layout = QHBoxLayout()
        self.search_input = QLineEdit()
        self.search_input.setPlaceholderText("輸入文件名或擴(kuò)展名 (例如: *.txt, report.docx)")
        self.search_input.returnPressed.connect(self.start_search)
        search_layout.addWidget(self.search_input)
        
        self.search_button = QPushButton("搜索")
        self.search_button.clicked.connect(self.start_search)
        self.search_button.setFixedWidth(100)
        search_layout.addWidget(self.search_button)
        layout.addLayout(search_layout)
        
        # 選項(xiàng)區(qū)域
        options_layout = QHBoxLayout()
        self.include_hidden_check = QCheckBox("包含隱藏文件和系統(tǒng)文件")
        options_layout.addWidget(self.include_hidden_check)
        
        self.search_drives_combo = QCheckBox("搜索所有驅(qū)動(dòng)器")
        self.search_drives_combo.stateChanged.connect(self.toggle_drive_search)
        options_layout.addWidget(self.search_drives_combo)
        
        self.browse_button = QPushButton("選擇搜索目錄...")
        self.browse_button.clicked.connect(self.browse_directory)
        self.browse_button.setFixedWidth(120)
        options_layout.addWidget(self.browse_button)
        options_layout.addStretch()
        layout.addLayout(options_layout)
        
        # 當(dāng)前搜索目錄顯示
        self.current_dir_label = QLabel("搜索目錄: 用戶目錄")
        self.current_dir_label.setStyleSheet("color: #666; font-style: italic;")
        layout.addWidget(self.current_dir_label)
        
        # 驅(qū)動(dòng)器選擇區(qū)域
        self.drive_selection_widget = QWidget()
        drive_layout = QHBoxLayout(self.drive_selection_widget)
        drive_layout.addWidget(QLabel("選擇要搜索的驅(qū)動(dòng)器:"))
        
        # 添加可用驅(qū)動(dòng)器
        self.drive_buttons = []
        for drive in "ABCDEFGHIJKLMNOPQRSTUVWXYZ":
            if os.path.exists(f"{drive}:\\"):
                btn = QCheckBox(f"{drive}:")
                btn.setChecked(True)
                self.drive_buttons.append(btn)
                drive_layout.addWidget(btn)
        
        drive_layout.addStretch()
        layout.addWidget(self.drive_selection_widget)
        self.drive_selection_widget.setVisible(False)
        
        # 進(jìn)度顯示
        progress_layout = QHBoxLayout()
        self.progress_label = QLabel("準(zhǔn)備搜索...")
        progress_layout.addWidget(self.progress_label)
        progress_layout.addStretch()
        
        self.file_count_label = QLabel("已找到: 0 文件")
        self.file_count_label.setAlignment(Qt.AlignRight)
        progress_layout.addWidget(self.file_count_label)
        layout.addLayout(progress_layout)
        
        # 進(jìn)度條
        self.progress_bar = QProgressBar()
        self.progress_bar.setRange(0, 100)
        self.progress_bar.setValue(0)
        layout.addWidget(self.progress_bar)
        
        return panel

    def create_result_panel(self):
        """創(chuàng)建結(jié)果面板"""
        panel = QWidget()
        layout = QVBoxLayout(panel)
        layout.addWidget(QLabel("搜索結(jié)果:"))
        
        self.result_list = QListWidget()
        self.result_list.itemSelectionChanged.connect(self.update_selected_file_info)
        self.result_list.itemDoubleClicked.connect(self.open_file)
        self.result_list.setContextMenuPolicy(Qt.CustomContextMenu)
        self.result_list.customContextMenuRequested.connect(self.show_context_menu)
        layout.addWidget(self.result_list, 1)
        
        return panel

    def apply_styles(self):
        """應(yīng)用UI樣式"""
        self.setStyleSheet("""
            QMainWindow { background-color: #f0f0f0; }
            QLineEdit {
                padding: 8px;
                font-size: 14px;
                border: 1px solid #ccc;
                border-radius: 4px;
            }
            QListWidget {
                font-family: Consolas, 'Courier New', monospace;
                font-size: 12px;
                border: 1px solid #ddd;
            }
            QProgressBar {
                text-align: center;
                height: 20px;
            }
            QPushButton {
                padding: 6px 12px;
                background-color: #4CAF50;
                color: white;
                border: none;
                border-radius: 4px;
            }
            QPushButton:hover { background-color: #45a049; }
            QPushButton:pressed { background-color: #3d8b40; }
            QCheckBox { padding: 5px; }
        """)
        
        # 設(shè)置全局字體
        font = QFont("Segoe UI", 10)
        QApplication.setFont(font)

    def toggle_drive_search(self, state):
        """切換驅(qū)動(dòng)器搜索選項(xiàng)顯示"""
        self.drive_selection_widget.setVisible(state == Qt.Checked)

    def browse_directory(self):
        """選擇搜索目錄"""
        directory = QFileDialog.getExistingDirectory(self, "選擇搜索目錄", os.path.expanduser("~"))
        if directory:
            self.current_dir_label.setText(f"搜索目錄: {directory}")
            self.search_drives_combo.setChecked(False)
            self.drive_selection_widget.setVisible(False)

    def start_search(self):
        """開始搜索操作"""
        # 防抖處理
        current_time = time.time()
        if current_time - self.last_search_time < 1.0:
            return
        self.last_search_time = current_time
        
        # 驗(yàn)證輸入
        search_term = self.search_input.text().strip()
        if not search_term:
            QMessageBox.warning(self, "輸入錯(cuò)誤", "請(qǐng)輸入搜索關(guān)鍵詞")
            return
        
        # 準(zhǔn)備搜索
        self.result_list.clear()
        self.file_count_label.setText("已找到: 0 文件")
        self.status_bar.showMessage("正在搜索...")
        
        # 確定搜索路徑
        if self.search_drives_combo.isChecked():
            selected_drives = []
            for btn in self.drive_buttons:
                if btn.isChecked():
                    drive = btn.text().replace(":", "")
                    selected_drives.append(f"{drive}:\\")
            
            if not selected_drives:
                QMessageBox.warning(self, "選擇錯(cuò)誤", "請(qǐng)至少選擇一個(gè)驅(qū)動(dòng)器")
                return
                
            self.current_dir_label.setText(f"搜索目錄: {len(selected_drives)}個(gè)驅(qū)動(dòng)器")
            search_paths = selected_drives
        else:
            current_dir_text = self.current_dir_label.text()
            if current_dir_text.startswith("搜索目錄: "):
                search_path = current_dir_text[6:]
                if not os.path.exists(search_path):
                    search_path = os.path.expanduser("~")
            else:
                search_path = os.path.expanduser("~")
            
            search_paths = [search_path]
            self.current_dir_label.setText(f"搜索目錄: {search_path}")
        
        # 創(chuàng)建并啟動(dòng)搜索線程
        if self.search_thread and self.search_thread.isRunning():
            self.search_thread.cancel()
            self.search_thread.wait(1000)
        
        self.search_thread = SearchWorker(
            search_term, 
            search_paths, 
            self.include_hidden_check.isChecked()
        )
        
        # 連接信號(hào)
        self.search_thread.found_file.connect(self.add_result)
        self.search_thread.update_progress.connect(self.update_progress)
        self.search_thread.search_complete.connect(self.search_finished)
        self.search_thread.error_occurred.connect(self.show_error)
        self.search_thread.current_dir.connect(self.update_current_dir)
        
        # 更新UI狀態(tài)
        self.search_button.setEnabled(False)
        self.search_button.setText("停止搜索")
        self.search_button.clicked.disconnect()
        self.search_button.clicked.connect(self.cancel_search)
        
        # 啟動(dòng)線程
        self.search_thread.start()

    def add_result(self, file_path):
        """添加搜索結(jié)果"""
        self.result_list.addItem(file_path)
        count = self.result_list.count()
        self.file_count_label.setText(f"已找到: {count} 文件")
        self.status_bar.showMessage(f"找到 {count} 個(gè)匹配文件")

    def update_progress(self, file_count):
        """更新進(jìn)度顯示"""
        self.progress_label.setText(f"已掃描 {file_count} 個(gè)文件...")
        self.progress_bar.setRange(0, 0)  # 不確定模式

    def update_current_dir(self, directory):
        """更新當(dāng)前搜索目錄"""
        self.status_bar.showMessage(f"正在搜索: {directory}")

    def search_finished(self):
        """搜索完成處理"""
        self.progress_bar.setRange(0, 100)
        self.progress_bar.setValue(100)
        
        if self.search_thread:
            self.progress_label.setText(
                f"搜索完成！共掃描 {self.search_thread.file_count} 個(gè)文件，"
                f"找到 {self.search_thread.found_count} 個(gè)匹配項(xiàng)"
            )
            self.status_bar.showMessage(f"搜索完成，找到 {self.result_list.count()} 個(gè)匹配文件")
        
        # 重置搜索按鈕
        self.search_button.setText("搜索")
        self.search_button.clicked.disconnect()
        self.search_button.clicked.connect(self.start_search)
        self.search_button.setEnabled(True)

    def cancel_search(self):
        """取消搜索"""
        if self.search_thread and self.search_thread.isRunning():
            self.search_thread.cancel()
            self.search_thread.wait(1000)
            
            self.progress_bar.setRange(0, 100)
            self.progress_bar.setValue(0)
            self.progress_label.setText("搜索已取消")
            self.status_bar.showMessage(f"已取消搜索，找到 {self.result_list.count()} 個(gè)文件")
            
            self.search_button.setText("搜索")
            self.search_button.clicked.disconnect()
            self.search_button.clicked.connect(self.start_search)
            self.search_button.setEnabled(True)

    def open_file(self, item):
        """打開文件"""
        file_path = item.text()
        self.current_selected_file = file_path
        try:
            os.startfile(file_path)
        except Exception as e:
            QMessageBox.critical(self, "打開文件錯(cuò)誤", f"無(wú)法打開文件:\n{str(e)}")

    def show_error(self, error_msg):
        """顯示錯(cuò)誤信息"""
        QMessageBox.critical(self, "搜索錯(cuò)誤", f"搜索過程中發(fā)生錯(cuò)誤:\n{error_msg}")
        self.cancel_search()

    def show_context_menu(self, position):
        """顯示右鍵菜單"""
        if not self.result_list.selectedItems():
            return
            
        selected_item = self.result_list.currentItem()
        file_path = selected_item.text()
        self.current_selected_file = file_path
        
        menu = QMenu()
        
        # 添加菜單項(xiàng)
        open_action = QAction("打開文件", self)
        open_action.triggered.connect(lambda: self.open_file(selected_item))
        menu.addAction(open_action)
        
        open_location_action = QAction("打開文件所在位置", self)
        open_location_action.triggered.connect(self.open_file_location)
        menu.addAction(open_location_action)
        
        copy_path_action = QAction("復(fù)制文件路徑", self)
        copy_path_action.triggered.connect(self.copy_file_path)
        menu.addAction(copy_path_action)
        
        menu.exec_(self.result_list.mapToGlobal(position))
    
    def open_file_location(self):
        """打開文件所在位置"""
        if not self.current_selected_file:
            return
            
        try:
            subprocess.Popen(f'explorer /select,"{self.current_selected_file}"')
        except Exception as e:
            QMessageBox.critical(self, "打開位置錯(cuò)誤", f"無(wú)法打開文件所在位置:\n{str(e)}")
    
    def copy_file_path(self):
        """復(fù)制文件路徑"""
        if not self.current_selected_file:
            return
            
        clipboard = QApplication.clipboard()
        clipboard.setText(self.current_selected_file)
        self.status_bar.showMessage(f"已復(fù)制路徑: {self.current_selected_file}", 3000)
    
    def update_selected_file_info(self):
        """更新文件信息顯示"""
        selected_items = self.result_list.selectedItems()
        if not selected_items:
            self.selected_file_label.setText("未選擇文件")
            self.current_selected_file = ""
            return
            
        file_path = selected_items[0].text()
        self.current_selected_file = file_path
        
        try:
            # 獲取文件信息
            file_size = os.path.getsize(file_path)
            file_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(os.path.getmtime(file_path)))
            
            # 格式化文件大小
            if file_size < 1024:
                size_str = f"{file_size} B"
            elif file_size < 1024*1024:
                size_str = f"{file_size/1024:.2f} KB"
            else:
                size_str = f"{file_size/(1024*1024):.2f} MB"
            
            # 更新狀態(tài)欄
            file_name = os.path.basename(file_path)
            self.selected_file_label.setText(
                f"{file_name} | 大小: {size_str} | 修改時(shí)間: {file_time}"
            )
        except:
            self.selected_file_label.setText(file_path)


if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = FileSearchApp()
    window.show()
    sys.exit(app.exec_())

with os.scandir(path) as entries:
    for entry in entries:
        if entry.is_dir():
            # 處理目錄
        else:
            # 處理文件

if entry.name.lower() in [
    '$recycle.bin', 
    'system volume information', 
    'windows', 
    'program files', 
    'program files (x86)'
]:
    continue

self.file_count += 1
if self.file_count % 100 == 0:
    self.update_progress.emit(self.file_count)

self.search_thread = SearchWorker(...)
self.search_thread.start()