获取指定目录下所有py文件里的类名

最新推荐文章于 2023-03-07 20:10:13 发布

置顶 Arvin.袁

最新推荐文章于 2023-03-07 20:10:13 发布

阅读量1.9k

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python

本文链接：https://blog.youkuaiyun.com/weixin_43751803/article/details/106471073

python 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种使用Python递归遍历目录并从.py文件中提取类名的方法。通过正则表达式匹配class声明，然后加载源文件并获取类名，最后将所有找到的类名收集到一个列表中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

需要考虑的点

目录下可能还有目录，文件里可能有类似类名格式的字符串

import imp
import os
import re

cls_name_list = []
def get_dir(path):
    try:
        file_list = os.listdir(path)
    except:
        file_list = []
        print("the path is not dir")
    if file_list:
        for file in file_list:
            file = os.path.join(path, file)
            print(file)
            if os.path.isdir(file):
                get_dir(file)
            else:
                if file.endswith(".py"):
                    with open(file, encoding="utf-8") as f:
                        for line in f.readlines():
                            cls_match = re.match(r"class\s(.*?)[\(:]", line)
                            if cls_match:
                                cls_name = cls_match.group(1)
                                try:
                                    module = imp.load_source('mycl', file)
                                    cls_a = getattr(module, cls_name)
                                    if cls_a:
                                        cls_name_list.append(cls_name)
                                except:
                                    pass

if __name__ == '__main__':
    path = "/home/study"
    get_dir(path)
    print(cls_name_list)