Refactor streamers configs

src/modules/games_list/discord.py

@@ -8,15 +8,18 @@ from discord import app_commands
 from modules.games_list.games_list import GameList, GameItem
 
 from core.config import config
+from repositories.streamers import StreamerConfigRepository
 
 
 logger = logging.getLogger(__name__)
 
 
-def get_game_list_channel_to_message_map() -> dict[int, int]:
+async def get_game_list_channel_to_message_map() -> dict[int, int]:
     result = {}
 
-    for streamer in config.STREAMERS:
+    streamers = await StreamerConfigRepository.all()
+
+    for streamer in streamers:
         if (integration := streamer.integrations.discord) is None:
             continue
 
@@ -41,7 +44,9 @@ class DiscordClient(discord.Client):
         self.tree = app_commands.CommandTree(self)
 
     async def setup_hook(self):
-        for streamer in config.STREAMERS:
+        streamers = await StreamerConfigRepository.all()
+
+        for streamer in streamers:
             if (integration := streamer.integrations.discord) is None:
                 continue
 
@@ -82,7 +87,7 @@ async def add(
     game: str,
     date: str | None = None
 ):
-    channel_to_message = get_game_list_channel_to_message_map()
+    channel_to_message = await get_game_list_channel_to_message_map()
 
     if interaction.channel is None:
         await interaction.response.send_message("Команда не доступна в этом канале (#1)", ephemeral=True)
@@ -114,7 +119,7 @@ async def game_list_autocomplete(
     if not isinstance(interaction.channel, Messageable):
         return []
 
-    channel_to_message = get_game_list_channel_to_message_map()
+    channel_to_message = await get_game_list_channel_to_message_map()
     message_id = channel_to_message.get(interaction.channel.id)
     if message_id is None:
         return []
@@ -130,7 +135,7 @@ async def game_list_autocomplete(
 @app_commands.describe(game="Игра")
 @app_commands.autocomplete(game=game_list_autocomplete)
 async def delete(interaction: discord.Interaction, game: str):
-    channel_to_message = get_game_list_channel_to_message_map()
+    channel_to_message = await get_game_list_channel_to_message_map()
 
     if interaction.channel is None:
         await interaction.response.send_message("Команда не доступна в этом канале (#1)", ephemeral=True)

src/modules/scheduler_sync/synchronizer.py

@@ -2,7 +2,10 @@ from asyncio import sleep
 import logging
 from datetime import datetime
 
-from core.config import config, TwitchConfig
+from core.config import config
+
+from domain.streamers import TwitchConfig
+from repositories.streamers import StreamerConfigRepository
 
 from .twitch_events import get_twitch_events, TwitchEvent
 from .discord_events import (
@@ -101,7 +104,9 @@ async def start_synchronizer():
 
     while True:
         try:
-            for streamer in config.STREAMERS:
+            streamers = await StreamerConfigRepository().all()
+
+            for streamer in streamers:
                 if (integration := streamer.integrations.discord) is None:
                     continue
 
@@ -109,4 +114,4 @@ async def start_synchronizer():
         except Exception as e:
             logging.error(e)
 
-        await sleep(5 * 30)
+        await sleep(5 * 60)

src/modules/stream_notifications/twitch/twitch.py

@@ -8,8 +8,9 @@ from twitchAPI.twitch import Twitch
 from twitchAPI.type import AuthScope
 from twitchAPI.object.eventsub import StreamOnlineEvent, ChannelUpdateEvent
 
-from core.config import config, StreamerConfig
+from core.config import config
 from modules.stream_notifications.notification import notify
+from repositories.streamers import StreamerConfigRepository
 
 from .state import State
 from .token_storage import TokenStorage
@@ -50,19 +51,12 @@ class TwitchService:
 
         return twitch
 
-    def get_streamer_config(self, streamer_id: int) -> StreamerConfig:
-        for streamer in config.STREAMERS:
-            if streamer.twitch.id == streamer_id:
-                return streamer
-
-        raise ValueError(f"Streamer with id {streamer_id} not found")
-
     async def notify_online(self, streamer_id: int):
         current_state = self.state.get(streamer_id)
         if current_state is None:
             raise RuntimeError("State is None")
 
-        streamer = self.get_streamer_config(streamer_id)
+        streamer = await StreamerConfigRepository.get_by_twitch_id(streamer_id)
 
         if streamer.notifications.start_stream is None:
             return
@@ -78,7 +72,7 @@ class TwitchService:
         if (datetime.now() - current_state.last_live_at).seconds >= self.ONLINE_NOTIFICATION_DELAY:
             return
 
-        streamer = self.get_streamer_config(streamer_id)
+        streamer = await StreamerConfigRepository.get_by_twitch_id(streamer_id)
 
         if streamer.notifications.change_category is None:
             return
@@ -154,7 +148,9 @@ class TwitchService:
             message_deduplication_history_length=50
         )
 
-        for streamer in config.STREAMERS:
+        streamers = await StreamerConfigRepository.all()
+
+        for streamer in streamers:
             current_stream = await self.get_current_stream(streamer.twitch.id)
 
             if current_stream:
@@ -173,7 +169,7 @@ class TwitchService:
 
             logger.info("Subscribe to events...")
 
-            for streamer in config.STREAMERS:
+            for streamer in streamers:
                 logger.info(f"Subscribe to events for {streamer.twitch.name}")
                 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)
@@ -184,7 +180,7 @@ class TwitchService:
             while True:
                 await sleep(self.UPDATE_DELAY)
 
-                for streamer in config.STREAMERS:
+                for streamer in streamers:
                     await self._on_stream_online(streamer.twitch.id)
         finally:
             await eventsub.stop()