Python 打印调用者的类名|极客教程

Python 打印调用者的类名

在 Python 中，我们经常需要打印调用者的类名来进行调试或日志记录。通过获取当前调用函数的栈信息，我们可以轻松地获得调用者的类名。本文将介绍如何在 Python 中实现这一功能，并通过示例代码来演示具体的实现方法。

使用inspect模块获取调用者的类名

Python 标准库中的 inspect 模块提供了一种方便的方式来获取当前调用函数的栈信息。通过 inspect 模块中的 currentframe() 函数可以获取当前的调用栈帧对象，然后我们可以通过该对象获取调用者的类名。下面是一个示例代码：

import inspect

def get_caller_class_name():
    caller_frame = inspect.currentframe().f_back
    caller_class_name = caller_frame.f_locals.get('__class__')
    return caller_class_name

class Foo:
    def method1(self):
        print(get_caller_class_name())

class Bar:
    def method2(self):
        foo = Foo()
        foo.method1()

bar = Bar()
bar.method2()

运行以上代码，将会输出：

<class '__main__.Bar'>

在上面的示例代码中，我们定义了两个类 Foo 和 Bar，Bar 类中调用了 Foo 类的 method1 方法。通过调用 get_caller_class_name() 函数，我们可以获取到调用者的类名为 Bar。

使用inspect模块获取调用者的类名（另一种方式）

除了获取调用者的类名外，我们还可以通过 inspect 模块的 stack() 函数来获取调用者的类名。该函数返回一个包含当前栈帧信息的列表，其中第一个元素表示调用者的栈帧。下面是另一种实现方式的示例代码：

import inspect

def get_caller_class_name():
    stack = inspect.stack()
    caller_frame = stack[1].frame
    caller_class_name = caller_frame.f_locals.get('__class__')
    return caller_class_name

class Foo:
    def method1(self):
        print(get_caller_class_name())

class Bar:
    def method2(self):
        foo = Foo()
        foo.method1()

bar = Bar()
bar.method2()

运行以上代码，将会输出：