Python の contextlib モジュールの contextmanager のようなものを自前で書くとどんな感じになるか？

gistfile1.py

def my_contextmanager(function):
    class Context(object):
        def __init__(self, *args, **kw):
            self.ctx = function(*args, **kw)
        def __enter__(self):
            return self.ctx.next()
        def __exit__(self, exception_type, exception_value, exception_traceback):
            try:
                if exception_type is None:
                    self.ctx.next()
                else:
                    self.ctx.throw(exception_type, exception_value, exception_traceback)
            except StopIteration:
                return True
            else:
                return True

    def _my_contextmanager(*args, **kw):
        return Context(*args, **kw)

    return _my_contextmanager