Python合并指定目录下的文件

                                             Python合并指定目录下的文件

 

  该脚本用于合并指定的目录下的文件。

#!/usr/bin/env python
#coding:utf-8
'''
export_gongyi.py

'''
__author__ = 'Gran'

import os
import sys
import glob

def merge_txt_file(dirs,outfile):
        if not os.path.isdir(dirs):
                print('\nThe path %s you input is not a real path,please input again\n'%dirs)
                return False
        if os.path.exists(outfile):
                print('Delete files %s\n'%outfile)
                os.remove(outfile)

        outputfile = open(outfile,'a')
        i = 0
        for txtfiles in glob.glob(os.path.join(dirs,"*.txt")):
                i = i+1
                print('Merge file: %s'%txtfiles)
                inputfile = open(txtfiles,'rb')
                for line in inputfile:
                        outputfile.write(line)
                inputfile.close()

        print('Merge total files: %i\n'%i)
        outputfile.close()
        return True

if __name__ == "__main__":
        if len(sys.argv) < 3:
                print('-----------------------------------------\n')
                print("Usage: %s dir outputFileName\n" %sys.argv[0])
                sys.exit()
        merge_txt_file(sys.argv[1],sys.argv[2])