Python 脚本 批量删除目录下文件的特定内容

# -*- coding: gbk -*-
#/bin/python
__author__ = 'yli108'

import os,sys,time,codecs
reload(sys) 
sys.setdefaultencoding("gbk")

READ_CODEC = 'gbk'
HomeDir = os.path.split(os.path.abspath(sys.argv[0]))[0]
SP_DIR = 'D:\\Git Surepay\\SUREPAY-CORE\\sp'
SP_OUT_DIR = 'D:\\Git Surepay\\SUREPAY-CORE\\sp1'

def getfilelist(dir):
    '''list all files under the dir'''
    filelist = []
    for dirpath,dirname,filenames in os.walk(dir):
        for name in filenames:
            if name.strip().endswith('.tcl'):
                tcl = os.path.join(dirpath, name)
                print tcl
                filelist.append(tcl)
    return filelist

def frd(filepath):
    ''' Get file context, 
        and rename it to name.bak
        and generate buffer.
     '''
    Dir, filename = os.path.split(filepath)
    file_bk = Dir + '\\' + filename.strip('.tcl')+'_tcl.bak'
    file_in = filepath
    count =0
    buffer = []
    if not os.path.isfile(file_in):
        print "Can't find "+file_in
        return 0,False,buffer
    else:
        with codecs.open(file_in, 'r', encoding = READ_CODEC) as tcl:
            flag = False
            Change_flag = False
            for line in tcl:
                count = count + 1
                if line.strip().startswith(r'%% if {$R27_ONLY == "FALSE"} {'):
                    flag = True
                    Change_flag = True
                    count = 1
                elif flag and line.strip().startswith(r'%%'):
                    count = count + line.strip().count('{') - line.strip().count('}')
                    if count == 0:
                        flag = False
                        line = ''
                if not flag:
                    buffer.append(line.strip('\n'))
        tcl.close()
        #os.remove(file_in)
        if Change_flag:
            pass
            #os.rename(file_in,file_bk)
        else:
            buffer = []
        return 1,Change_flag,buffer

def fwt(filepath,buffer):
    '''        
        write buffer to newfile, newfile  use old filename.
        for text that between 
            "%% R27_ONLY == FALSE {} {
                ...
             %% }
        will not be included to new file.
    '''
    try:
        fout = open(filepath+'new','w')
        fout.write(''.join(buffer))
        fout.close()
        return 1
    except:
        return 0

if __name__ == '__main__':

    '''Usage: delete_code.py   No parameter'''

    file_done = HomeDir + '\\' + 'done_list.txt'
    filelist = getfilelist(SP_DIR)
    
    # open record file to see done files.
    fd = open(file_done,'r+')
    fd_list = fd.readlines()

    for f in filelist[56:57]:
        if f not in fd_list:
            code,c_flag,buffer = frd(f)
            if code and c_flag :
                if fwt(f,buffer):
                    print "Task Success!     " + f
                    fd.write(f+'\n')
                else:
                    print "Write File Fail!!     " + f
            elif code and not c_flag:
                print "TCL File no need change !    " + f
            else:
                print "Task Fail!!    " + f  
    fd.close()