ContextDecorator for Python 2.6/2.7.

contextdecorator.py

from functools import wraps

class ContextDecorator(object):
    def __call__(self, fn):
        @wraps(fn)
        def decorator(*args, **kw):
            with self:
                return fn(*args, **kw)
     
    def __enter__(self):
        # Do whatever setup.

    def __exit__(self, type, value, tb):
        # Do whatever cleanup.
        if any((type, value, tb)):
            raise type, value, tb

mycontextdecorator = ContextDecorator()

with mycontextdecorator:
    # Do something!

@mycontextdecorator
def myfunc():
    # Do something!