去除utf-8文件bom头的py脚本，刚写的，共享下

t6760915

我写的有个项目不知道怎么的再上传后文件全加了bom头，一个一个修改起来比较麻烦，所以查了下资料写了个批处理的脚本，共享给大家看看。这个脚本如果在有中文目录的环境下执行可能会有点编码问题，不过一般的程序文件好像没中文目录，哈哈

1. #coding=utf-8
   
2. 
   
3. '''
   
4. * 去除指定类型文件的bom头
   
5. * 版权所有
   
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.     #运行函数入口
    
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])

复制代码

trim_bom.rar (971 Bytes, 下载次数: 126)

2010-05-28 11:38 上传

点击文件名下载附件

2gua

不错啊，原创内容，开源精神，值得赞扬。

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

复制代码

t6760915 发表于 2010-05-28 11:38

既然做异常处理  前面两个if 不是多余么。 另外如果子目录下的文件也要处理就不如用walk了 省得isdir  isfile 的判断了。

t6760915

既然做异常处理  前面两个if 不是多余么。 另外如果子目录下的文件也要处理就不如用walk了 省得isdir  is ...
luffy.deng 发表于 2010-05-28 13:23

    看了你的建议还是将前面两个处理去掉的好，学习了