合并目录中的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') + '\r\n'
                 buf[index] = '第' + str(loop) + '节 ' +line
                 loop=loop+1
             else:
                 #buf[index] = line.decode('gb2312').encode('utf8') + '\r\n'
                 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\n')
                     r.write(b + '\r\n')
                     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'