Time a python funtion

timeit.py

# Timer for functions
from functools import wraps


def timeit(f):
    @wraps(f)
    def wrap(*args, **kwargs):
        start = time()
        r = f(*args, **kwargs)
        print(f.__name__, time() - start)
        return r

    return wrap

# Timer Context Manager
from time import time

class Timer:
    def __enter__(self):
        self.start = time()
        return self

    def __exit__(self, *args, **kwargs):
        self.end = time() - self.start

    def time(self):
        return self.end