40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import httpx
|
|
|
|
from ...settings import Settings
|
|
from .alertmanager import AlertmanagerNotifier
|
|
from .base import Notifier
|
|
from .debug import DebugNotifier
|
|
from .telegram import TelegramNotifier
|
|
from .webhook import WebhookNotifier
|
|
|
|
|
|
def build_registry(settings: Settings, client: httpx.AsyncClient) -> dict[str, Notifier]:
|
|
reg: dict[str, Notifier] = {}
|
|
if settings.notifier_debug_enabled:
|
|
reg["debug"] = DebugNotifier(settings.notification_time_zone)
|
|
if (
|
|
settings.notifier_telegram_enabled
|
|
and settings.notifier_telegram_token
|
|
and settings.notifier_telegram_chat_id
|
|
):
|
|
reg["telegram"] = TelegramNotifier(
|
|
client,
|
|
settings.notifier_telegram_token,
|
|
settings.notifier_telegram_chat_id,
|
|
settings.notification_time_zone,
|
|
)
|
|
if settings.notifier_webhook_enabled and settings.notifier_webhook_url:
|
|
reg["webhook"] = WebhookNotifier(
|
|
client,
|
|
settings.notifier_webhook_url,
|
|
settings.notifier_webhook_token,
|
|
settings.notification_time_zone,
|
|
)
|
|
if settings.notifier_alertmanager_enabled and settings.notifier_alertmanager_url:
|
|
reg["alertmanager"] = AlertmanagerNotifier(
|
|
client, settings.notifier_alertmanager_url, settings.notification_time_zone
|
|
)
|
|
return reg
|