1. #coding:utf-8 
2. import os 
3. #-------代碼段一 
4. #獲取當(dāng)前文件夾 
5. filePath = os.getcwd() 
6. #獲取當(dāng)前文件列表 
7. fileNameList = os.listdir(filePath) 
8. fileDirList = [] 
9. #獲取文件路徑列表 
10. for fileName in fileNameList: 
11.   fileDirList.append(os.path.join(filePath, fileName)) 
12. #--------代碼段二 
13. f = open('code.txt', 'w') 
14. f.write("開始寫文件/n") 
15. f.close() 
16. for code in fileDirList: 
17.   f = open('code.txt', 'a') 
18.   split = os.path.splitext(code) 
19.   if(split[1] == '.h' or split[1] == '.cpp'): 
20.     fz = open(code, 'r') 
21.     string = "源代碼文件" + code +"代碼:/n" 
22.     f.write(string) 
23.     content = fz.read() 
24.     print "文件 %s 讀寫成功" % code 
25.     f.write(content) 
26.     fz.close() 
27.   f.close() 
28. print "讀寫成功"