python 指定目录位置，查找指定扩展名的文件是否有指定的关键字

import os

def search_in_file(file_path, search_term):
    """检查文件中是否包含指定的搜索关键字"""
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            content = file.read()
            if search_term in content:
                print(f"Found '{search_term}' in {file_path}")
    except Exception as e:
        print(f"Could not read {file_path}: {e}")

def find_files_with_extensions(root_dir, extensions):
    """递归查找指定目录及子目录下符合扩展名的文件"""
    for root, dirs, files in os.walk(root_dir):
        for file in files:
            if any(file.endswith(ext) for ext in extensions):
                file_path = os.path.join(root, file)
                search_in_file(file_path, 'imvect') #imvect

if __name__ == "__main__":
    # 指定要搜索的根目录
    directory_to_search = './mp_lists'  # 替换为实际路径
    
    # 要搜索的扩展名
    extensions = ['.h', '.cpp', '.hpp']
    
    # 调用函数，查找并搜索文件
    find_files_with_extensions(directory_to_search, extensions)
 

import os

def search_in_file(file_path, search_term):
    """检查文件中是否包含指定的搜索关键字"""
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            content = file.read()
            if search_term in content:
                print(f"Found '{search_term}' in {file_path}")
    except Exception as e:
        print(f"Could not read {file_path}: {e}")

def find_files_with_extensions(root_dir, extensions):
    """递归查找指定目录及子目录下符合扩展名的文件"""
    for root, dirs, files in os.walk(root_dir):
        for file in files:
            if any(file.endswith(ext) for ext in extensions):
                file_path = os.path.join(root, file)
                search_in_file(file_path, 'imvect') #imvect

if __name__ == "__main__":
    # 指定要搜索的根目录
    directory_to_search = './mp_lists'  # 替换为实际路径
    
    # 要搜索的扩展名
    extensions = ['.h', '.cpp', '.hpp']
    
    # 调用函数，查找并搜索文件
    find_files_with_extensions(directory_to_search, extensions)