python实现获取文件中的类方式

 更新时间:2025年12月31日 09:26:16   作者:我一个搬砖人  
本文介绍了两种获取Python文件中类的方法:一种是通过导入文件,另一种是通过不导入文件获取类名,不导入文件的方法通过解析抽象语法树(AST)并使用正则表达式来过滤出特定类名,从而避免了全量导入带来的问题

1. 导入文件的方式获取文件中的类

由于工作需求,需要获得文件中的某一些类的内容。

一般比较常用的方法是通过遍历文件,然后通过调用importlib包导入文件,就能获得文件中的内容,从而获取想要的类。

代码如下:

for importer, package_name, is_pkg in pkgutil.walk_packages('.'):
	importlib.import_module(package_name)
	module = sys.modules[package_name]

	for cls_name, cls in inspect.getmembers(module, inspect.isclass):
		if is_want_class(cls):  
			do_something()

该方法十分简单,但是也有弊端。由于需要导入文件,因此文件中的内容都会被执行。而在一个项目中,存在着成百上千的文件,有一些代码是相斥的。

如果通过这种方法获取类,轻则出现逻辑问题,重则出现异常,报错。

2. 通过不导入文件的方式获取文件中的类名,从而过滤文件

为了防止上面说到的问题影响项目,我们不能将所有文件通通导入进来。

但是换个思路,可以先通过不导入文件的方式获取文件中中存在的类名,如果存在的类名中有想要获取的类,再导入文件。通过这种方式避免导入所有文件,从而大大降低出错的可能性。

我们知道,python解释器在运行时候,对文件做了下列操作:

Parse source code into a parse tree (Parser/pgen.c)
Transform parse tree into an Abstract Syntax Tree (Python/ast.c)
Transform AST into a Control Flow Graph (Python/compile.c)
Emit bytecode based on the Control Flow Graph (Python/compile.c)

其中1,2步执行之后,会生成一棵抽象语法树,从而帮助解释器去解析代码结构。 

因此想要获得文件中的类,可以通过ast包获得生成的ast树,再通过遍历树节点的方式获得相应的内容。通过该方法,可以获得定义的变量和类名的文本信息。

在项目代码中,一般对于类的命名有特定的规范。因此需要正则表达式,可以过滤出文件中是否含有想要查找的类。

代码如下:

for root, dirs, files in os.walk("."):
	for file in files:
	    file_name = os.path.join(root, file)

	    with open(file_name, "r") as source:
                    # 将文件转化为ast树
                tree = ast.parse(source.read())
                # 获取树中的类
                class_name_list = [node.name for node in ast.walk(tree) if isinstance(node, ast.ClassDef)]
                # 这里通过正则want_pattern去匹配类,从而过滤出想要的类
                want_class_name_list = [name for name in class_name_list if re.search(want_pattern, name) is not None]

最后得到的want_class_name_list就是通过正则匹配出来的类的类名。通过判断want_class_name_list是否为空,就能知道文件中是否定义了自己想要的类。

如果定义了,再用方法1获取文件中的类即可。

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • 浅谈用VSCode写python的正确姿势

    浅谈用VSCode写python的正确姿势

    本篇文章主要介绍了浅谈用VSCode写python的正确姿势,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-12-12
  • 教你怎么用python selenium实现自动化测试

    教你怎么用python selenium实现自动化测试

    今天带大家学习怎么用python selenium实现自动化测试,文中有非常详细的介绍及代码示例,对正在学习python的小伙伴们很有帮助,需要的朋友可以参考下
    2021-05-05
  • Django执行源生mysql语句实现过程解析

    Django执行源生mysql语句实现过程解析

    这篇文章主要介绍了Django执行源生mysql语句实现过程解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-11-11
  • 从原理详解Python进行Base64解码的实战技巧

    从原理详解Python进行Base64解码的实战技巧

    Base64编码解码是处理二进制数据的常见需求,无论是处理API响应,还是数据存储,今天我们从实际应用角度出发,深入探讨Base64解码的技术细节,感兴趣的小伙伴可以跟随小编一起学习一下
    2025-08-08
  • Python实现统计文本中字符的方法小结

    Python实现统计文本中字符的方法小结

    在计算机编程中,经常需要对文本数据进行处理和分析,字符统计是其中一个常见任务,本文将详细介绍如何使用Python进行字符统计,希望对大家有所帮助
    2024-01-01
  • 利用python计算均值、方差和标准差(Numpy和Pandas)

    利用python计算均值、方差和标准差(Numpy和Pandas)

    这篇文章主要给大家介绍了关于利用python计算均值、方差和标准差的相关资料,Numpy在Python中是一个通用的数组处理包,它提供了一个高性能的多维数组对象和用于处理这些数组的工具,它是使用Python进行科学计算的基础包,需要的朋友可以参考下
    2023-11-11
  • Python2.7下安装Scrapy框架步骤教程

    Python2.7下安装Scrapy框架步骤教程

    本篇文章主要介绍了Python2.7下安装Scrapy框架步骤教程,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-12-12
  • pandas 如何保存数据到excel,csv

    pandas 如何保存数据到excel,csv

    这篇文章主要介绍了pandas 如何保存数据到excel,csv的实现方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2022-07-07
  • 深入探讨PythonLogging模块的高级用法与性能优化

    深入探讨PythonLogging模块的高级用法与性能优化

    在Python应用程序中,日志处理是一项至关重要的任务,本文将探索Logging模块的高级用法,包括日志级别、格式化、处理程序等方面的功能,需要的可以参考下
    2024-04-04
  • Python实现合并两个列表的方法分析

    Python实现合并两个列表的方法分析

    这篇文章主要介绍了Python实现合并两个列表的方法,结合实例形式对比分析了常见的Python列表合并操作技巧,需要的朋友可以参考下
    2018-05-05

最新评论