在所有子目录文件的文本中搜索字符串然后打印到创建的文件的脚本

Question

我是一个脚本菜鸟，我被这个问题困住了。

我希望代码能做一些事情：

1. 要求用户输入要搜索的字符串。
2. 遍历给定文件路径的子目录。
3. 打开具有所列扩展名类型之一的文件。
4. 打开文件并搜索用户输入的字符串。
5. 将查询结果打印到文本文件中。

代码似乎需要一些时间才能运行，但什么也没有出现。

import os.path

# Ask the user to enter string to search
search_str = input("Keyword or phrase:\n")

# Store file names for later printing 
file_names = []

# Path to search 
path = os.path.dirname(os.path.realpath(__file__))

# Acceptable file extensions
extensions = {".xlsx", ".txt", ".pdf", ".doc", ".docx", ".mb", ".xlsm", ".xltx", ".xltm"}

# Create file to store search results
search_files = open('search.txt', 'w')
search_files.write(f'I searched for "{search_str}" in your files.\n\nHere is what I found:\n\n')


# Program to search files for keyword
def search_all_files_by_keyword(path):

    # Store file count number
    file_count = 0

    for root, dirs, files in os.walk(path):

        for file in files:

            try:

                # Apply file type filter, search for acceptable ext in extension
                ext = os.path.splitext(file)
                if ext in extensions:

                    # Define file pathway
                    file_path = os.path.join(root, file)

                    # Open file for reading
                    with open(file, 'r') as f:

                        # Read file and search for keyword or phrase
                        if search_str in f.read():

                            # Add file path to file_names and increase file_count, then close file
                            file_names.append(file_path)
                            file_count += 1
                            f.close()

                        # If keyword or phrase is not found, do nothing and close file
                        else:
                            f.close()

            except:
                pass

    # Print search results to file
    if file_count >= 1:
        search_files.write(f"{file_names}\n")
    else:
        search_files.write(f'No results found for "{search_str}".')

# Run program 
search_all_files_by_keyword(path)

Answer 1

尝试路径模块以这种方式搜索所有文件夹/子文件夹：

import re
from pathlib import Path

# Ask the user to enter string to search
search_str = input("Keyword or phrase:\n")

# Store file names for later printing
file_names = []

# Path to search
path = Path("path/to/directory") # Replace with your actual file path

# Acceptable file extensions
extensions = {".xlsx", ".txt", ".pdf", ".doc", ".docx", ".mb"]

# to search for string in a file
def search_in_file(file_path, search_str):
    with open(file_path, 'r', encoding='utf-8') as f:
        file_content = f.read()
        matches = re.findall(search_str, file_content)
        if matches:
            return matches
        else:
            return None

# Iterate through the sub-directories 
for file_path in path.glob("**/*"):
    if file_path.suffix in extensions:
        matches = search_in_file(file_path, search_str)
        
        if matches:
            file_names.append(str(file_path))
            
            with open("results.txt", "a", encoding='utf-8') as results_file:
                results_file.write(f"{file_path}\n")
                results_file.write(f"{matches}\n\n")