20190227-做一个简易代码统计工具

做一个代码统计工具：

要求：
　　1.如果是文件，就直接统计文件行数，并打印结果
　　2.判断是否是目录，是就遍历统计目录下所有的文件
文件统计规则：
　　1.开头是#_*_或者#encoding的需要算作代码统计
　　2.注释#,'''或者"""判断某一行开头是"""或者'''就不计
　　3.空行不统计
　　4.统计当前文件行数，还有累计行数

要求算法：

　　1.判断是否是文件，且文件存在，则打开文件进行统计

　　2.判断是目录，则变量目录下所有的文件，进行统计

统计规则算法中，注释种类有如下几种:

a.以#开头进行单行注释

#XXXXX

b.以三个双引号"""开头和结尾的单行注释

"""XXXXX"""
'''XXXXX'''

c.以三个双引号或者单引号开头换行进行注释格式如下：

"""
XXXXX
"""
'''
XXX
'''

d.以三个双引号或者单引号开头不换行进行注释格式如下:

"""XXXXX
"""
'''XXXXX
'''

e.以三个双引号或者单引号结尾不换行进行注释格式如下:

"""
XXXXX"""
'''
XXXXX'''

根据以上多行注释情况，考虑设置标志位，当标志位是True的时候统计代码，当标志位是False的时候不统计，注意考虑标志位还原的情况编写统计规则如下：

def coding_rule(filename):
    count = 0
    import string
    with open(filename,'r',encoding ='utf-8') as fp:
        flag = True
        for line in fp:
            #print(line)
            if line.strip().startswith('#encoding') or line.strip().startswith('#_*_'):
                count+=1
                #统计开头是#_*_或者#encoding的代码
            elif not line.strip():
                continue
                #如果是空行，则不计
            elif line.strip().startswith('#'):
                #如果是单行注释格式如#xxx,则不统计，匹配a的情况
                continue
            elif line.strip().startswith('"""') and line.strip().endswith('"""') and len(line.strip())>3:
                #如果是单行注释格式如"""xxx""",则不统计，匹配b的情况
                continue
            elif line.strip().startswith("'''") and line.strip().endswith("'''") and len(line.strip())>3:
                continue
                #如果是单行注释格式如'''xxx''',则不统计，匹配b的情况
            elif line.strip().startswith("'''") and len(line.strip())==3 and flag ==True:
                #如果是以三个单引号'''开始进行注释的，设置标志位为False，匹配c的情况
                flag =False
                continue
            elif line.strip().startswith('"""') and len(line.strip())==3 and flag ==True:
                #如果是以三个双引号"""开始进行注释的，设置标志位为False，匹配c的情况
                flag =False
                continue
            elif line.strip().startswith('"""') and not line.strip().endswith('"""'):
                #如果是三个双引号开头不换行进行注释，设置标志位为False，匹配d的情况
                flag =False
                continue
            elif line.strip().startswith("'''") and not line.strip().endswith("'''"):
                #如果是三个单引号引号开头不换行进行注释，设置标志位为False，匹配d的情况
                flag =False
                continue
            elif not line.strip().startswith('"""') and line.strip().endswith('"""'):
                #如果以三个双引号结尾不换行进行注释,还原标志位为True，匹配e的情况
                flag =True
            elif not line.strip().startswith("'''") and line.strip().endswith("'''"):
                #如果以三个单引号结尾不换行进行注释,还原标志位为True，匹配e的情况
                flag =True
            elif line.strip().startswith("'''") and len(line.strip())==3 and flag ==False:
                #如果是三个单引号引号开头不换行进行注释，还原标志位为True，匹配d的情况
                flag =False
                flag =True
            elif line.strip().startswith('"""') and len(line.strip())==3 and flag ==False:
                #如果是三个单引号引号开头不换行进行注释，还原标志位为True，匹配d的情况
                flag =False
                flag =True
            elif flag ==True:
                count+=1
            #print(flag)
            #print(count)
    return count

另外写一个函数，来写代码统计的要求部分，如下：

def coding_count(*source_dir):
    import os
    result ={}
    total_count =0
    except_file=[]
    for source in source_dir:
        if os.path.exists(source) and (source[-2:]=='py' or source[-3:]=='txt'):
            #如果文件存在，且是.py文件
            result[source]=coding_rule(source)
            total_count +=coding_rule(source)
        #读文件
        elif os.path.exists(source) and os.path.isdir(source):
            #如果输入的dir是一个目录，且存在
            for root,dirs,files in os.walk(source):
                #print('当前目录：%s'%root)
                for file in files:
                    file_dir = os.path.join(root,file)
                    if os.path.basename(file)[-2:]=='py' or os.path.basename(file)[-3:]=='txt':
                        try:
                            result[file]=coding_rule(file_dir)
                            total_count +=coding_rule(file_dir)
                        except:
                            except_file.append(file)
        result['total']=total_count
    return result,except_file
#print(coding_count(r'C:Users何发奋DesktopPython 习题a.txt',r'C:Users何发奋DesktopPython 习题习题库'))
print(coding_count(r'C:Users何发奋DesktopPython 习题a.txt'))