checktheroads-hyperglass/hyperglass/external/webhooks.py

"""Convenience functions for webhooks."""

# Project
from hyperglass.exceptions import HyperglassError
from hyperglass.external._base import BaseExternal
from hyperglass.external.slack import SlackHook
from hyperglass.external.generic import GenericHook
from hyperglass.external.msteams import MSTeams

PROVIDER_MAP = {
    "generic": GenericHook,
    "msteams": MSTeams,
    "slack": SlackHook,
}


class Webhook(BaseExternal):
    """Get webhook for provider name."""

    def __new__(cls, config):
        """Return instance for correct provider handler."""
        try:
            provider_class = PROVIDER_MAP[config.provider]
            return provider_class(config)
        except KeyError:
            raise HyperglassError(
                f"'{config.provider.title()}' is not yet supported as a webhook target."
            )