Source code for pylasu.support

def extension_method(cls):
    """Installs the decorated function as an extension method on cls.
    See https://mail.python.org/pipermail/python-dev/2008-January/076194.html"""
    def decorator(func):
        name = func.__name__
        if name in cls.__dict__:
            raise Exception(f"{cls} already has a member called {name}")
        setattr(cls, name, func)
        return func
    return decorator