profiler.py 文件源码

def do_cprofile(func):
    """
    ?????cProfile???????????
    ???
    @do_cprofile
    def expensive_function():
        for x in range(10000):
            _ = x ** x
        return 'OK!'

    result = expensive_function()
    """
    def profiled_func(*args, **kwargs):
        profile = cProfile.Profile()
        try:
            profile.enable()
            result = func(*args, **kwargs)
            profile.disable()
            return result
        finally:
            profile.print_stats()
    return profiled_func