contextmanager as with block and as decorator

test-context-manager.py

#!/usr/bin/env python3
import time
from contextlib import contextmanager


@contextmanager
def timing():
    start = time.time()
    yield
    elapsed = time.time() - start

    print("Took", elapsed, "seconds to run")


@timing()
def foo():
    print("foo will sleep")
    time.sleep(0.2)


def bar():
    with timing():
        print("bar will sleep")
        time.sleep(0.3)


if __name__ == '__main__':
    foo()
    bar()