python每日一练：0007题

第 0007 题： 有个目录，里面是你自己写过的程序，统计一下你写过多少行代码。包括空行和注释，但是要分别列出来。

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

import os

def countCode(path):

    if not os.path.exists(path):

        print('路径不存在！')

    else:

        fileNameList = os.listdir(path)

        for fileName in fileNameList:

            nbspCount = 0

            commentCount = 0

            codeCount = 0

            multiCommentEndFlag = 0

            if ('.py' in fileName) and (fileName[-3:] == '.py'):

                with open(path+'\\'+fileName,encoding='utf-8') as f:

                    code = f.readlines()

                    for i in code:

                        codeCount += 1

                        i = i.strip()

                        if multiCommentEndFlag == 1:   #多行注释中内容

                            commentCount += 1

                            if len(i) >= 3 and (i[-3:] == '\'\'\'' or i[-3:] == '\"\"\"'): #多行注释结尾

                                multiCommentEndFlag = 0

                        elif len(i) >= 3 and (i[0:3] == '\'\'\'' or i[-3:] == '\"\"\"'):    #多行注释开头

                            commentCount += 1

                            multiCommentEndFlag = 1

                            if len(i) > 3 and (i[-3:] == '\'\'\'' or i[-3:] == '\"\"\"'):   #多行注释符在同一行的情况

                                multiCommentEndFlag = 0

                        elif i == '':   #空行

                            nbspCount += 1

                        elif len(i) >= 1 and i[0] == '#':   #单行注释（#）

                            commentCount += 1

                print('{0}中代码行数为：{1}，其中空行数：{2}，注释数：{3}'.format(fileName, codeCount, nbspCount, commentCount))

if __name__ == '__main__':

    countCode(r'C:\Users\Admin\PycharmProjects\untitled\python每日一练')