profiling decorator

profiling_decorator.py

import functools
import time

class Stack(object):
    hits = {}
    times = {}

    @classmethod
    def register(cls, func_name, call_time):
        if func_name not in Stack.hits:
            Stack.hits[func_name] = 0
            Stack.times[func_name] = 0
        Stack.hits[func_name] += 1
        Stack.times[func_name] += call_time

    @classmethod
    def report(cls):
        print('Calls made:')
        for k in sorted(Stack.hits):
            print(k, Stack.hits[k], Stack.times[k])

    @staticmethod
    def usage():
        def decorator(fn):
            @functools.wraps(fn)
            def inner(*args, **kwargs):
                start = time.time()
                result = fn(*args, **kwargs)
                end = time.time()
                Stack.register(fn.__name__, call_time=end-start)
                return result
            return inner
        return decorator


@Stack.usage()
def this():
    time.sleep(0.01)
    print('this')

@Stack.usage()
def that():
    print('that')
    time.sleep(0.1)

@Stack.usage()
def them():
    time.sleep(0.001)
    print('them')


def main():
    for i in range(3):
        this()
        that()
        for i in range(3):
            them()
        this()


main()
Stack.report()

# outputs:
# this
# that
# them
# ...
# them
# this
# 
# Calls made:
# that 3 0.32779526710510254
# them 9 0.1409895420074463
# this 6 0.09374713897705078