class MyContext:
    def __enter__(self):
        print("enter")
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("exit")
        return True

with MyContext():
    print("run")
    raise ValueError("test")
print("after")