profile_this.py

import cProfile


def profile_this(fn):
    def profiled_fn(*args, **kwargs):
        filename = fn.__name__ + '.profile'
        prof = cProfile.Profile()
        ret = prof.runcall(fn, *args, **kwargs)
        prof.dump_stats(filename)
        return ret
    return profiled_fn

if __name__ == '__main__':
    def f1():
        a = [x * x for x in range(10000)]
        return a

    def f2():
        a = [x * x for x in range(20000)]
        return a

    def f3():
        a = [x * x for x in range(30000)]
        return a

    @profile_this
    def test():
        f1()
        f2()
        f3()

    test()
    print('Profile is available in test.profile.'
          'Use runsnake or snakeviz to view it')