tracing_decorator.py

# Modified from https://cscheid.net/2017/12/11/minimal-tracing-decorator-python-3.html

import inspect

trace_indent = 0
def tracing(f):
    sig = inspect.signature(f)
    def do_it(*args, **kwargs):
        global trace_indent
        ws = ' ' * (trace_indent * 2)

        params = ','.join([
            f"{param.name}: {repr(args[ix-1])}"
            for ix, param in enumerate(sig.parameters.values())
            if param.name != 'self'
        ])

        print(f"{ws}ENTER {f.__module__}::{f.__qualname__}({params}):")
        sys.stdout.flush()
        trace_indent += 1
        result = f(*args, **kwargs)
        trace_indent -= 1
        print(f"{ws}EXIT {f.__module__}::{f.__qualname__}:")
        sys.stdout.flush()
        return result
    return do_it