python处理代码文档

一段python代码，把某个路径下的代码文件复制到一个文件中，方便编写代码文档。

代码的样式可以自己控制（换行，空行）

2.x版本

import os, sys
path = r'D:\\onion\\code\\java\\'
fns = [os.path.join(root, fn) for root, dirs, files in os.walk(path) for fn in files]
outfile = open('D:\\onion\\code\\code5.txt', 'wb')
for f in fns:
    # print(f)  
    f1 = open(f, "r")  
    lines = f1.readlines() 
    for line in lines:        
         if line.split():
             #sys.stdout.write(line)
             outfile.write(line)           
outfile.close()

print(len(fns))

**

python3.x 版本

**

# coding:utf-8
import os
num = input("请输入一个整数：")
num=int(num);
path = r'D:\\data\\code\\vue\\'
fns = [os.path.join(root, fn) for root, dirs, files in os.walk(path) for fn in files]
outfile = open('D:\\data\\code\\code.txt', 'wb')
linum = 1;
for f in fns:
    print("文件名称：" + f)
    f1 = open(f, "r", encoding="utf-8")
    lines = f1.readlines()
    for line in lines:
        linum=linum+1;
        print (linum)
        if(linum>=num):
            break
        if not len(line) or line == '\n':
            continue
        outfile.write(line.encode())
    if (linum >= num):
        break

outfile.close()

print("文件总数：" + len(fns))