python快速文件查询

jackson_cu

大家好, 接触python 时间不久, 现在想实现一个功能如下, 不知道怎么样效率高?

因为服务器上有很多的zip, rar 压缩文件, 我的主要想法是遍历各个目录, 找到所有zip, rar 的文件, 找到这些文件后, 然后再查看当前目录下有没有与zip/rar 文件同名的目录, 有的话, 就将zip/rar 文件删除.

自己大致的想法是用os.walk 来做, 但估计会很慢.

jackson_cu

试着做了一个...

1. 
   
2. import os
   
3. sep = os.path.sep
   
4. extsep = os.path.extsep
   
5. 
   
6. def RemoveCompressFiles(dstFolder):
   
7.     for roots,paths,files in os.walk(dstFolder):
   
8.         if len(files) > 0:
   
9.             #print roots,paths
   
10.             if len(paths)!=0:
    
11.                 for subpaths in paths:
    
12.                     #print subpaths
    
13.                     for subfiles in files:
    
14.                         file_name = os.path.splitext(subfiles)
    
15.                         file_prefix = file_name[0]
    
16.                         file_ext = file_name[1]
    
17.                         #print file_prefix,file_ext
    
18.                         if file_ext in ['.zip','.rar','.7z','.log']:
    
19.                             if subpaths == file_prefix:
    
20.                                 zipfile=os.path.join(roots,subfiles)
    
21.                                 file_size = os.path.getsize(zipfile)/1024/1024
    
22.                                 print zipfile,">",file_size,"MB"
    
23. 
    
24. 
    
25. if __name__ == '__main__':
    
26.     RemoveCompressFiles('c:\intel\logs')
    

复制代码

ssfjhh

1. import os
   
2. COMFILES = (".rar", ".zip", ".7z", ".log")
   
3. def RemoveCompressFiles(path):
   
4.     os.chdir(path)
   
5.     for r, dlst, flst in os.walk('.'):
   
6.         os.chdir(r)
   
7.         for f in flst:
   
8.             file_prefix, file_ext = os.path.splitext(f)
   
9. 
   
10.             if file_ext in COMFILES and file_prefix in dlst:
    
11.                 os.remove(f)
    
12.     os.chdir(path)
    
13. 
    
14. if __name__ == "__main__":
    
15.     RemoveCompressFiles(your path here)
    

复制代码

简陋版的，没有完全按照你的程序作同样的输出。

jackson_cu

回复 3# ssfjhh

兄台的代码确实精简了很多, 佩服!

我在电脑上跑了一下, 但没有输出, 不知道哪里出错了.

1. #!/usr/bin/env python
   
2. import os
   
3. COMFILES = (".rar", ".zip", ".7z", ".log")
   
4. def RemoveCompressFiles(path):
   
5.     os.chdir(path)
   
6.     for r, dlst, flst in os.walk('.'):
   
7.         os.chdir(r)
   
8.         for f in flst:
   
9.             file_prefix, file_ext = os.path.splitext(f)
   
10. 
    
11.             if file_ext in COMFILES and file_prefix in dlst:
    
12.                 print f
    
13.     os.chdir(path)
    
14. 
    
15. if __name__ == "__main__":
    
16.     RemoveCompressFiles('/data')
    

复制代码

jackson_cu

回复 4# jackson_cu [/b

在windows 上试了一下, 有输出了!

谢谢啦.

   

HH106

回复 3# ssfjhh

1. if file_ext in COMFILES and file_prefix in dlst:

复制代码

file_prefix in dlst红色部分是否有误？



   

ssfjhh

回复 6# HH106


    没看出有什么错误，你给点指示？

HH106

回复 7# ssfjhh
文件名应该不在目录列表元组中