Python装饰器函数调用深度，结构化输出

Question

我正在打印函数的输入和输出，但这会导致日志难以阅读，尤其是嵌套函数调用过多(非递归)时。

示例:

Calling test1()
Calling test2()
Calling test3()
Calling test4()
'test4' returned None
'test3' returned None
'test2' returned None
'test1' returned None

我希望看到这样的输出:

Calling test1()
    Calling test2()
        Calling test3()
            Calling test4()
            'test4' returned None
        'test3' returned None
    'test2' returned None
'test1' returned None

如果我可以保留调用计数并通过每次递增将其转移到下一个调用，我就可以实现这一目标。然后我什至可以根据此计数的值对输出进行着色并生成更易于阅读的日志。

有没有办法可以将当前调用的深度保留在迭代器内？

作为引用，这里是装饰器:

def trace(func):
    """Print the function signature and return value"""
    @functools.wraps(func)
    def wrapper_debug(*args, **kwargs):
        args_repr = [repr(a) for a in args]                      # 1
        kwargs_repr = [f"{k}={v!r}" for k, v in kwargs.items()]  # 2
        signature = ", ".join(args_repr + kwargs_repr)           # 3
        print(start + f"Calling {func.__name__}({signature})" + end)
        value = func(*args, **kwargs)
        print(f"{func.__name__!r} returned {value!r}")           # 4
        return value
    return wrapper_debug

Answer 1

Python 随时知道您的调用深度。

您还可以随时打印您当前所在位置的回溯。

无需将其保存在单独的变量中。

所以给你:

import traceback
len(traceback.format_stack())

由您来进行缩进并修复任何相差一位的错误。