Python Decorator to determine execution time of a given function

timeit.py

"""Decorator to determine execution time of a given function."""
from time import time


def timeit(method: func):
  """Print execution time of decorated function."""
  
    def timed(*args, **kw):
        ts = time()
        result = method(*args, **kw)
        te = time()
        if 'log_time' in kw:
            name = kw.get('log_name', method.__name__.upper())
            kw['log_time'][name] = int((te - ts) * 1000)
        else:
            print(f"%r  %2.2f ms" % \
                  (method.__name__, (te - ts) * 1000))
        return result
    return timed