#coding:utf-8
import os
#-------代码段一
#获取当前文件夹
filePath = os.getcwd()
#获取当前文件列表
fileNameList = os.listdir(filePath)
fileDirList = []
#获取文件路径列表
for fileName in fileNameList:
fileDirList.append(os.path.join(filePath, fileName))
#--------代码段二
f = open('code.txt', 'w')
f.write("开始写文件n")
f.close()
for code in fileDirList:
f = open('code.txt', 'a')
split = os.path.splitext(code)
if(split[1] == '.h' or split[1] == '.cpp'):
fz = open(code, 'r')
string = "源代码文件" + code +"代码:n"
f.write(string)
content = fz.read()
print "文件 %s 读写成功" % code
f.write(content)
fz.close()
f.close()
print "读写成功"