代码量简单统计

代码量简单统计

def count_code(file_path):  # 单文件代码统计
    count=0
    tag=False
    with open(file_path, 'r', encoding='utf-8') as f:
        for line in f:
            if tag and line.startswith('"""') or line.startswith("'''"):
                tag=False
            if tag and  not line.startswith('"""') or line.startswith("'''"):
                continue
            if line.startswith("#"):
                continue
            if line.startswith('\n'):
                continue
            count+=1
        return count

# 所有代码统计量
def count_file_code(top_packge):
    count_sum = 0
    # 针对文件做处理
    if os.path.isfile(top_packge):
        count=count_code(top_packge)
        return count

    # 针对文件夹做处理

    res = os.walk(top_packge)  # 只针对文件夹
    for dir, _, files in res:
        # print(files)
        for file in files:
            file_path = os.path.join(dir, file)
            # print(file_path)
            if file_path.endswith('.py'):
            # print(file_path)
                count=count_code(file_path)   # 单文件代码统计
                count_sum += count
    return count_sum

top=r'G:\Python代码日常\第一阶段\1阶段\模块\md.py'
count_sum=count_file_code(top)
print(f'{top}文件夹统计代码量为：{count_sum}')