刪除指定格式的文件

gjfer

嘗試使用remove()，一直不知道怎么能夠使用通配符，似乎也沒找到說明。于是就自己弄了一個簡單的小函數(shù)。默認刪除當前目錄下的所有指定類型的文件，包括子目錄。
前面有一小段廢代碼，懶得刪除了。因為我這邊本來是用來代替makefile里面clean功能的。帶了一個自己編譯環(huán)境下的目錄。后來刪除其他文件的時候，就改成了現(xiàn)在這樣子。
我知道有簡單的方法實現(xiàn)這個功能，比如批處理。但是我就是想用一下新學習的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!"

復制代碼