Time Decorator #python #decorator

time_decorator.py

from time import time

def timeit(method):
  def timed(*args, **kw):
    time_start = time()
    result = method(*args, **kw)
    time_end = time()
    time_diff = (time_end - time_start) * 1000
    print('{}: {:2.2f} ms'.format(method.__name__, time_diff))
    return result
  return timed