add: custom ids on interval buttons

Author: mystique09

config/dev-config-void.yml

@@ -0,0 +1,23 @@
+bot:
+  prefix: "pph-"
+  token: !ENV "token"
+  staff_roles: [ 748554755265134635, 809311700560904201, 1064179977639165952, 1062998739012296795 ]
+
+database:
+  name: !ENV "db_name"
+  host: !ENV "db_host"
+  user: !ENV "db_user"
+  password: !ENV "db_pass"
+  port: !ENV "db_port"
+
+api:
+  api_ninja: !ENV "api_ninja_key"
+
+guild:
+  staff_roles: [ 748554755265134635, 809311700560904201, 1064179977639165952, 1062998739012296795 ]
+  dev_help_forum: 1260076443505786993
+  log_channel: 1076405709916344360
+
+logger:
+  log_channel: 1076405709916344360
+  log_level: "DEBUG"

pyrightconfig.json

@@ -0,0 +1,4 @@
+{
+  "venvPath": ".",
+  "venv": "C:/Users/benji/AppData/Local/pypoetry/Cache/virtualenvs/progphilbot-QxuRZnFx-py3.12"
+}

src/bot/main.py

@@ -85,7 +85,7 @@ async def load_cogs(self, module: str, cogs: list[str]) -> None:
         :param cogs: list of cogs to load, basically the files under the cogs/<category> that ends with .py
         """
         for cog in cogs:
-            if cog.startswith("__init__"):
+            if cog.startswith("__init__") or cog.startswith("currency"):
                 continue
             if cog.endswith(".py"):
                 await self.load_extension(f"src.cogs.{module}.{cog[:-3]}")
@@ -129,7 +129,7 @@ def migrate_db(db: Database, logger: Logger) -> None:
 
 
 async def main():
-    config = get_config("config/dev-config.yml")
+    config = get_config("config/dev-config-void.yml")
     logger_config = config.logger
     logger = BotLogger(logger_config)
     logger.add_handler(StreamHandler())

src/ui/views/forum_showcase.py

@@ -111,12 +111,12 @@ def __init__(self,
         self.logger = logger
         self.selected_interval: str | None = None
 
-    @button(label="Skip", style=ButtonStyle.red)  # type: ignore
+    @button(label="Skip", style=ButtonStyle.red, custom_id="cancel-interval-select")  # type: ignore
     async def cancel_button(self, interaction: Interaction, button: Button):
         await interaction.response.send_message("Skipped!", ephemeral=True)
         self.stop()
 
-    @button(label="Submit", style=ButtonStyle.green) # type: ignore
+    @button(label="Submit", style=ButtonStyle.green, custom_id="submit-interval-select") # type: ignore
     async def submit_button(self, interaction: Interaction, button: Button):
         if self.selected_interval is None:
             await interaction.response.send_message(

src/utils/scheduler.py

@@ -59,8 +59,9 @@ def __init__(
         self._day = day
         self._task = task
 
-    async def run(self) -> None:
+    def run(self) -> None:
         self._spawn_loop.start()
+        print(f"Schedule {self._task} started")
 
     @loop()
     async def _spawn_loop(self):
@@ -132,33 +133,43 @@ def cancel(self) -> None:
 class Scheduler:
     schedules: dict[str, Schedule]
 
-    def __init__(self) -> None:
-        self.schedules = {}
+    def __init__(self, schedules: dict[str, Schedule]) -> None:
+        self.schedules = schedules
+
+    def has(self, key: str) -> bool:
+        return key in self.schedules
 
     async def spawn_schedule(self, key: str, schedule: Schedule):
-        if key in self.schedules:
+        if self.has(key):
             raise KeyError(f"Schedule with key {key} already exists")
         self.schedules[key] = schedule
 
-        await schedule.run()
+        schedule.run()
 
     def cancel_schedule(self, key: str) -> None:
-        if key not in self.schedules:
+        if not self.has(key):
             raise KeyError(f"Schedule with key {key} does not exist")
 
         if self.schedules[key].is_running():
             self.schedules[key].cancel()
 
     async def resume_schedule(self, key: str) -> None:
-        if key not in self.schedules:
+        if not self.has(key):
             raise KeyError(f"Schedule with key {key} does not exist")
 
         if not self.schedules[key].is_running():
-            await self.schedules[key].run()
+            self.schedules[key].run()
 
     def remove_schedule(self, key: str) -> None:
-        if key not in self.schedules:
+        if not self.has(key):
             raise KeyError(f"Schedule with key {key} does not exist")
 
         self.cancel_schedule(key)
         self.schedules.pop(key)
+
+    def restart_schedule(self, key: str) -> None:
+        if not self.has(key):
+            raise KeyError(f"Schedule with key {key} does not exist")
+
+        self.schedules[key].cancel()
+        self.schedules[key].reschedule()