用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文件;
欢迎转载,同时请注明出处