cProfile_decorator.py

""" A decorator to easily use cProfile on function """
import cProfile

def profiler(sort=-1, use_globals=False, use_locals=True):
    def decr(func):
        def wrp(*args, **kwargs):
            g = globals() if use_globals else None
            l = locals() if use_locals else None
            cProfile.runctx("func(*args, **kwargs)", g, l, sort=sort)
            return func(*args, **kwargs)
        return wrp
    return decr