Error Registry

error_registry.py

class Jug:
    def __init__(self):
        self.exception_handlers = {}

    def errorhandler(self, exception):
        def decorator(f):
            self.exception_handlers[exception] = f
            return f
        return decorator

    def call_function(self, func):
        try:
            func()
        except Exception as e:
            self.exception_handlers[type(e)](e)


jug = Jug()


@jug.errorhandler(ValueError)
def foo(e):
    print("THIS IS FOO")


@jug.errorhandler(KeyError)
def bar(e):
    print("THIS IS BAR")


def func1():
    print("Func1!")
    raise ValueError()


def func2():
    print("Func2!")
    raise KeyError()


jug.call_function(func1)
jug.call_function(func2)