Auto_Bangumi/backend/src/module/notification/providers/__init__.py at 7bf1c75674e61375db45fc26b1ecf3eb5b839ee9 · AMYdd00/Auto_Bangumi · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
"""Notification providers registry."""

from typing import TYPE_CHECKING

from module.notification.providers.telegram import TelegramProvider
from module.notification.providers.discord import DiscordProvider
from module.notification.providers.bark import BarkProvider
from module.notification.providers.server_chan import ServerChanProvider
from module.notification.providers.wecom import WecomProvider
from module.notification.providers.gotify import GotifyProvider
from module.notification.providers.pushover import PushoverProvider
from module.notification.providers.webhook import WebhookProvider
from module.notification.providers.onebot import OneBotProvider

if TYPE_CHECKING:
    from module.notification.base import NotificationProvider

# Registry mapping provider type names to their classes
PROVIDER_REGISTRY: dict[str, type["NotificationProvider"]] = {
    "telegram": TelegramProvider,
    "discord": DiscordProvider,
    "bark": BarkProvider,
    "server-chan": ServerChanProvider,
    "serverchan": ServerChanProvider,  # Alternative name
    "wecom": WecomProvider,
    "gotify": GotifyProvider,
    "pushover": PushoverProvider,
    "webhook": WebhookProvider,
    "onebot": OneBotProvider,
}

__all__ = [
    "PROVIDER_REGISTRY",
    "TelegramProvider",
    "DiscordProvider",
    "BarkProvider",
    "ServerChanProvider",
    "WecomProvider",
    "GotifyProvider",
    "PushoverProvider",
    "WebhookProvider",
    "OneBotProvider",
]