python计算项目内的总代码行数

毕设实用小工具，通过递归查找的形式来计算项目内自己需要计算的文件内的代码行数，同时这个精简的递归算法可以用于其他文件操作，如文件夹上传下载云云

import os


# fp为你想要计算的根目录
fp = "D:\Code_Proagram\Project\Pycharm\Graduate-design-kg-construction\frontend\src\"
globals()['total_lines'] = 0
calculate_list = ['py', 'vue', 'js', 'css']

def recursion_search_dir(fp, cur_path):
    if os.path.isdir(fp):
        print('----', fp)
        cur_files = os.listdir(fp)
        for fn in cur_files:
            print(fn)
            if '.' not in fn:  # 目录
                recursion_search_dir(fp + '\' + fn, cur_path=cur_path + fn + '/')
            else:  # 文件
                local_path = fp + '\' + fn
                if fn.split('.')[-1] in calculate_list:
                    with open(local_path, 'r', encoding='utf-8') as f:
                        globals()['total_lines'] += len(f.readlines())

recursion_search_dir(fp, '')
print(globals()['total_lines'])