如何在Python类中获取函数的返回值？|极客笔记

如何在Python类中获取函数的返回值？

在Python中，类是我们用来定义对象的模板，而函数是具有特定功能的代码块。当我们在类中调用一个函数时，有时需要在类中获取该函数的返回值。

下面是一个示例代码：

class Calculator:
    def add(self, a, b):
        result = a + b
        return result

my_calculator = Calculator()
result = my_calculator.add(2, 3)
print(result)

在这个示例中，我们定义了一个名为Calculator的类，其中包含一个add方法。该方法将两个数相加，并返回计算结果。我们通过创建一个新的Calculator对象，然后调用add方法，将2和3作为参数传递。最后，我们打印出该方法的返回结果。

但是，如果我们尝试从类中获取该函数的返回值呢？这时候我们可以通过使用Python内置的inspect模块来实现。下面是一个示例代码：

import inspect

class Calculator:
    def add(self, a, b):
        result = a + b
        return result

    def get_result(self, method_name, *args):
        method = getattr(self, method_name)
        result = method(*args)
        frame = inspect.currentframe().f_back
        return frame.f_locals['result']

my_calculator = Calculator()
result = my_calculator.get_result('add', 2, 3)
print(result)

在这个示例中，我们添加了一个新的get_result方法，该方法获取函数名和参数，然后使用getattr函数获取函数对象。接下来，我们使用该对象调用函数，并获取函数的返回结果。最后，我们通过使用inspect模块获得当前帧并检查本地变量，来获取函数的实际返回值。

阅读更多：Python 教程