Catching exception in context manager __enter__()

by

Like this:

import sys

class Context(object):
    def __enter__(self):
        try:
            raise Exception("Oops in __enter__")
        except:
            # Swallow exception if __exit__ returns a True value
            if self.__exit__(*sys.exc_info()):
                pass
            else:
                raise


    def __exit__(self, e_typ, e_val, trcbak):
        print "Now it's running"


with Context():
    pass

To let the program continue on its merry way without executing the context block you need to inspect the context object inside the context block and only do the important stuff if __enter__ succeeded.

class Context(object):
    def __init__(self):
        self.enter_ok = True

    def __enter__(self):
        try:
            raise Exception("Oops in __enter__")
        except:
            if self.__exit__(*sys.exc_info()):
                self.enter_ok = False
            else:
                raise
        return self

    def __exit__(self, e_typ, e_val, trcbak):
        print "Now this runs twice"
        return True


with Context() as c:
    if c.enter_ok:
        print "Only runs if enter succeeded"

print "Execution continues"

As far as I can determine, you can’t skip the with-block entirely. And note that this context now swallows all exceptions in it. If you wish not to swallow exceptions if __enter__ succeeds, check self.enter_ok in __exit__ and return False if it’s True.

More Related Contents:

  1. Create a “with” block on several context managers? [duplicate]
  2. Explaining Python’s ‘__enter__’ and ‘__exit__’
  3. How to save traceback / sys.exc_info() values in a variable?
  4. Python Multiprocessing Lib Error (AttributeError: __exit__)
  5. Understanding the Python with statement and context managers
  6. Get exception description and stack trace which caused an exception, all as a string
  7. Alternative to contextlib.nested with variable number of context managers
  8. Context manager for Python’s MySQLdb
  9. Can Syntax Errors be handled?
  10. Can Syntax Errors be handled?
  11. Execute python script while open terminal
  12. a python library that accepts some text, and replaces phone numbers, names, and so on with tokens
  13. How do I check if a variable exists?
  14. Division in Python 2.7. and 3.3 [duplicate]
  15. Integer division in Python 2 and Python 3
  16. How do I mock an open used in a with statement (using the Mock framework in Python)?
  17. How to check the version of Python modules
  18. Cleanest way to get last item from Python iterator
  19. Python: Catching specific exception
  20. Default arguments with *args and **kwargs
  21. rounding errors in Python floor division
  22. How to Install and Use TkDnD with Python Tkinter on OSX?
  23. Why python’s list slicing doesn’t produce index out of bound error? [duplicate]
  24. Byte Array to Hex String
  25. BaseException.message deprecated in Python 2.6
  26. Read file with timeout in Python
  27. What code can I use to check if script is running in IDLE?
  28. Run process with realtime output to a Tkinter GUI
  29. ‘sys.excepthook’ and threading
  30. Google Search Web Scraping with Python

Leave a Comment