python核心编程学习笔记-2016-08-03-01-习题9-9

习题9-9

一开始打算导入各个模块，再利用__doc__属性，但是感觉不是太好。浏览了网上的一些答案，发现他们都是利用打开文件，直接遍历文件内容，查找两个三引号"""之间内容的思路，因此参考了这一思路，写了代码。

#-*-coding: utf-8-*-

import os

path = r'c:\python27\Lib'
os.chdir(path)
files = []
for i in os.listdir(path):
    if i[-3:] == '.py':
        files.append(i)

have_doc_file = {}
none_doc_file = []

for file in files:
    f = open(file, 'r')
    has_doc = False
    doc = []
    for eachLine in f:
        if eachLine[:3] == '"""' and not has_doc: # 以"""开头的行，首先确定有文档字符串
            has_doc = True
        if has_doc:
            doc.append(eachLine)
        if eachLine[-4:-1] == '"""' and has_doc: # 结尾字符串是"""\n，不要漏掉\n
            break
    f.close()
    if has_doc: 
        have_doc_file[file] = ''.join(doc)
    else:
        none_doc_file.append(file)

f_has_doc_obj = open(r'e:\Learning_python\coreprogramming\has_doc.txt', 'w')
for k, v in have_doc_file.iteritems():
    f_has_doc_obj.write(k + '\n' + v + '\n')
f_has_doc_obj.close()

f_none_doc_obj = open(r'e:\Learning_python\coreprogramming\none_doc.txt', 'w')
for i in none_doc_file:
    f_none_doc_obj.write(i + '\n')
f_none_doc_obj.close()