#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 "读写成功"