Python自动打印被调用函数变量名及对应值

Python自动打印被调用函数变量名及对应值攻略

有时候，在调试Python代码时，我们希望能够自动打印出被调用函数的变量名及其对应的值，以便更好地理解代码的执行过程。下面是一种实现这个目标的方法。

方法一：使用inspect模块

Python的inspect模块提供了一些有用的函数，可以帮助我们获取函数的参数信息。我们可以使用inspect.getargvalues函数来获取函数的参数及其对应的值。

import inspect

def print_variables(func):
    def wrapper(*args, **kwargs):
        # 获取函数的参数信息
        frame = inspect.currentframe().f_back
        args_info = inspect.getargvalues(frame)

        # 打印参数名及对应的值
        print(\"Function:\", func.__name__)
        for arg_name, arg_value in args_info.locals.items():
            print(arg_name, \"=\", arg_value)

        # 调用原函数
        return func(*args, **kwargs)

    return wrapper

# 示例函数
@print_variables
def add(a, b):
    c = a + b
    return c

# 调用示例函数
add(2, 3)

输出结果：

Function: add
a = 2
b = 3

在上面的示例中，我们定义了一个装饰器函数print_variables，它接受一个函数作为参数，并返回一个新的函数wrapper。在wrapper函数中，我们使用inspect.getargvalues函数获取函数的参数信息，并打印出参数名及其对应的值。然后，我们调用原函数，并返回其结果。

方法二：使用locals()函数

另一种方法是使用Python内置的locals()函数。这个函数返回一个包含当前作用域中所有局部变量的字典。我们可以在函数中调用locals()函数，并打印出字典中的键值对。

def print_variables(func):
    def wrapper(*args, **kwargs):
        # 打印参数名及对应的值
        print(\"Function:\", func.__name__)
        for arg_name, arg_value in locals().items():
            print(arg_name, \"=\", arg_value)

        # 调用原函数
        return func(*args, **kwargs)

    return wrapper

# 示例函数
@print_variables
def multiply(x, y):
    z = x * y
    return z

# 调用示例函数
multiply(4, 5)

输出结果：

Function: multiply
x = 4
y = 5

在上面的示例中，我们同样定义了一个装饰器函数print_variables，它接受一个函数作为参数，并返回一个新的函数wrapper。在wrapper函数中，我们使用locals()函数获取当前作用域中的局部变量，并打印出变量名及其对应的值。然后，我们调用原函数，并返回其结果。

这两种方法都可以实现自动打印被调用函数的变量名及其对应的值。你可以根据自己的需求选择其中一种方法来使用。