[code]统计项目代码量

功能
统计一个文件夹中所有指定后缀名文件中的数据行数

应用场景
统计项目代码行数

代码

# coding=utf-8

import os

def get_line_counts(path, file_types, include_blank_line=True):
    """
    path:str:文件夹全路径
    file_types:tuple or list:要统计的文件后缀列表:如['py', 'cpp', 'java']
    include_blank_line:bool:空行是否也计入总的统计行数中
    """
    total_lines = 0
    for parent, _, filenames in os.walk(path):  # 遍历path下的所有子文件夹
        for filename in filenames:  # 遍历文件夹下的所有文件
            file_type = filename.split('.')[-1]  # 获取文件后缀名
            if file_type in file_types:
                file_full_path = os.path.join(parent, filename)  # 获取文件的全路径
                if not include_blank_line:
                    for line in open(file_full_path).readlines():
                        if len(line.strip()) != 0:
                            total_lines += 1
                else:
                    total_lines += len(open(file_full_path).readlines())

    return total_lines

用例

line_count = get_line_counts(r"C:\Users\Administrator\Desktop\my_python_tools", ['py'])