debug_on_error.py

import pdb
import traceback


class DebugOnError:
    def __init__(self, suppress=False):
        self.suppress = suppress

    def __enter__(self):
        return None

    def __exit__(self, _type, _value, _traceback):
        if _traceback:
            traceback.print_exception(_type, _value, _traceback)
            pdb.post_mortem(_traceback)
            return self.suppress
        return False