cmwrap.py

import functools
import contextlib


def cmwrap(cm, *cmargs, **cmkwargs):
    """Decorator wrapping a context manager around a function call."""
    def wrapper(f):
        @functools.wraps(f)
        def wrapped_f(*args, **kwargs):
            with cm(*cmargs, **cmkwargs):
                return f(*args, **kwargs)

        return wrapped_f
    return wrapper


@contextlib.contextmanager
def verbose(before, after):
    print(before)
    yield
    print(after)


@cmwrap(verbose, "hello", "bye")
def foo(bar):
    print(bar)


if __name__ == '__main__':
    foo("doing something")