runtime_decorator

runtime.py

def runtime(func):
    @wraps(func)
    def decorated(*args, **kwargs):
        # Decorate front of func
        before  = time.time()

        output  = func(*args, **kwargs)

        # Decorate back of func
        after   = time.time()
        print('Elapsed time: {}'.format(after-before))

        return output
    return decorated