CodePlex Issue #31043 Plain Text Attachments

context_bug.py

from contextlib import contextmanager
from functools import wraps

@contextmanager
def enter_stage(message):
    # Start stage
    print("Start '{0}' stage".format(message))

    try:
        yield
    finally:
        # Finish stage
        print("End '{0}' stage".format(message))
        
def enter_stage_decorator(message):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            with enter_stage(message):
                return func(*args, **kwargs)
        return wrapper
    return decorator

# Examples of how it is intended to use defined above managers:
#
# with enter_stage("test 1"):
#     print "  Do test 1 stuff"
#
# @enter_stage_decorator("test 2")
# def test_2(x):
#     print "  Do test 2 stuff (x = {0})".format(x)
#
# test_2(30)


# Bug is rising below:

@enter_stage_decorator("test")
def test():
    print "  test() call f()"
    f()
    print "  f() is done" # never reached

def f():
    print "  f() call g()"
    g()

def g():
    print "  g() raise exception now"
    raise Exception("some exception")

try:
    test()
except:
    import traceback
    traceback.print_exc()