用Python合并文本文件

初入某个项目,一般都要看代码。有时候,想把代码文件打印下来看,不过一般代码文件数量都在两位数或更多,逐一打开、打印,确实太耗费精力了,此外,也会出现某个代码文件打印到纸上只占了一两行的情况,很浪费纸。如果可以合并到一个文本文件里面上面这些问题就解决。于是用Python写了个合并代码文件的小程序,功能还很单调,实用性不高,暂时只能对某个目录的文本文件进行合并,没有一些精细的判断和处理,留作以后细化。做好后,应该有一定的使用价值。

废话少说,上代码:

import os

def hebing(fdir,outfile):
	file_list = os.listdir(fdir)
	file_to_write = file(outfile,'w')
	for f in file_list:
		file_to_read = file(fdir+str(f),'r')
		
		file_to_write.write('\r\n/*------------')
		file_to_write.write(str(f))
		file_to_write.write('------------*/\r\n')
		file_to_write.write('\r\n')
		
		while True:
			line = file_to_read.readline();
			if len(line) == 0:
				break
			else:
				file_to_write.write(line)
		file_to_read.close()
	
	file_to_write.close()

if __name__ == '__main__':
	hebing('./code/','hebing.txt') #fdir must end with '/'

改进计划(TODO):
1 增加目录深层遍历的功能,能合并多级目录的所有文本文件;
2 增加一些错误的检测,比如文件类型、编码等;
3 格式化输出功能增强;
4 使用pypdf库,生成pdf文件; 

posted @ 2011-11-18 11:24  tt-0411  阅读(4255)  评论(0编辑  收藏  举报