合并目录中的txt文件

txt 文件 合并 并 按章节 分隔

# -*- coding: utf8 -*-
import os

result_name = 'result.txt'

def resplit_txt_in_folder(folder_name):
    for root,dirs,files in os.walk(folder_name):
        for file_ in files:
            if -1 != file_.find('.txt'):
                resplit_txt_with_name(os.path.dirname(folder_name) + '/' +file_)

def resplit_txt_with_name(txt_name):
    '''
    resplit the txt file, replace prefix with '章节'
    '''
    if True != os.path.exists(txt_name):
        print 'resplit txt failed: file do not exist:' + txt_name
        return
    try:
        print txt_name
        handler = open(txt_name,'r')
        result = open(result_name,'a')
        buf = handler.readlines()
        loop = 1
        for index,line in enumerate(buf):
            if -1 != line.find('@#'):
                #buf[index] = '第' + str(loop) + '节 ' +line.decode('gb2312').encode('utf8') + '
'
                buf[index] = '第' + str(loop) + '节 ' +line
                loop=loop+1
            else:
                #buf[index] = line.decode('gb2312').encode('utf8') + '
'
                pass
        result.writelines(buf)
        result.close()

    finally:
        handler.close()
        print 'resplit done'

def merge_txt_in_folder(folder_name):
    '''
    merge all of the text file into one
    '''

    r = open(result_name,'a')
    loop = 1

    for root,dirs,files in os.walk(folder_name):
        for file_ in files:
            if -1 != file_.find('.txt'):
                try:
                    print file_
                    handler = open(file_,'r')
                    b = handler.read()
                    r.write('第' + str(loop) + '章 ' + file_.split('.')[0].decode('gb2312').encode('utf8') +'
') 
                    r.write(b + '
')
                    loop = loop+1
                finally:
                    handler.close()

        r.close()

if __name__ == '__main__':
    path_name = raw_input("input the path :")
    print 'Start'
    if '' == path_name:
        #merge_txt_in_folder('.')
        resplit_txt_in_folder('.')
    else:
        if True == os.path.exists(path_name):
            #merge_txt_in_folder(path_name)
            resplit_txt_in_folder(path_name)
        else:
            print 'path do not exist:',path_name
    print 'End'