Python處理utf-8 添加和刪除BOM頭

以下代碼只處理了assic和utf8文件。其它文件編碼為保險起見並未加入支持。
參數
exts 需要處理文件的擴展名
folders 需要處理的文件夾及子目錄
處理目錄為當前目錄
運行：
添加bom頭
python proc_bom.py
刪除bom頭
python proc_bom.py -r

運行缺少chardet報錯 

方法1:在線安裝

　　pip install chardet

方法2:離線安裝

https://github.com/chardet/chardet

https://pypi.python.org/pypi/chardet#downloads

去下載chardet包

進入到chardet 3.0.4 的解壓包里，執行：python setup.py install 就可以完成安裝了。

 

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import os;
import sys;
import codecs;
import chardet;

#獲取腳本文件的當前路徑
def cur_file_dir():
     #獲取腳本路徑
     path = sys.path[0]
     #判斷為腳本文件還是py2exe編譯后的文件，如果是腳本文件，則返回的是腳本的目錄，如果是py2exe編譯后的文件，則返回的是編譯后的文件路徑
     if os.path.isdir(path):
         return path
     elif os.path.isfile(path):
         return os.path.dirname(path)
#打印結果


#pip install chardet 安裝相應插件
def procBOM(strPath,curLen, bAdd):
    newcontent = '';
    f = open(strPath, "rb");
    fcontent = f.read();
    f.close();
    printBuffer = strPath[curLen:]
    codeType = chardet.detect(fcontent)["encoding"]  #檢測編碼方式  
     printBuffer = printBuffer + "  "+str(codeType) 

     if codeType.lower().find('utf-8') == -1 and codeType.lower().find('ascii') == -1 :
         #非utf8文件保險起見先退出,並輸出錯誤提示,todo后續再添加其它轉碼到utf8
         print printBuffer + " error OK"
         return
    
    #不需要轉換，已經添加bom頭    

    if bAdd and fcontent[:3] != codecs.BOM_UTF8:
        print  printBuffer+" add bom",
        newcontent = codecs.BOM_UTF8;
        newcontent += fcontent;
    elif not bAdd and fcontent[:3] == codecs.BOM_UTF8:
        newcontent = fcontent[3:];
        print  printBuffer+" del bom",
    else:
        return;
    fnew = open(strPath, "wb+")
    fnew.write(newcontent);
    fnew.close();
    print "done"
    return

if __name__ == "__main__":

    bAdd = True;
    exts = ['.h', '.c', '.cpp'];
    folders = ["GNaviInterface/search","src","tester"]
    bAdd = True;
    if(len(sys.argv) > 1 and sys.argv[1] == '-r'):
        bAdd = False;
    curLen = len(cur_file_dir())
    for folderName in folders:
        folderPath = cur_file_dir()+"/"+folderName+"/"
        #print "procBOM:folder path = "+folderPath+",add = "+str(bAdd)
        for parent,dirnames,filenames in os.walk(folderPath):
            for f in filenames:
                bTargetFile = False;
                for e in exts:
                    if(f.endswith(e)):
                        bTargetFile = True;
                if(bTargetFile):
                    procBOM(os.path.join(parent,f),curLen, bAdd);
#print 'file:%s add:%s' % (os.path.join(parent, f), bAdd);