kurbezz/discord-bot
1
0
Fork 0
mirror of https://github.com/kurbezz/discord-bot.git synced 2025-12-06 07:05:36 +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
import logging
from typing import NoReturn
from typing import NoReturn, Literal
from twitchAPI.eventsub.webhook import EventSubWebhook
from twitchAPI.twitch import Twitch
@@ -30,18 +30,23 @@ class TwitchService:
async def subscribe_with_retry(
self,
method: Literal["listen_channel_update_v2"] | Literal["listen_stream_online"],
eventsub: EventSubWebhook,
streamer: StreamerConfig,
retry: int = 10
):
try:
await eventsub.listen_channel_update_v2(str(streamer.twitch.id), self.on_channel_update)
await eventsub.listen_stream_online(str(streamer.twitch.id), self.on_stream_online)
if method == "listen_channel_update_v2":
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:
if retry > 0:
await sleep(1)
await self.subscribe_with_retry(eventsub, streamer, retry - 1)
await self.subscribe_with_retry(method, eventsub, streamer, retry - 1)
else:
raise e
@@ -64,7 +69,8 @@ class TwitchService:
for streamer in streamers:
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("Twitch service started")