Decorator for timing functions in python

timeit.py

import time

def timeit(method):
    """
    Decorator useful for timing functions.
    Source: https://medium.com/pythonhive/python-decorator-to-measure-the-execution-time-of-methods-fa04cb6bb36d
    
    Usage:
    @timeit
    def my_function():
        pass
    """
    def timed(*args, **kw):
        ts = time.time()
        result = method(*args, **kw)
        te = time.time()

        if 'log_time' in kw:
            name = kw.get('log_name', method.__name__.upper())
            kw['log_time'][name] = int((te - ts) * 1000)
        else:
            print('%r  %2.2f ms' %
                  (method.__name__, (te - ts) * 1000))
        return result

    return timed

If you changed log_time to be a module var and append times to a list, you could store all call times for all decorated functions without the need to pass in the kwarg. Then after the end of your program, before exit, you could post-process (average, stats, graph, whatever) the log_time data at that point. No need to inject it into the function kwargs then. Which also saves you from having to change all timed functions to accept **kwargs. Maybe some don't etc....

👍 to your comment @jlecount. That's a great idea. Will allow for more portability of the decorator.