kurbezz/discord-bot
1
0
Fork 0
mirror of https://github.com/kurbezz/discord-bot.git synced 2025-12-06 15:15:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update retries

Browse Source
This commit is contained in:
Bulat Kurbanov
2024-11-17 22:20:23 +01:00
parent 698fe4b40d
commit 3fc362e6ff
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/modules/stream_notifications/twitch/webhook.py
View File

@@ -1,6 +1,6 @@
from asyncio import sleep from asyncio import sleep
import logging import logging
from typing import NoReturn from typing import NoReturn, Literal
from twitchAPI.eventsub.webhook import EventSubWebhook from twitchAPI.eventsub.webhook import EventSubWebhook
from twitchAPI.twitch import Twitch from twitchAPI.twitch import Twitch
@@ -30,18 +30,23 @@ class TwitchService:
async def subscribe_with_retry( async def subscribe_with_retry(
self, self,
method: Literal["listen_channel_update_v2"] | Literal["listen_stream_online"],
eventsub: EventSubWebhook, eventsub: EventSubWebhook,
streamer: StreamerConfig, streamer: StreamerConfig,
retry: int = 10 retry: int = 10
): ):
try: try:
await eventsub.listen_channel_update_v2(str(streamer.twitch.id), self.on_channel_update) if method == "listen_channel_update_v2":
await eventsub.listen_stream_online(str(streamer.twitch.id), self.on_stream_online) await eventsub.listen_channel_update_v2(str(streamer.twitch.id), self.on_channel_update)
elif method == "listen_stream_online":
await eventsub.listen_stream_online(str(streamer.twitch.id), self.on_stream_online)
else:
raise ValueError("Unknown method")
except Exception as e: except Exception as e:
if retry > 0: if retry > 0:
await sleep(1) await sleep(1)
await self.subscribe_with_retry(eventsub, streamer, retry - 1) await self.subscribe_with_retry(method, eventsub, streamer, retry - 1)
else: else:
raise e raise e
@@ -64,7 +69,8 @@ class TwitchService:
for streamer in streamers: for streamer in streamers:
logger.info(f"Subscribe to events for {streamer.twitch.name}") logger.info(f"Subscribe to events for {streamer.twitch.name}")
await self.subscribe_with_retry(eventsub, streamer) await self.subscribe_with_retry("listen_channel_update_v2", eventsub, streamer)
await self.subscribe_with_retry("listen_stream_online", eventsub, streamer)
logger.info(f"Subscribe to events for {streamer.twitch.name} done") logger.info(f"Subscribe to events for {streamer.twitch.name} done")
logger.info("Twitch service started") logger.info("Twitch service started")