decorators/ctxmgrs

wrapped_ctxmgr.py

from contextlib import contextmanager


def to_mgr(fn):
    fn.__converted__ = True

    @contextmanager
    def fn2(*args, **kwargs):
        try:
            fn(*args, **kwargs)
            yield
        except Exception:
            pass
        finally:
            print('DOne')
    return fn2


@to_mgr
def greetings(name='anon'):
    print('HEEEY, {}'.format(name))
    print('HEee2')


if __name__ == '__main__':
    with greetings(name='me'):
        print('Hello back')