Changing the project to being called Nessa now

2024-05-05 16:38:32 -04:00
parent e5a189d9cb
commit ce1d4626a8
12 changed files with 105 additions and 51 deletions
--- a/nessa/confirm.py
+++ b/nessa/confirm.py
@@ -0,0 +1,48 @@
+from discord.ui import View, Button
+
+class ConfirmView(View):
+    def __init__(self, project_id, timeout=180):  # 180 seconds until the view expires
+        super().__init__(timeout=timeout)
+        self.project_id = project_id
+        self.value = None
+
+    @discord.ui.button(label="Confirm", style=discord.ButtonStyle.danger)
+    async def confirm(self, interaction: discord.Interaction, button: Button):
+        # Attempt to remove the project
+        try:
+            await remove_project(self.project_id)
+            await interaction.response.edit_message(content=f"Project ID {self.project_id} and all associated tasks have been successfully removed.", view=None)
+            logger.info(f"Project ID {self.project_id} and all associated tasks removed successfully.")
+        except Exception as e:
+            error_message = f"Failed to remove the project. Error: {e}"
+            await interaction.response.edit_message(content=error_message, view=None)
+            logger.error(f"Error in remove_project_command: {e}")
+            self.stop()  # Stopping the view after an error is usually a good practice to clean up the UI.
+
+    @discord.ui.button(label="Cancel", style=discord.ButtonStyle.secondary)
+    async def cancel(self, interaction: discord.Interaction, button: Button):
+        await interaction.response.edit_message(content="Project removal cancelled.", view=None)
+        self.stop()
+
+class ConfirmTaskDeletionView(View):
+    def __init__(self, task_id, timeout=180):  # 180 seconds until the view expires
+        super().__init__(timeout=timeout)
+        self.task_id = task_id
+        self.value = None
+
+    @discord.ui.button(label="Confirm", style=discord.ButtonStyle.danger)
+    async def confirm(self, interaction: discord.Interaction, button: Button):
+        # Attempt to remove the task
+        try:
+            await remove_task(self.task_id)
+            await interaction.response.edit_message(content=f"Task ID {self.task_id} has been successfully removed.", view=None)
+            logger.info(f"Task ID {self.task_id} removed successfully.")
+        except Exception as e:
+            error_message = f"Failed to remove the task. Error: {e}"
+            await interaction.response.edit_message(content=error_message, view=None)
+            logger.error(f"Error in remove_task_command: {e}")
+
+    @discord.ui.button(label="Cancel", style=discord.ButtonStyle.secondary)
+    async def cancel(self, interaction: discord.Interaction, button: Button):
+        await interaction.response.edit_message(content="Task removal cancelled.", view=None)
+        self.stop()