python代码遍历文件夹所有子文件夹下的py文件的代码行数

上次写了一篇如何统计文件夹下python代码行数的文章

https://www.toutiao.com/article/7325058522729939497/

后来发现文件夹下可能还有其他的子文件夹，因此优化了一下代码：

#!/usr/bin/python3.9
# -*- coding: utf-8 -*-
#
# Copyright (C) 2024 , Inc. All Rights Reserved
#
# @Time      : 2024/1/17 20:45
# @Author    : fangel
# @FileName  : python代码行数统计.py
# @Software  : PyCharm

import os

def all_file(folder):
    # 先获取指定路径下的文件以及子文件夹下的文件，将其加入到filePathList列表中
    for file in os.listdir(folder):
        file_path = os.path.join(folder, file)
        if os.path.isdir(file_path):
            all_file(file_path)
        else:
            filePathList.append(file_path)
    return filePathList

def count_lines(folder):
    #统计指定文件夹下所有文件的代码行数
    sum = 0

    for file in all_file(folder):
        # 只统计后缀为.py的文件，其它文件类型跳过
        if not file.endswith('.py'):
            continue
        # 打开文件
        with open(file, 'r', encoding='UTF-8') as f:
            # total统计单个文件的代码行数
            total = 0
            # 读取所有的行数
            lines = f.readlines()
            # 逐行进行分析
            for line in lines:
                # 去掉空格，目的是下面的不统计空行
                line = line.strip()
                # 判断是否是空行，空行就不统计，进入下一轮循环
                if len(line) == 0:
                    continue
                # 不统计以#开头的注释行数
                if not line.startswith("#"):
                    total += 1
            # 每个文件读取结束后，总数sum应该要加上每个文件的行数
            sum += total
        print(file, ":", total, "行")
        f.close

    return sum

if __name__ == '__main__':
    filePathList = []
    # 调用函数并传入目标文件夹路径
    total_lines = count_lines('D:\\学习视频\\Python相关软件\\fangelPythonProject\\2024年代码')
    print('\nAll File Total Lines:', total_lines)