【每日一坑 6】 查找文件内容

jodie

1. import os
   
2. import chardet
   
3. 
   
4. def match(search_path,match_text):
   
5.     result_list = []
   
6. 
   
7.     for root, dirs, files in os.walk(search_path):
   
8.         # 遍历所有的文件
   
9.         for file in files:
   
10.             # 判断文件名是否以“.txt”结尾
    
11.             if file.endswith(".txt"):
    
12. 
    
13.                 # 获取文件的绝对路径
    
14.                 file_path = os.path.join(root,file)
    
15.                 # print(file_path)
    
16. 
    
17.                 # 判断文件的编码格式
    
18.                 with open(file_path, "rb") as obj1:
    
19.                     text_b = obj1.read()
    
20.                     text_encoding = chardet.detect(text_b)["encoding"]
    
21.                     # print(text_encoding)
    
22. 
    
23.                 # 用指定的编码格式打开文件，判断文本参数是否存在于文件内容
    
24.                 with open(file_path, "r", encoding=text_encoding) as obj2:
    
25.                     text = obj2.read()
    
26.                     if match_text in text:
    
27.                         result_list.append(file_path)
    
28. 
    
29.     return result_list
    
30. 
    
31. 
    
32. if __name__ == "__main__":
    
33. 
    
34.     # 提示用户输入路径
    
35.     while True:
    
36.         search_path = input("请输入搜索路径： ").strip()
    
37.         # 判断路径是否存在
    
38.         if os.path.exists(search_path):
    
39.             break
    
40.         print("路径不存在！")
    
41.     # 提示用户输入匹配的文字
    
42.     match_text = input("请输入匹配的文字： ")
    
43.     file_list = match(search_path,match_text)
    
44.     # 展示匹配到的文件
    
45.     print("匹配到的文件如下： ")
    
46.     for one in file_list:
    
47.         print(one)

复制代码