python 编写一个程序,要求用户输入一个文件夹路径和一个关键词,程序需要在该文件夹下的所有文件中查找该关键词,并返回包含该关键词的文件名。
时间: 2024-05-10 17:19:22 浏览: 157
python 查找文件夹下所有文件 实现代码
可以使用Python的os和re模块来实现该功能,具体代码如下:
```python
import os
import re
def search_files(keyword, path):
files_with_keyword = []
# 遍历文件夹下的所有文件
for dirpath, dirnames, filenames in os.walk(path):
for filename in filenames:
file_path = os.path.join(dirpath, filename)
# 打开文件并逐行查找关键词
with open(file_path, 'r', encoding='utf-8') as f:
for line in f:
if re.search(keyword, line):
files_with_keyword.append(filename)
break # 如果找到了关键词就不需要再继续查找该文件了
return files_with_keyword
# 测试代码
path = input("请输入文件夹路径:")
keyword = input("请输入关键词:")
result = search_files(keyword, path)
if not result:
print("未找到包含关键词的文件")
else:
print("包含关键词的文件名如下:")
for filename in result:
print(filename)
```
其中,os.walk()函数可以递归遍历文件夹下的所有文件和子文件夹,re.search()函数用于在一行文本中查找关键词。程序会返回所有包含关键词的文件名。
阅读全文