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

时间:2018-10-05 09:11:36

标签: python python-3.x decorator

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

示例:

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

1 个答案:

答案 0 :(得分:1)

Python随时知道您的通话深度。

您还可以随时打印当前位置的追溯。

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

所以你去了:

<strong>

由您自己决定缩进并修复所有一次性错误。