去除utf-8文件bom頭的py腳本，剛寫的，共享下

t6760915

我寫的有個項目不知道怎么的再上傳后文件全加了bom頭，一個一個修改起來比較麻煩，所以查了下資料寫了個批處理的腳本，共享給大家看看。這個腳本如果在有中文目錄的環(huán)境下執(zhí)行可能會有點編碼問題，不過一般的程序文件好像沒中文目錄，哈哈

1. #coding=utf-8
   
2. 
   
3. '''
   
4. * 去除指定類型文件的bom頭
   
5. * 版權(quán)所有
   
6. * @author      t6760915<t6760915@gmail.com>
   
7. * @version     $Id: trim_bom.py $
   
8. '''
   
9. 
   
10. import os
    
11. import sys
    
12. import codecs
    
13. 
    
14. class TrimBom:
    
15.    
    
16.     basePath = ''
    
17.     fileList = []
    
18.     trimExtList = ['php', 'css', 'js', 'py', 'pl', 'html', 'htm']
    
19. 
    
20.     #去除字符串中的bom部分
    
21.     def trim_bom(self, str):
    
22.         if not str:
    
23.             return ''
    
24.         
    
25.         if len(str) < 3:
    
26.             return ''
    
27.         
    
28.         if str[:3] == codecs.BOM_UTF8:
    
29.             return str[3:]
    
30.         
    
31.         return str
    
32. 
    
33.     #去除某文件的bom頭
    
34.     def trim_file_bom(self, fileName):
    
35.         if not fileName:
    
36.             return False
    
37.         
    
38.         if not os.path.exists(fileName):
    
39.             return False
    
40. 
    
41.         try:
    
42.             fp = open(fileName, 'r')
    
43.             html_sourse = fp.read()
    
44.             fp.close()
    
45.         except:
    
46.             return False
    
47. 
    
48.         html_result = self.trim_bom(html_sourse)
    
49. 
    
50.         if html_result == html_sourse:
    
51.             return False
    
52. 
    
53.         try:
    
54.             fp = open(fileName, 'w')
    
55.             fp.write(html_result)
    
56.             fp.close()
    
57.         except:
    
58.             return False
    
59. 
    
60.         return True
    
61. 
    
62.     #獲取指定類型文件名列表
    
63.     def getFileListByExt(self, path):
    
64.         if not path:
    
65.             return False
    
66. 
    
67.         path = os.path.normpath(path)
    
68. 
    
69.         if not os.path.exists(path):
    
70.             return False
    
71. 
    
72.         if os.path.isfile(path) and path.split('.')[-1] in self.trimExtList:
    
73.             trimFlag = self.trim_file_bom(path)
    
74. 
    
75.             if trimFlag:
    
76.                 print 'process %s success...' % path.replace(self.basePath, '').replace('\\', '/')
    
77.             
    
78.         elif os.path.isdir(path):
    
79.             fileNameList = os.listdir(path)
    
80.             for fileName in fileNameList:
    
81.                 fileName = os.path.normpath('%s/%s' % (path,fileName))
    
82.                 self.getFileListByExt(fileName)
    
83.         
    
84.         return False
    
85. 
    
86.     #運行函數(shù)入口
    
87.     def run(self, path):
    
88.         self.basePath = os.path.normpath(path)
    
89.         self.getFileListByExt(path)
    
90. 
    
91. if __name__ == '__main__':
    
92.     if len(sys.argv) < 2:
    
93.         print 'USEAGE:python %s dirName' % __file__
    
94.         sys.exit(0)
    
95.    
    
96.     tObj = TrimBom()
    
97.     tObj.run(sys.argv[1])

復(fù)制代碼

trim_bom.rar (971 Bytes, 下載次數(shù): 126)

2010-05-28 11:38 上傳

點擊文件名下載附件

2gua

不錯啊，原創(chuàng)內(nèi)容，開源精神，值得贊揚。

luffy.deng

1. 
   
2.         if not fileName:
   
3.             return False
   
4.         
   
5.         if not os.path.exists(fileName):
   
6.             return False
   
7. 
   
8.         try:
   
9.             fp = open(fileName, 'r')
   
10.             html_sourse = fp.read()
    
11.             fp.close()
    
12.         except:
    
13.             return False

復(fù)制代碼

t6760915 發(fā)表于 2010-05-28 11:38

既然做異常處理  前面兩個if 不是多余么。 另外如果子目錄下的文件也要處理就不如用walk了 省得isdir  isfile 的判斷了。

t6760915

既然做異常處理  前面兩個if 不是多余么。 另外如果子目錄下的文件也要處理就不如用walk了 省得isdir  is ...
luffy.deng 發(fā)表于 2010-05-28 13:23

    看了你的建議還是將前面兩個處理去掉的好，學(xué)習(xí)了