Python??inspect?模块的简单使用及常用方法

最新推荐文章于 2025-01-02 10:29:16 发布

原创最新推荐文章于 2025-01-02 10:29:16 发布 · 1k 阅读

7 ·

CC 4.0 BY-SA版权

文章来源：https://www.piaodoo.com/

文章标签：

#python #开发语言

本文介绍了Pythoninspect模块的强大功能，包括获取当前函数名、文件路径，检查调用者，查看函数参数，获取源代码，以及检查类和实例。着重展示了在自动化场景下，如动态函数调用和文档生成中的应用。

教程下载地址：网赚博客http://www.piaodoo.com/创业项目排行榜前十名http://www.piaodoo.com/

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 自动化测试项目