引子

近日，单位应上级要求清理一批文件，关键字这里就不写了。所有职工都要进行清理，通过口头转述自行清理，但是领导不放心要求每个人的电脑再查看一看，是否清理干净。如果通知windows内置的搜索，输入关键是都是一大片，非常麻烦。又因学习了一段时间的python，所以起了用python遍历全盘查找文件的心思。

完整代码，高手勿喷

  1. import os
  2. import subprocess
  3. import sys
  4. import time
  6. def searth(path,keys):
  7.        "对path中的关键字进行查找"
  8.        searth_ls=[]
  9.        error=[]
  10.        try:
  11.               for root, dirnames,files in os.walk(path,topdown=False):
  12.                      if root.count(r"C:\Windows")>0:
  13.                             continue
  14.                      print(root)
  15.                      for name in files:
  16.                             for key in keys:                                 
  17.                                    if key in name:
  18.                                           # print(os.path.join(root,name))
  19.                                           searth_ls.append(os.path.join(root,name))
  21.        except Exception as e:
  22.               print("error:{}".format(e))
  23.               error.append(e)
  25.        return searth_ls, error                       
  27. def getdisk():
  28.        "获取盘符"
  29.        disk_ls=[]
  30.        for i in range(65,91):
  31.               vol = chr(i) + ':\\'
  32.               if os.path.isdir(vol):
  33.                      disk_ls.append(vol)      
  34.        return disk_ls
  36. def wirte_file(name,ls):
  37.        "把ls列表写入到name文件中"
  38.        f=open(name,'w',encoding='utf-8')
  39.        f.write("\n".join(ls))
  40.        f.close()
  42. def disproce_findall(find_all,start_time,end_time):
  43.        "对找到的数据进行处理" 
  44.        wirte_file('search_ls.txt',find_all)
  45.        print('*'*50)
  46.        print("查找共用了{}分{:2f}秒。".format((end_time-start_time)//60,(end_time-start_time)%60))
  47.        print(f"找到文件并已存入到search_ls.txt中：")
  48.        for i in find_all:
  49.               print(i)
  50.        print('*'*50)
  51.        sure=input("是否删除文件(y/n)?")
  52.        if sure =="y":
  53.               for i in find_all: 
  54.                      try:                           
  55.                             os.remove(i)                                   
  56.                             print("删除:{}".format(i))
  57.                      except:
  58.                             pass
  59. def main():
  60.        "程序主函数"
  61.        keystr=['**来','**军','**才']  # 关键字不止这几个
  62.        falg=True
  63.        while falg:
  64.               start_time=time.time()
  65.               disk_ls=getdisk()
  66.               print(f'找到磁盘：{disk_ls}')
  67.               disk_all_search=input('是否进行全盘搜索(y)？')
  68.               find_all=[]
  69.               # 全盘检索
  70.               if disk_all_search=='y':
  71.                      for disk in disk_ls:
  72.                             searth_ls,error_ls=searth(disk,keystr)
  73.                             find_all=find_all+searth_ls
  74.               # 独立磁盘或目录检查
  75.               else:
  76.                      start_disk=input("请输入你要检查的磁盘(q退出)：")
  77.                      if start_disk=="q":
  78.                             falg=False
  79.                      searth_ls,error_ls=searth(start_disk,keystr)
  80.                      find_all=searth_ls
  81.               end_time=time.time()
  82.               # 对找到的数据进行处理
  83.               if len(find_all)>0:
  84.                      disproce_findall(find_all,start_time,end_time) 
  85.               else:
  86.                      print('*'*50)
  87.                      print(f"没有找到相关文件")
  89. if __name__=="__main__":
  90.        main()