generic decorator with functools.wraps

timethis.py

import functools

def timethis(func):
    """A profiling function used to record the time consumed on a function call."""

    @functools.wraps(func)
    def _(self, *a, **kw):

        start = time()
        ret = func(self, *a, **kw)
        end = time()
        total_time = end - start

        self.queries.append({
            'func': func.__name__,
            'start': start,
            'end': end,
            'total': total_time,
        })

        return ret

    return _