TheLovinator/discord-reminder-bot
1
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

Make pause button have the correct label

Browse Source
This commit is contained in:
Joakim Hellsén
2025-01-08 21:08:12 +01:00
parent 90b934107b
commit 1b1deb72d9
1 changed files with 24 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
discord_reminder_bot/main.py
View File

@ -301,6 +301,7 @@ class JobManagementView(discord.ui.View):
self.job: Job = job self.job: Job = job
self.scheduler: settings.AsyncIOScheduler = scheduler self.scheduler: settings.AsyncIOScheduler = scheduler
self.add_item(JobSelector(scheduler)) self.add_item(JobSelector(scheduler))
self.update_buttons()
@discord.ui.button(label="Delete", style=discord.ButtonStyle.danger) @discord.ui.button(label="Delete", style=discord.ButtonStyle.danger)
async def delete_button(self, interaction: discord.Interaction, button: Button) -> None: # noqa: ARG002 async def delete_button(self, interaction: discord.Interaction, button: Button) -> None: # noqa: ARG002
@ -416,20 +417,36 @@ class JobManagementView(discord.ui.View):
await interaction.response.send_modal(modal) await interaction.response.send_modal(modal)
@discord.ui.button(label="Pause/Resume", style=discord.ButtonStyle.secondary) @discord.ui.button(label="Pause/Resume", style=discord.ButtonStyle.secondary)
async def pause_button(self, interaction: discord.Interaction, button: Button) -> None: # noqa: ARG002 async def pause_button(self, interaction: discord.Interaction, button: Button) -> None:
"""Pause or resume the job. """Pause or resume the job.
Args: Args:
interaction: The interaction object for the command. interaction: The interaction object for the command.
button: The button that was clicked. button: The button that was clicked.
""" """
if self.job.next_run_time is None: if hasattr(self.job, "next_run_time"):
self.job.resume() if self.job.next_run_time is None:
status = "resumed" logger.info("State: %s", self.job.__getstate__())
self.job.resume()
status = "resumed"
button.label = "Pause"
else:
logger.info("State: %s", self.job.__getstate__())
self.job.pause()
status = "paused"
button.label = "Resume"
else: else:
self.job.pause() status: str = f"What is this? {self.job.__getstate__()}"
status = "paused" button.label = "What?"
await interaction.response.send_message(f"Job '{self.job.name}' has been {status}.", ephemeral=True)
self.update_buttons()
await interaction.response.edit_message(view=self)
msg: str = f"Job '{self.job.name}' has been {status}."
if hasattr(self.job, "next_run_time"):
msg += f"\nNext run time: {self.job.next_run_time} {calculate(self.job)}"
await interaction.followup.send(msg)
def update_buttons(self) -> None: def update_buttons(self) -> None:
"""Update the visibility of buttons based on job status.""" """Update the visibility of buttons based on job status."""