基于Python实现一个目录/文件递归检索工具

Written by

in

文章目录
  • 项目结构分析 :快速查看项目目录结构 文件系统清理 :识别特定类型的文件(如临时文件) 文档编制 :生成项目目录树文档 资产盘点 :统计特定类型的文件数量 系统维护 :查找分散的配置文件或日志文件
  • import os import sys import re import json import fnmatch from PyQt5.QtWidgets import ( QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QPushButton, QRadioButton, QButtonGroup, QGroupBox, QFileDialog, QTextEdit, QComboBox, QMessageBox, QCheckBox, QListWidget, QListWidgetItem, QInputDialog ) from PyQt5.QtCore import Qt from PyQt5.QtGui import QFont, QPalette, QColor # 获取脚本所在目录 if getattr(sys, ‘frozen’, False): # 如果是打包后的可执行文件 APP_DIR = os.path.dirname(sys.executable) else: # 如果是脚本文件 APP_DIR = os.path.dirname(os.path.abspath(__file__)) # 修改配置文件路径为脚本所在目录 CONFIG_FILE = os.path.join(APP_DIR, “config.json”) class FileSearchApp(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle(“目录/文件递归检索工具”) self.setGeometry(300, 300, 800, 650) # 初始化变量 self.ignore_configs = [] self.current_ignore_config = {“name”: “默认配置”, “patterns”: []} self.load_config() # 创建UI self.init_ui() class FileSearchApp(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle(“目录/文件递归检索工具 V1.0 by:Sunf10wer”) self.setGeometry(300, 300, 800, 650) # 初始化变量 self.ignore_configs = [] self.current_ignore_config = {“name”: “默认配置”, “patterns”: []} self.load_config() # 创建UI self.init_ui() def init_ui(self): # 主布局 main_widget = QWidget() main_layout = QVBoxLayout() main_widget.setLayout(main_layout) self.setCentralWidget(main_widget) # 设置标题样式 title_label = QLabel(“目录/文件递归检索工具”) title_font = QFont(“Arial”, 16, QFont.Bold) title_label.setFont(title_font) title_label.setAlignment(Qt.AlignCenter) title_label.setStyleSheet(“color: #FFFFFF; padding: 10px;”) main_layout.addWidget(title_label) # 目录选择 dir_layout = QHBoxLayout() dir_label = QLabel(“目标目录:”) self.dir_entry = QLineEdit() self.dir_entry.setPlaceholderText(“请选择或输入要检索的目录…”) browse_button = QPushButton(“浏览…”) browse_button.clicked.connect(self.browse_directory) dir_layout.addWidget(dir_label) dir_layout.addWidget(self.dir_entry, 4) dir_layout.addWidget(browse_button, 1) main_layout.addLayout(dir_layout) # 输出文件选择 output_layout = QHBoxLayout() output_label = QLabel(“输出文件:”) self.output_entry = QLineEdit() self.output_entry.setPlaceholderText(“输出文件名…”) output_browse_button = QPushButton(“浏览…”) output_browse_button.clicked.connect(self.browse_output_file) output_layout.addWidget(output_label) output_layout.addWidget(self.output_entry, 4) output_layout.addWidget(output_browse_button, 1) main_layout.addLayout(output_layout) # 检索类型选择 search_type_group = QGroupBox(“检索类型”) search_layout = QHBoxLayout() self.folder_radio = QRadioButton(“仅文件夹”) self.file_radio = QRadioButton(“仅文件”) self.both_radio = QRadioButton(“文件和文件夹”) self.both_radio.setChecked(True) self.search_type_group = QButtonGroup() self.search_type_group.addButton(self.folder_radio) self.search_type_group.addButton(self.file_radio) self.search_type_group.addButton(self.both_radio) # 文件后缀过滤 suffix_layout = QHBoxLayout() suffix_label = QLabel(“文件后缀(用分号分隔):”) self.suffix_entry = QLineEdit() self.suffix_entry.setPlaceholderText(“例如: .txt; .py; .jpg”) suffix_layout.addWidget(suffix_label) suffix_layout.addWidget(self.suffix_entry) search_layout.addWidget(self.folder_radio) search_layout.addWidget(self.file_radio) search_layout.addWidget(self.both_radio) search_layout.addStretch() search_type_group.setLayout(search_layout) main_layout.addWidget(search_type_group) main_layout.addLayout(suffix_layout) # 屏蔽词管理 ignore_group = QGroupBox(“屏蔽词管理”) ignore_layout = QVBoxLayout() # 屏蔽词配置选择 config_layout = QHBoxLayout() config_label = QLabel(“当前配置:”) self.config_combo = QComboBox() self.config_combo.setMinimumWidth(150) self.config_combo.currentIndexChanged.connect(self.config_selected) new_config_btn = QPushButton(“新建配置”) new_config_btn.clicked.connect(self.create_new_config) config_layout.addWidget(config_label) config_layout.addWidget(self.config_combo, 1) config_layout.addWidget(new_config_btn) # 屏蔽词列表 ignore_list_layout = QVBoxLayout() list_label = QLabel(“屏蔽词列表(支持通配符,如 *.tmp; backup_*)”) self.ignore_list = QListWidget() self.ignore_list.setAlternatingRowColors(True) add_btn = QPushButton(“添加屏蔽词”) add_btn.clicked.connect(self.add_ignore_pattern) remove_btn = QPushButton(“移除选中”) remove_btn.clicked.connect(self.remove_selected_pattern) list_btn_layout = QHBoxLayout() list_btn_layout.addWidget(add_btn) list_btn_layout.addWidget(remove_btn) ignore_list_layout.addWidget(list_label) ignore_list_layout.addWidget(self.ignore_list) ignore_list_layout.addLayout(list_btn_layout) ignore_layout.addLayout(config_layout) ignore_layout.addLayout(ignore_list_layout) ignore_group.setLayout(ignore_layout) main_layout.addWidget(ignore_group) # 操作按钮 button_layout = QHBoxLayout() self.search_btn = QPushButton(“开始检索”) self.search_btn.setStyleSheet( “background-color: #3498db; color: white; font-weight: bold; padding: 8px;” ) self.search_btn.clicked.connect(self.start_search) export_btn = QPushButton(“导出配置”) export_btn.clicked.connect(self.export_config) button_layout.addStretch() button_layout.addWidget(self.search_btn) button_layout.addWidget(export_btn) button_layout.addStretch() main_layout.addLayout(button_layout) # 状态栏 self.status_bar = self.statusBar() self.status_label = QLabel(“就绪”) self.status_bar.addWidget(self.status_label) # 更新UI self.update_config_combo() self.update_ignore_list() # 连接信号 self.dir_entry.textChanged.connect(self.update_output_filename) def load_config(self): “””从配置文件加载屏蔽词配置””” if os.path.exists(CONFIG_FILE): try: with open(CONFIG_FILE, ‘r’, encoding=’utf-8′) as f: self.ignore_configs = json.load(f) # 确保至少有一个默认配置 if not any(cfg[‘name’] == ‘默认配置’ for cfg in self.ignore_configs): self.ignore_configs.insert(0, {“name”: “默认配置”, “patterns”: []}) except: self.ignore_configs = [{“name”: “默认配置”, “patterns”: []}] else: self.ignore_configs = [{“name”: “默认配置”, “patterns”: []}] self.current_ignore_config = self.ignore_configs[0] def save_config(self): “””保存屏蔽词配置到文件””” try: with open(CONFIG_FILE, ‘w’, encoding=’utf-8′) as f: json.dump(self.ignore_configs, f, ensure_ascii=False, indent=2) return True except Exception as e: QMessageBox.critical(self, “保存错误”, f”保存配置时出错: {str(e)}”) return False def update_config_combo(self): “””更新配置下拉框””” self.config_combo.clear() for config in self.ignore_configs: self.config_combo.addItem(config[‘name’]) # 选择当前配置 current_index = next( (i for i, config in enumerate(self.ignore_configs) if config[‘name’] == self.current_ignore_config[‘name’]), 0 ) self.config_combo.setCurrentIndex(current_index) def update_ignore_list(self): “””更新屏蔽词列表””” self.ignore_list.clear() for pattern in self.current_ignore_config[‘patterns’]: self.ignore_list.addItem(pattern) def config_selected(self, index): “””配置选择改变事件””” if 0 <= index < len(self.ignore_configs): self.current_ignore_config = self.ignore_configs[index] self.update_ignore_list() def create_new_config(self): “””创建新的屏蔽词配置””” name, ok = QInputDialog.getText( self, “新建配置”, “输入配置名称:”, text=f”配置{len(self.ignore_configs)+1}” ) if ok and name: # 检查名称是否已存在 if any(cfg[‘name’] == name for cfg in self.ignore_configs): QMessageBox.warning(self, “名称冲突”, f”配置名 ‘{name}’ 已存在!”) return new_config = {“name”: name, “patterns”: []} self.ignore_configs.append(new_config) self.current_ignore_config = new_config self.update_config_combo() self.update_ignore_list() self.save_config() def add_ignore_pattern(self): “””添加屏蔽词””” pattern, ok = QInputDialog.getText( self, “添加屏蔽词”, “请输入屏蔽词(支持通配符):” ) if ok and pattern: if pattern not in self.current_ignore_config[‘patterns’]: self.current_ignore_config[‘patterns’].append(pattern) self.update_ignore_list() self.save_config() def remove_selected_pattern(self): “””移除选中的屏蔽词””” selected_items = self.ignore_list.selectedItems() if not selected_items: return for item in selected_items: pattern = item.text() if pattern in self.current_ignore_config[‘patterns’]: self.current_ignore_config[‘patterns’].remove(pattern) self.update_ignore_list() self.save_config() def browse_directory(self): “””浏览目录””” directory = QFileDialog.getExistingDirectory(self, “选择目录”) if directory: self.dir_entry.setText(directory) self.update_output_filename() def browse_output_file(self): “””浏览输出文件””” # 获取默认输出路径 default_path = self.output_entry.text() or self.get_default_output_path() # 确保默认路径包含文件名 if os.path.isdir(default_path): default_path = os.path.join(default_path, self.get_default_filename()) # 打开文件对话框 file_path, _ = QFileDialog.getSaveFileName( self, “保存结果”, default_path, “文本文件 (*.txt)” ) if file_path: # 确保文件扩展名正确 if not file_path.endswith(‘.txt’): file_path += ‘.txt’ self.output_entry.setText(file_path) def get_default_output_path(self): “””获取默认输出目录””” # 优先使用目标目录所在位置 if self.dir_entry.text(): return os.path.dirname(self.dir_entry.text()) # 使用当前工作目录作为备选 return os.getcwd() def update_output_filename(self): “””当目录改变时更新默认输出文件名””” # 仅当输出框为空时更新 if not self.output_entry.text(): self.output_entry.setText(self.get_default_output_path()) def get_default_filename(self): “””获取默认文件名””” directory = self.dir_entry.text() if directory: # 获取目录名作为文件名基础 base_name = os.path.basename(directory) or “root” return f”{base_name}_检索结果.txt” return “检索结果.txt” def export_config(self): “””导出当前配置””” if not self.current_ignore_config[‘patterns’]: QMessageBox.information(self, “导出配置”, “当前配置没有屏蔽词!”) return file_path, _ = QFileDialog.getSaveFileName( self, “导出配置”, “”, “JSON文件 (*.json)” ) if file_path: if not file_path.endswith(‘.json’): file_path += ‘.json’ try: with open(file_path, ‘w’, encoding=’utf-8′) as f: json.dump(self.current_ignore_config, f, ensure_ascii=False, indent=2) QMessageBox.information(self, “导出成功”, f”配置已导出到:n{file_path}”) except Exception as e: QMessageBox.critical(self, “导出错误”, f”导出配置时出错: {str(e)}”) def start_search(self): “””开始检索””” # 获取输入参数 target_dir = self.dir_entry.text().strip() if not target_dir or not os.path.isdir(target_dir): QMessageBox.warning(self, “目录错误”, “请选择有效的目标目录!”) return # 获取输出文件路径 output_file = self.output_entry.text().strip() if not output_file: # 如果没有指定输出文件,使用默认文件名 output_file = os.path.join( self.get_default_output_path(), self.get_default_filename() ) self.output_entry.setText(output_file) if not output_file.endswith(‘.txt’): output_file += ‘.txt’ # 确保输出目录存在 output_dir = os.path.dirname(output_file) if output_dir and not os.path.exists(output_dir): try: os.makedirs(output_dir) except Exception as e: QMessageBox.critical(self, “目录错误”, f”无法创建输出目录: {str(e)}”) return # 检查检索类型 search_folders = self.folder_radio.isChecked() search_files = self.file_radio.isChecked() search_both = self.both_radio.isChecked() file_extensions = [] if search_files or search_both: ext_str = self.suffix_entry.text().strip() if ext_str: file_extensions = [ext.strip().lower() for ext in ext_str.split(‘;’) if ext.strip()] # 获取屏蔽词 ignore_patterns = self.current_ignore_config[‘patterns’] # 执行检索 self.status_label.setText(“正在检索…”) QApplication.processEvents() # 更新UI try: # 获取层级结构的检索结果 results = [] self.recursive_traverse( target_dir, target_dir, results, 0, search_folders, search_files, search_both, file_extensions, ignore_patterns ) # 写入文件 with open(output_file, ‘w’, encoding=’utf-8′) as f: # 写入头部信息 f.write(f”检索目录: {target_dir}n”) if search_folders: f.write(f”检索类型: 仅文件夹n”) elif search_files: f.write(f”检索类型: 仅文件n”) else: f.write(f”检索类型: 文件和文件夹n”) if (search_files or search_both) and file_extensions: f.write(f”文件后缀: {‘, ‘.join(file_extensions)}n”) if ignore_patterns: f.write(f”屏蔽配置: {self.current_ignore_config[‘name’]}n”) f.write(f”屏蔽词: {‘, ‘.join(ignore_patterns)}n”) f.write(“n” + “=” * 70 + “nn”) # 写入层级结构 total_items = 0 total_folders = 0 total_files = 0 for item in results: # 根据层级深度添加缩进 indent = “│ ” * (item[‘depth’] – 1) prefix = “├── ” if item[‘depth’] > 0 else “” # 添加文件夹/文件标识 if item[‘type’] == ‘folder’: line = f”{indent}{prefix}📁 {item[‘name’]}/” total_folders += 1 else: line = f”{indent}{prefix}📄 {item[‘name’]}” total_files += 1 f.write(line + “n”) total_items += 1 # 添加统计信息 f.write(“n” + “=” * 70 + “nn”) f.write(f”统计信息:n”) f.write(f”总项目数: {total_items}n”) f.write(f”文件夹数: {total_folders}n”) f.write(f”文件数: {total_files}n”) self.status_label.setText(f”检索完成!找到 {total_items} 个项目,结果已保存到: {output_file}”) QMessageBox.information(self, “完成”, f”检索完成!n” f”总项目数: {total_items}n” f”文件夹数: {total_folders}n” f”文件数: {total_files}n” f”结果已保存到:n{output_file}” ) except Exception as e: self.status_label.setText(“检索出错”) QMessageBox.critical(self, “错误”, f”检索过程中出错: {str(e)}”) def recursive_traverse(self, root_dir, current_dir, results, depth, search_folders, search_files, search_both, file_extensions, ignore_patterns): “””递归遍历目录,保持实际目录结构””” try: # 获取当前目录下的条目 entries = os.listdir(current_dir) except Exception as e: # 跳过无权访问的目录 return # 排序条目 entries.sort(key=lambda s: s.lower()) # 获取当前目录相对于根目录的相对路径 rel_dir = os.path.relpath(current_dir, root_dir) # 如果是根目录,添加根目录项 if current_dir == root_dir: results.append({ ‘name’: os.path.basename(root_dir) or os.path.splitdrive(root_dir)[0], ‘path’: root_dir, ‘depth’: depth, ‘type’: ‘folder’ }) # 处理文件夹 folders = [e for e in entries if os.path.isdir(os.path.join(current_dir, e))] for folder in folders: folder_path = os.path.join(current_dir, folder) rel_path = os.path.relpath(folder_path, root_dir) # 检查是否在屏蔽列表中 if self.is_ignored(rel_path, ignore_patterns): continue # 添加到结果(如果需要检索文件夹) if search_folders or search_both: results.append({ ‘name’: folder, ‘path’: folder_path, ‘depth’: depth + 1, ‘type’: ‘folder’ }) # 递归处理子目录 self.recursive_traverse( root_dir, folder_path, results, depth + 1, search_folders, search_files, search_both, file_extensions, ignore_patterns ) # 处理文件 files = [e for e in entries if os.path.isfile(os.path.join(current_dir, e))] for file in files: file_path = os.path.join(current_dir, file) rel_path = os.path.relpath(file_path, root_dir) # 检查是否在屏蔽列表中 if self.is_ignored(rel_path, ignore_patterns): continue # 检查文件后缀 if (search_files or search_both) and file_extensions: ext = os.path.splitext(file)[1].lower() if ext not in file_extensions: continue # 添加到结果 if search_files or search_both: results.append({ ‘name’: file, ‘path’: file_path, ‘depth’: depth + 1, ‘type’: ‘file’ }) def is_ignored(self, path, patterns): “””检查路径是否与任何屏蔽模式匹配””” for pattern in patterns: if fnmatch.fnmatch(path, pattern): return True if pattern in path: return True return False if __name__ == “__main__”: app = QApplication([]) app.setStyle(“Fusion”) # 设置应用样式 palette = QPalette() palette.setColor(QPalette.Window, QColor(53, 53, 53)) palette.setColor(QPalette.WindowText, Qt.white) palette.setColor(QPalette.Base, QColor(35, 35, 35)) palette.setColor(QPalette.AlternateBase, QColor(53, 53, 53)) palette.setColor(QPalette.ToolTipBase, Qt.white) palette.setColor(QPalette.ToolTipText, Qt.white) palette.setColor(QPalette.Text, Qt.white) palette.setColor(QPalette.Button, QColor(53, 53, 53)) palette.setColor(QPalette.ButtonText, Qt.white) palette.setColor(QPalette.BrightText, Qt.red) palette.setColor(QPalette.Highlight, QColor(142, 45, 197).lighter()) palette.setColor(QPalette.HighlightedText, Qt.black) app.setPalette(palette) window = FileSearchApp() window.show() app.exec_()
  • PyQt5==5.15.9 pip install PyQt5 pyinstaller –onefile -w so.py 到此这篇关于基于Python实现一个目录/文件递归检索工具的文章就介绍到这了,更多相关Python目录/文件递归检索内容请搜索风君子博客以前的文章或继续浏览下面的相关文章希望大家以后多多支持风君子博客! 您可能感兴趣的文章: 使用Python检索数据库sql格式的文件 Python实现检索指定网段内所有的数据库服务器 Python字符串检索方式 Python使用Oracle向量数据库实现文本检索系统 python题解LeetCode303区域和检索示例详解
  • 目录
    • 核心功能
      • 1. 目录结构检索
      • 2. 智能过滤系统
      • 3. 结果输出
    • 特色功能
      • 4. 用户友好界面
      • 5. 配置管理
      • 6. 高效性能
    • 使用场景
      • 附源代码
        • 文件递归检索工具依赖库

          递归扫描 :深度遍历指定目录及其所有子目录

          多种检索模式

          • 仅文件夹模式:只显示目录结构
          • 仅文件模式:只显示文件列表
          • 文件+文件夹模式:完整显示目录树结构(默认模式)

          文件后缀过滤 :支持多后缀过滤(如:.txt; .py; .jpg)

          屏蔽词管理

          • 支持通配符(如 .tmp; backup_ )
          • 可创建和管理多个屏蔽配置
          • 支持导入/导出配置

          • 树形结构展示 :直观显示目录层级关系
          • 可视化标识

          • 统计信息 :自动生成项目数量统计
          • 结果导出 :一键导出为文本文件

          • 直观操作 :清晰的按钮布局和分组
          • 深色主题 :减轻视觉疲劳
          • 实时状态提示 :显示当前操作状态
          • 智能路径建议 :自动生成默认输出路径

          • 配置文件存储 :配置文件保存在程序同目录
          • 多配置支持 :可创建和管理多个屏蔽配置
          • 配置导出 :支持将配置导出为JSON文件

          • 快速扫描 :优化递归算法提高效率
          • 错误处理 :自动跳过无权限访问的目录
          • 排序功能 :文件和文件夹按名称排序

          1. 项目结构分析 :快速查看项目目录结构
          2. 文件系统清理 :识别特定类型的文件(如临时文件)
          3. 文档编制 :生成项目目录树文档
          4. 资产盘点 :统计特定类型的文件数量
          5. 系统维护 :查找分散的配置文件或日志文件

          import os
          import sys
          import re
          import json
          import fnmatch
          from PyQt5.QtWidgets import (
              QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit,
              QPushButton, QRadioButton, QButtonGroup, QGroupBox, QFileDialog, QTextEdit,
              QComboBox, QMessageBox, QCheckBox, QListWidget, QListWidgetItem, QInputDialog
          )
          from PyQt5.QtCore import Qt
          from PyQt5.QtGui import QFont, QPalette, QColor
          
          # 获取脚本所在目录
          if getattr(sys, 'frozen', False):
              # 如果是打包后的可执行文件
              APP_DIR = os.path.dirname(sys.executable)
          else:
              # 如果是脚本文件
              APP_DIR = os.path.dirname(os.path.abspath(__file__))
          
          # 修改配置文件路径为脚本所在目录
          CONFIG_FILE = os.path.join(APP_DIR, "config.json")
          
          class FileSearchApp(QMainWindow):
              def __init__(self):
                  super().__init__()
                  self.setWindowTitle("目录/文件递归检索工具")
                  self.setGeometry(300, 300, 800, 650)
                  
                  # 初始化变量
                  self.ignore_configs = []
                  self.current_ignore_config = {"name": "默认配置", "patterns": []}
                  self.load_config()
                  
                  # 创建UI
                  self.init_ui()
          
          class FileSearchApp(QMainWindow):
              def __init__(self):
                  super().__init__()
                  self.setWindowTitle("目录/文件递归检索工具 V1.0 by:Sunf10wer")
                  self.setGeometry(300, 300, 800, 650)
                  
                  # 初始化变量
                  self.ignore_configs = []
                  self.current_ignore_config = {"name": "默认配置", "patterns": []}
                  self.load_config()
                  
                  # 创建UI
                  self.init_ui()
                  
              def init_ui(self):
                  # 主布局
                  main_widget = QWidget()
                  main_layout = QVBoxLayout()
                  main_widget.setLayout(main_layout)
                  self.setCentralWidget(main_widget)
                  
                  # 设置标题样式
                  title_label = QLabel("目录/文件递归检索工具")
                  title_font = QFont("Arial", 16, QFont.Bold)
                  title_label.setFont(title_font)
                  title_label.setAlignment(Qt.AlignCenter)
                  title_label.setStyleSheet("color: #FFFFFF; padding: 10px;")
                  main_layout.addWidget(title_label)
                  
                  # 目录选择
                  dir_layout = QHBoxLayout()
                  dir_label = QLabel("目标目录:")
                  self.dir_entry = QLineEdit()
                  self.dir_entry.setPlaceholderText("请选择或输入要检索的目录...")
                  browse_button = QPushButton("浏览...")
                  browse_button.clicked.connect(self.browse_directory)
                  
                  dir_layout.addWidget(dir_label)
                  dir_layout.addWidget(self.dir_entry, 4)
                  dir_layout.addWidget(browse_button, 1)
                  main_layout.addLayout(dir_layout)
                  
                  # 输出文件选择
                  output_layout = QHBoxLayout()
                  output_label = QLabel("输出文件:")
                  self.output_entry = QLineEdit()
                  self.output_entry.setPlaceholderText("输出文件名...")
                  output_browse_button = QPushButton("浏览...")
                  output_browse_button.clicked.connect(self.browse_output_file)
                  
                  output_layout.addWidget(output_label)
                  output_layout.addWidget(self.output_entry, 4)
                  output_layout.addWidget(output_browse_button, 1)
                  main_layout.addLayout(output_layout)
                  
                  # 检索类型选择
                  search_type_group = QGroupBox("检索类型")
                  search_layout = QHBoxLayout()
                  
                  self.folder_radio = QRadioButton("仅文件夹")
                  self.file_radio = QRadioButton("仅文件")
                  self.both_radio = QRadioButton("文件和文件夹")
                  self.both_radio.setChecked(True)
                  
                  self.search_type_group = QButtonGroup()
                  self.search_type_group.addButton(self.folder_radio)
                  self.search_type_group.addButton(self.file_radio)
                  self.search_type_group.addButton(self.both_radio)
                  
                  # 文件后缀过滤
                  suffix_layout = QHBoxLayout()
                  suffix_label = QLabel("文件后缀(用分号分隔):")
                  self.suffix_entry = QLineEdit()
                  self.suffix_entry.setPlaceholderText("例如: .txt; .py; .jpg")
                  
                  suffix_layout.addWidget(suffix_label)
                  suffix_layout.addWidget(self.suffix_entry)
                  
                  search_layout.addWidget(self.folder_radio)
                  search_layout.addWidget(self.file_radio)
                  search_layout.addWidget(self.both_radio)
                  search_layout.addStretch()
                  
                  search_type_group.setLayout(search_layout)
                  main_layout.addWidget(search_type_group)
                  main_layout.addLayout(suffix_layout)
                  
                  # 屏蔽词管理
                  ignore_group = QGroupBox("屏蔽词管理")
                  ignore_layout = QVBoxLayout()
                  
                  # 屏蔽词配置选择
                  config_layout = QHBoxLayout()
                  config_label = QLabel("当前配置:")
                  self.config_combo = QComboBox()
                  self.config_combo.setMinimumWidth(150)
                  self.config_combo.currentIndexChanged.connect(self.config_selected)
                  
                  new_config_btn = QPushButton("新建配置")
                  new_config_btn.clicked.connect(self.create_new_config)
                  
                  config_layout.addWidget(config_label)
                  config_layout.addWidget(self.config_combo, 1)
                  config_layout.addWidget(new_config_btn)
                  
                  # 屏蔽词列表
                  ignore_list_layout = QVBoxLayout()
                  list_label = QLabel("屏蔽词列表(支持通配符,如 *.tmp; backup_*)")
                  self.ignore_list = QListWidget()
                  self.ignore_list.setAlternatingRowColors(True)
                  
                  add_btn = QPushButton("添加屏蔽词")
                  add_btn.clicked.connect(self.add_ignore_pattern)
                  remove_btn = QPushButton("移除选中")
                  remove_btn.clicked.connect(self.remove_selected_pattern)
                  
                  list_btn_layout = QHBoxLayout()
                  list_btn_layout.addWidget(add_btn)
                  list_btn_layout.addWidget(remove_btn)
                  
                  ignore_list_layout.addWidget(list_label)
                  ignore_list_layout.addWidget(self.ignore_list)
                  ignore_list_layout.addLayout(list_btn_layout)
                  
                  ignore_layout.addLayout(config_layout)
                  ignore_layout.addLayout(ignore_list_layout)
                  ignore_group.setLayout(ignore_layout)
                  main_layout.addWidget(ignore_group)
                  
                  # 操作按钮
                  button_layout = QHBoxLayout()
                  self.search_btn = QPushButton("开始检索")
                  self.search_btn.setStyleSheet(
                      "background-color: #3498db; color: white; font-weight: bold; padding: 8px;"
                  )
                  self.search_btn.clicked.connect(self.start_search)
                  
                  export_btn = QPushButton("导出配置")
                  export_btn.clicked.connect(self.export_config)
                  
                  button_layout.addStretch()
                  button_layout.addWidget(self.search_btn)
                  button_layout.addWidget(export_btn)
                  button_layout.addStretch()
                  
                  main_layout.addLayout(button_layout)
                  
                  # 状态栏
                  self.status_bar = self.statusBar()
                  self.status_label = QLabel("就绪")
                  self.status_bar.addWidget(self.status_label)
                  
                  # 更新UI
                  self.update_config_combo()
                  self.update_ignore_list()
                  
                  # 连接信号
                  self.dir_entry.textChanged.connect(self.update_output_filename)
                  
              def load_config(self):
                  """从配置文件加载屏蔽词配置"""
                  if os.path.exists(CONFIG_FILE):
                      try:
                          with open(CONFIG_FILE, 'r', encoding='utf-8') as f:
                              self.ignore_configs = json.load(f)
                          # 确保至少有一个默认配置
                          if not any(cfg['name'] == '默认配置' for cfg in self.ignore_configs):
                              self.ignore_configs.insert(0, {"name": "默认配置", "patterns": []})
                      except:
                          self.ignore_configs = [{"name": "默认配置", "patterns": []}]
                  else:
                      self.ignore_configs = [{"name": "默认配置", "patterns": []}]
                  
                  self.current_ignore_config = self.ignore_configs[0]
              
              def save_config(self):
                  """保存屏蔽词配置到文件"""
                  try:
                      with open(CONFIG_FILE, 'w', encoding='utf-8') as f:
                          json.dump(self.ignore_configs, f, ensure_ascii=False, indent=2)
                      return True
                  except Exception as e:
                      QMessageBox.critical(self, "保存错误", f"保存配置时出错: {str(e)}")
                      return False
              
              def update_config_combo(self):
                  """更新配置下拉框"""
                  self.config_combo.clear()
                  for config in self.ignore_configs:
                      self.config_combo.addItem(config['name'])
                  
                  # 选择当前配置
                  current_index = next(
                      (i for i, config in enumerate(self.ignore_configs) 
                       if config['name'] == self.current_ignore_config['name']),
                      0
                  )
                  self.config_combo.setCurrentIndex(current_index)
              
              def update_ignore_list(self):
                  """更新屏蔽词列表"""
                  self.ignore_list.clear()
                  for pattern in self.current_ignore_config['patterns']:
                      self.ignore_list.addItem(pattern)
              
              def config_selected(self, index):
                  """配置选择改变事件"""
                  if 0 <= index < len(self.ignore_configs):
                      self.current_ignore_config = self.ignore_configs[index]
                      self.update_ignore_list()
              
              def create_new_config(self):
                  """创建新的屏蔽词配置"""
                  name, ok = QInputDialog.getText(
                      self, "新建配置", "输入配置名称:", 
                      text=f"配置{len(self.ignore_configs)+1}"
                  )
                  if ok and name:
                      # 检查名称是否已存在
                      if any(cfg['name'] == name for cfg in self.ignore_configs):
                          QMessageBox.warning(self, "名称冲突", f"配置名 '{name}' 已存在!")
                          return
                      
                      new_config = {"name": name, "patterns": []}
                      self.ignore_configs.append(new_config)
                      self.current_ignore_config = new_config
                      self.update_config_combo()
                      self.update_ignore_list()
                      self.save_config()
              
              def add_ignore_pattern(self):
                  """添加屏蔽词"""
                  pattern, ok = QInputDialog.getText(
                      self, "添加屏蔽词", "请输入屏蔽词(支持通配符):"
                  )
                  if ok and pattern:
                      if pattern not in self.current_ignore_config['patterns']:
                          self.current_ignore_config['patterns'].append(pattern)
                          self.update_ignore_list()
                          self.save_config()
              
              def remove_selected_pattern(self):
                  """移除选中的屏蔽词"""
                  selected_items = self.ignore_list.selectedItems()
                  if not selected_items:
                      return
                  
                  for item in selected_items:
                      pattern = item.text()
                      if pattern in self.current_ignore_config['patterns']:
                          self.current_ignore_config['patterns'].remove(pattern)
                  
                  self.update_ignore_list()
                  self.save_config()
              
              def browse_directory(self):
                  """浏览目录"""
                  directory = QFileDialog.getExistingDirectory(self, "选择目录")
                  if directory:
                      self.dir_entry.setText(directory)
                      self.update_output_filename()
              
              def browse_output_file(self):
                  """浏览输出文件"""
                  # 获取默认输出路径
                  default_path = self.output_entry.text() or self.get_default_output_path()
                  
                  # 确保默认路径包含文件名
                  if os.path.isdir(default_path):
                      default_path = os.path.join(default_path, self.get_default_filename())
                  
                  # 打开文件对话框
                  file_path, _ = QFileDialog.getSaveFileName(
                      self, "保存结果", default_path, "文本文件 (*.txt)"
                  )
                  
                  if file_path:
                      # 确保文件扩展名正确
                      if not file_path.endswith('.txt'):
                          file_path += '.txt'
                      self.output_entry.setText(file_path)
              
              def get_default_output_path(self):
                  """获取默认输出目录"""
                  # 优先使用目标目录所在位置
                  if self.dir_entry.text():
                      return os.path.dirname(self.dir_entry.text())
                  
                  # 使用当前工作目录作为备选
                  return os.getcwd()
              
              def update_output_filename(self):
                  """当目录改变时更新默认输出文件名"""
                  # 仅当输出框为空时更新
                  if not self.output_entry.text():
                      self.output_entry.setText(self.get_default_output_path())
              
              def get_default_filename(self):
                  """获取默认文件名"""
                  directory = self.dir_entry.text()
                  if directory:
                      # 获取目录名作为文件名基础
                      base_name = os.path.basename(directory) or "root"
                      return f"{base_name}_检索结果.txt"
                  return "检索结果.txt"
              
              def export_config(self):
                  """导出当前配置"""
                  if not self.current_ignore_config['patterns']:
                      QMessageBox.information(self, "导出配置", "当前配置没有屏蔽词!")
                      return
                  
                  file_path, _ = QFileDialog.getSaveFileName(
                      self, "导出配置", "", "JSON文件 (*.json)"
                  )
                  if file_path:
                      if not file_path.endswith('.json'):
                          file_path += '.json'
                      
                      try:
                          with open(file_path, 'w', encoding='utf-8') as f:
                              json.dump(self.current_ignore_config, f, ensure_ascii=False, indent=2)
                          QMessageBox.information(self, "导出成功", f"配置已导出到:n{file_path}")
                      except Exception as e:
                          QMessageBox.critical(self, "导出错误", f"导出配置时出错: {str(e)}")
              
              def start_search(self):
                  """开始检索"""
                  # 获取输入参数
                  target_dir = self.dir_entry.text().strip()
                  if not target_dir or not os.path.isdir(target_dir):
                      QMessageBox.warning(self, "目录错误", "请选择有效的目标目录!")
                      return
                  
                  # 获取输出文件路径
                  output_file = self.output_entry.text().strip()
                  if not output_file:
                      # 如果没有指定输出文件,使用默认文件名
                      output_file = os.path.join(
                          self.get_default_output_path(),
                          self.get_default_filename()
                      )
                      self.output_entry.setText(output_file)
                  
                  if not output_file.endswith('.txt'):
                      output_file += '.txt'
                  
                  # 确保输出目录存在
                  output_dir = os.path.dirname(output_file)
                  if output_dir and not os.path.exists(output_dir):
                      try:
                          os.makedirs(output_dir)
                      except Exception as e:
                          QMessageBox.critical(self, "目录错误", f"无法创建输出目录: {str(e)}")
                          return
                  
                  # 检查检索类型
                  search_folders = self.folder_radio.isChecked()
                  search_files = self.file_radio.isChecked()
                  search_both = self.both_radio.isChecked()
                  
                  file_extensions = []
                  if search_files or search_both:
                      ext_str = self.suffix_entry.text().strip()
                      if ext_str:
                          file_extensions = [ext.strip().lower() for ext in ext_str.split(';') if ext.strip()]
                  
                  # 获取屏蔽词
                  ignore_patterns = self.current_ignore_config['patterns']
                  
                  # 执行检索
                  self.status_label.setText("正在检索...")
                  QApplication.processEvents()  # 更新UI
                  
                  try:
                      # 获取层级结构的检索结果
                      results = []
                      self.recursive_traverse(
                          target_dir, 
                          target_dir, 
                          results, 
                          0,
                          search_folders, 
                          search_files,
                          search_both,
                          file_extensions, 
                          ignore_patterns
                      )
                      
                      # 写入文件
                      with open(output_file, 'w', encoding='utf-8') as f:
                          # 写入头部信息
                          f.write(f"检索目录: {target_dir}n")
                          
                          if search_folders:
                              f.write(f"检索类型: 仅文件夹n")
                          elif search_files:
                              f.write(f"检索类型: 仅文件n")
                          else:
                              f.write(f"检索类型: 文件和文件夹n")
                          
                          if (search_files or search_both) and file_extensions:
                              f.write(f"文件后缀: {', '.join(file_extensions)}n")
                          if ignore_patterns:
                              f.write(f"屏蔽配置: {self.current_ignore_config['name']}n")
                              f.write(f"屏蔽词: {', '.join(ignore_patterns)}n")
                          
                          f.write("n" + "=" * 70 + "nn")
                          
                          # 写入层级结构
                          total_items = 0
                          total_folders = 0
                          total_files = 0
                          
                          for item in results:
                              # 根据层级深度添加缩进
                              indent = "│   " * (item['depth'] - 1)
                              prefix = "├── " if item['depth'] > 0 else ""
                              
                              # 添加文件夹/文件标识
                              if item['type'] == 'folder':
                                  line = f"{indent}{prefix}📁 {item['name']}/"
                                  total_folders += 1
                              else:
                                  line = f"{indent}{prefix}📄 {item['name']}"
                                  total_files += 1
                              
                              f.write(line + "n")
                              total_items += 1
                          
                          # 添加统计信息
                          f.write("n" + "=" * 70 + "nn")
                          f.write(f"统计信息:n")
                          f.write(f"总项目数: {total_items}n")
                          f.write(f"文件夹数: {total_folders}n")
                          f.write(f"文件数: {total_files}n")
                      
                      self.status_label.setText(f"检索完成!找到 {total_items} 个项目,结果已保存到: {output_file}")
                      QMessageBox.information(self, "完成", 
                          f"检索完成!n"
                          f"总项目数: {total_items}n"
                          f"文件夹数: {total_folders}n"
                          f"文件数: {total_files}n"
                          f"结果已保存到:n{output_file}"
                      )
                  except Exception as e:
                      self.status_label.setText("检索出错")
                      QMessageBox.critical(self, "错误", f"检索过程中出错: {str(e)}")
              
              def recursive_traverse(self, root_dir, current_dir, results, depth, 
                                     search_folders, search_files, search_both,
                                     file_extensions, ignore_patterns):
                  """递归遍历目录,保持实际目录结构"""
                  try:
                      # 获取当前目录下的条目
                      entries = os.listdir(current_dir)
                  except Exception as e:
                      # 跳过无权访问的目录
                      return
                  
                  # 排序条目
                  entries.sort(key=lambda s: s.lower())
                  
                  # 获取当前目录相对于根目录的相对路径
                  rel_dir = os.path.relpath(current_dir, root_dir)
                  
                  # 如果是根目录,添加根目录项
                  if current_dir == root_dir:
                      results.append({
                          'name': os.path.basename(root_dir) or os.path.splitdrive(root_dir)[0],
                          'path': root_dir,
                          'depth': depth,
                          'type': 'folder'
                      })
                  
                  # 处理文件夹
                  folders = [e for e in entries if os.path.isdir(os.path.join(current_dir, e))]
                  for folder in folders:
                      folder_path = os.path.join(current_dir, folder)
                      rel_path = os.path.relpath(folder_path, root_dir)
                      
                      # 检查是否在屏蔽列表中
                      if self.is_ignored(rel_path, ignore_patterns):
                          continue
                      
                      # 添加到结果(如果需要检索文件夹)
                      if search_folders or search_both:
                          results.append({
                              'name': folder,
                              'path': folder_path,
                              'depth': depth + 1,
                              'type': 'folder'
                          })
                      
                      # 递归处理子目录
                      self.recursive_traverse(
                          root_dir, 
                          folder_path, 
                          results, 
                          depth + 1,
                          search_folders, 
                          search_files,
                          search_both,
                          file_extensions, 
                          ignore_patterns
                      )
                  
                  # 处理文件
                  files = [e for e in entries if os.path.isfile(os.path.join(current_dir, e))]
                  for file in files:
                      file_path = os.path.join(current_dir, file)
                      rel_path = os.path.relpath(file_path, root_dir)
                      
                      # 检查是否在屏蔽列表中
                      if self.is_ignored(rel_path, ignore_patterns):
                          continue
                      
                      # 检查文件后缀
                      if (search_files or search_both) and file_extensions:
                          ext = os.path.splitext(file)[1].lower()
                          if ext not in file_extensions:
                              continue
                      
                      # 添加到结果
                      if search_files or search_both:
                          results.append({
                              'name': file,
                              'path': file_path,
                              'depth': depth + 1,
                              'type': 'file'
                          })
              
              def is_ignored(self, path, patterns):
                  """检查路径是否与任何屏蔽模式匹配"""
                  for pattern in patterns:
                      if fnmatch.fnmatch(path, pattern):
                          return True
                      if pattern in path:
                          return True
                  return False
          
          
          if __name__ == "__main__":
              app = QApplication([])
              app.setStyle("Fusion")
              
              # 设置应用样式
              palette = QPalette()
              palette.setColor(QPalette.Window, QColor(53, 53, 53))
              palette.setColor(QPalette.WindowText, Qt.white)
              palette.setColor(QPalette.Base, QColor(35, 35, 35))
              palette.setColor(QPalette.AlternateBase, QColor(53, 53, 53))
              palette.setColor(QPalette.ToolTipBase, Qt.white)
              palette.setColor(QPalette.ToolTipText, Qt.white)
              palette.setColor(QPalette.Text, Qt.white)
              palette.setColor(QPalette.Button, QColor(53, 53, 53))
              palette.setColor(QPalette.ButtonText, Qt.white)
              palette.setColor(QPalette.BrightText, Qt.red)
              palette.setColor(QPalette.Highlight, QColor(142, 45, 197).lighter())
              palette.setColor(QPalette.HighlightedText, Qt.black)
              app.setPalette(palette)
              
              window = FileSearchApp()
              window.show()
              app.exec_()
          
          

          PyQt5==5.15.9

          pip install PyQt5

          pyinstaller --onefile -w so.py
          

          到此这篇关于基于Python实现一个目录/文件递归检索工具的文章就介绍到这了,更多相关Python目录/文件递归检索内容请搜索风君子博客以前的文章或继续浏览下面的相关文章希望大家以后多多支持风君子博客!

          您可能感兴趣的文章:

          • 使用Python检索数据库sql格式的文件
          • Python实现检索指定网段内所有的数据库服务器
          • Python字符串检索方式
          • Python使用Oracle向量数据库实现文本检索系统
          • python题解LeetCode303区域和检索示例详解

          站内搜索