Add undo command

BrigadaSOS · Jan 28, 2024 · 31fd3e5 · 31fd3e5
1 parent 875e1d3
commit 31fd3e5
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 6 deletions.
diff --git a/db/prod.db b/db/prod.db
diff --git a/immersion_bot/cogs/undo.py b/immersion_bot/cogs/undo.py
@@ -1,10 +1,14 @@
+import ast
 import os
+from typing import Optional
 
 import discord
 from discord import app_commands
 from discord.ext import commands
 from dotenv import load_dotenv
 
+import helpers
+from sql import Store
 
 #############################################################
 
@@ -26,14 +30,35 @@ async def on_ready(self):
         self.guild = self.bot.get_guild(GUILD_ID)
 
     @app_commands.command(name="undo", description="Deshacer el último log creado.")
-    async def undo(
-        self,
-        interaction: discord.Interaction,
-    ):
+    @app_commands.describe(
+        cantidad="Número registros a borrar de una vez (Máximo 5). CUALQUIER DATO ELIMINADO NO SE PUEDE RECUPERAR, ÚSELO CON PRECAUCIÓN."
+    )
+    async def undo(self, interaction: discord.Interaction, cantidad: Optional[int]):
         await interaction.response.defer()
-        return await interaction.edit_original_response(
-            content="Este comando aún no está implementado."
+
+        records_to_delete = cantidad if cantidad else 1
+        if records_to_delete > 5:
+            return await interaction.edit_original_response(
+                content="Por precaución no se pueden deshacer más de 5 registros de una vez."
+            )
+
+        store = Store(_DB_NAME)
+        deleted_rows = store.delete_last_log_user(
+            interaction.user.id, records_to_delete
         )
+        print(deleted_rows)
+
+        if len(deleted_rows) > 0:
+            bodyMessage = "### Se han eliminado los siguientes registros:\n"
+            for row in deleted_rows:
+                note = ast.literal_eval(row.note)
+                bodyMessage += f"* **[{row.created_at.strftime('%Y-%m-%d')}]** {row.media_type.value} {note[0]} - {round(row.amount, 2)} {helpers.media_type_format(row.media_type.value, (row.amount > 1))} - {row.time} minutos - {helpers._to_amount(row.media_type.value, row.amount)} puntos\n"
+                if note[1]:
+                    bodyMessage += f"\t* {note[1]}"
+        else:
+            bodyMessage = "No se ha encontrado ningún registro a eliminar."
+
+        return await interaction.edit_original_response(content=bodyMessage)
 
 
 #         message_link = message_link.split('/')

diff --git a/immersion_bot/sql.py b/immersion_bot/sql.py
@@ -144,6 +144,24 @@ def get_latest_content_by_user_autocomplete(
         print(content_results)
         return content_results
 
+    def delete_last_log_user(self, discord_user_id, limit):
+        query = f"""
+             DELETE FROM logs
+                    WHERE ROWID IN (SELECT ROWID
+                                    FROM logs WHERE discord_user_id == '{discord_user_id}'
+                                    ORDER BY created_at DESC
+                                    LIMIT {limit})
+                    RETURNING *
+        """
+
+        print(query)
+
+        cursor = self.conn.cursor()
+        cursor.execute(query)
+        deleted_rows = cursor.fetchall()
+        self.conn.commit()
+        return deleted_rows
+
     def get_logs_by_user(self, discord_user_id, media_type, timeframe):
         if media_type == None and timeframe == None:
             where_clause = f"discord_user_id={discord_user_id}"