删除指定格式的文件

gjfer

尝试使用remove()，一直不知道怎么能够使用通配符，似乎也没找到说明。于是就自己弄了一个简单的小函数。默认删除当前目录下的所有指定类型的文件，包括子目录。
前面有一小段废代码，懒得删除了。因为我这边本来是用来代替makefile里面clean功能的。带了一个自己编译环境下的目录。后来删除其他文件的时候，就改成了现在这样子。
我知道有简单的方法实现这个功能，比如批处理。但是我就是想用一下新学习的Python。
clean_all.py

1. #!/usr/bin/python
   
2. 
   
3. """
   
4. 2015.07.02
   
5. Clean all the genrated files during compile.             Edited by Grey
   
6. """
   
7. 
   
8. import os,re
   
9. 
   
10. def CleanAll(file_ext):
    
11.     # remove all files in bin
    
12.     for root,dirs,files in os.walk('./bin'):
    
13.         for each_file in files:
    
14.             try:
    
15.                 os.chdir('./bin')
    
16.                 os.remove(each_file)
    
17.                 os.chdir('../')
    
18.             except:
    
19.                 pass
    
20.     # remove all object files in source foulder
    
21.     for root,dirs,files in os.walk('./'):
    
22.         pwd = os.getcwd()
    
23.         os.chdir(root)
    
24.         object_file_list = [f for f in os.listdir('.') if f.endswith(file_ext)]
    
25.         for file in object_file_list:
    
26.             os.remove(file)
    
27.         os.chdir(pwd)
    
28. 
    
29. CleanAll('.o')
    
30. CleanAll('.rar')
    
31. CleanAll('.xls')
    
32. CleanAll('.xlsx')
    
33. CleanAll('.tmp')
    
34. CleanAll('.txt')
    
35. CleanAll('.lnk')
    
36. CleanAll('.doc')
    
37. CleanAll('.docx')
    
38. CleanAll('.ppt')
    
39. CleanAll('.pptx')
    
40. CleanAll('.bak')
    
41. CleanAll('.c')
    
42. CleanAll('.h')
    
43. CleanAll('.pdf')
    
44. print "All the files and all the files in 'bin' have been removed!"

复制代码