在 Python 中编写一个简单的文件搜索工具

在这篇博客中，我们将介绍如何使用 Python 编写一个简单的文件搜索工具。这个工具允许用户输入目录路径和文件名，并在指定目录及其子目录中搜索匹配的文件。我们将逐步讲解代码的实现，并展示如何运行这个工具。

环境准备

在开始之前，请确保你已经安装了以下工具和库：

• Python：确保你的系统上已经安装了 Python。

步骤

第一步：导入必要的模块

首先，我们需要导入 os 模块，该模块提供了与操作系统进行交互的功能，例如遍历目录结构。

import os

第二步：定义选择目录的函数

接下来，我们定义一个函数 select_directory，该函数使用 input 函数从用户获取要搜索的目录路径。

def select_directory():
    directory = input("请输入要搜索的目录路径: ")
    return directory

第三步：定义搜索文件的函数

然后，我们定义一个函数 search_files，该函数接受目录路径和搜索词作为参数，并在指定目录及其子目录中搜索匹配的文件。搜索结果将存储在一个列表中并返回。

def search_files(directory, search_term):
    results = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if search_term in file:
                results.append(os.path.join(root, file))
    return results

第四步：主程序

在主程序中，我们首先调用 select_directory 函数获取用户输入的目录路径，然后使用 input 函数获取要搜索的文件名。接着，我们调用 search_files 函数进行搜索，并根据搜索结果输出相应的信息。

if __name__ == "__main__":
    directory = select_directory()
    search_term = input("请输入要搜索的文件名: ")