Python – inspect 模块的简单使用

技术分享 1个月前 (03-28) 0 999+

Python中的`inspect`模块解析

Python的inspect模块是一个强大的内省工具，允许开发者检查（inspect）活动对象和源代码。它提供了一系列函数，用于获取信息关于正在运行的程序和调用堆栈，非常适合进行调试和动态分析。本文将通过介绍inspect模块的关键功能，并结合实际案例代码，来探索其在日常开发中的应用。

常用方法

1. 获取当前执行的函数或方法名、文件路径【并不是调用方】

在日志记录或调试时，知道当前执行的函数名是非常有用的

import inspect  def who_am_i():     # 输出当前文件绝对路径     print(inspect.currentframe().f_code.co_filename)     return inspect.currentframe().f_code.co_name  print(who_am_i())  # 输出: who_am_i

个人认为比较有用的就是 co_filename、co_name

2. 获取调用者信息

获取当前函数或方法的调用者信息

import inspect  def caller_info():     frame = inspect.currentframe().f_back     print(f调用者 {frame.f_code.co_filename} 调用行号 d{frame.f_lineno}")  def test():     caller_info()  # 调用以获取调用者信息  test()

这个例子显示了如何获取调用当前函数的代码位置，非常有助于调试复杂的调用链

3. 查看函数参数

inspect模块可以用来检查函数或方法的参数，这对于动态分析和生成文档非常有用

import inspect  def sample_function(name, age=25):     pass  sig = inspect.signature(sample_function) print(sig)  # 输出: (name, age=25)

4. 获取源代码

inspect还可以用来获取函数、类或模块的源代码

import inspect  def my_function():     """A simple function."""     pass  print(inspect.getsource(my_function))

5. 检查类和实例

inspect模块提供了多种方式来检查类和实例，比如获取类的所有方法、属性等

class MyClass:     def method_one(self):         pass          def method_two(self):         pass  # 获取类的所有成员方法 methods = inspect.getmembers(MyClass, predicate=inspect.isfunction) print(methods)  # 输出 MyClass 中定义的方法

实际案例：自动化场景下的应用

一个常见的使用场景是动态地调用函数或方法，并基于它们的签名自动生成文档。

 def test():     # 获取调用方     frame = inspect.currentframe().f_back     # 获取调用方文件绝对路径     caller_file = inspect.getfile(frame)     # 这种方式也可以     caller_file = frame.f_code.co_filename      ...          params = [         caller_file,         "--env-data",         env_data.json(),         f"--count={count}",         "-m",         mark,     ]

一个基于 Pytest 自动化测试项目