Added new functions to purchase_logger.py

- Made purchase functions subcommand under main command "purchase"
- Created display and remove commands with basic functionality

Author: iv0rye

purchase_logger.py

@@ -13,7 +13,7 @@ def __init__(self, bot):
         self.database = sqlite3.connect('database.db')
         self.cursor = self.database.cursor()
 
-        self.database.execute("CREATE TABLE IF NOT EXISTS purchases(user_id INT, category STRING, title STRING, description STRING, cost FLOAT)")
+        self.database.execute("CREATE TABLE IF NOT EXISTS purchases(user_id INT, category STRING, title STRING, description STRING, cost FLOAT, timestamp DATETIME DEFAULT CURRENT_TIMESTAMP)")
         self.database.execute("CREATE TABLE IF NOT EXISTS categories(user_id INT, category STRING)")
 
 
@@ -33,7 +33,7 @@ async def add_category(
     ):
         """Adds a category into the category list that is called when using a command involving any purchases"""
 
-        # Fetching all added categories by user in database
+        # Fetching all added categories by user in databasenextcord.errors.ApplicationInvokeError: Command raised an exception: OperationalError: table purchases has 6 columns but 5 values were supplied
         categories = self.fetch_categories(interaction)
 
         if category in categories:
@@ -58,7 +58,7 @@ async def clear_categories(
     ):
         """Removes a category from the category list that is called when using a command involving any purchases"""
         pass
-
+    
 
     @category.subcommand(description="Removes a category from the available list")
     async def remove_category(
@@ -69,8 +69,39 @@ async def remove_category(
         """Removes a category from the category list that is called when using a command involving any purchases"""
         pass
 
+
+    @nextcord.slash_command(guild_ids=[TESTING_GUILD_ID], description="main command")
+    async def purchase(self, interaction: nextcord.Interaction):
+        """
+        This is the main slash command that will be the prefix of all commands below.
+        This will never get called since it has subcommands.
+        """
+
+
+    @purchase.subcommand(description="Lists all purchases")
+    async def display(self, interaction: nextcord.Interaction):
+        query = "SELECT * FROM purchases WHERE user_id = ?"
+        data = self.cursor.execute(query, (interaction.user.id,))
+
+        await interaction.response.send_message(f"{data.fetchall()}")
+
+    @purchase.subcommand(description="Remove latest entry")
+    async def remove(self, interaction: nextcord.Interaction):
+        query = """DELETE FROM purchases
+                WHERE timestamp = (
+                SELECT timestamp
+                FROM purchases
+                WHERE user_id = ?  -- Replace ? with the user_id you want to target
+                ORDER BY timestamp DESC
+                LIMIT 1
+                );"""
+        self.cursor.execute(query, (interaction.user.id,))
+        self.database.commit()
+
+        await interaction.response.send_message("Done!")
 
-    @nextcord.slash_command(description="Logging new entry", guild_ids=[TESTING_GUILD_ID])
+
+    @purchase.subcommand(description="Logging new entry")
     async def add_purchase(
         self,
         interaction: nextcord.Interaction,
@@ -83,8 +114,9 @@ async def add_purchase(
     ):
         """Adds a new purchase into the database"""
 
+        # Check if category is in category DB, if title is present, and if cost is also present.
         # Inserting new purchase into purchase table in database
-        query = "INSERT INTO purchases VALUES (?, ?, ?, ?, ?)"
+        query = "INSERT INTO purchases (user_id, category, title, description, cost) VALUES (?, ?, ?, ?, ?)"
         self.cursor.execute(query, (interaction.user.id, category, title, description, cost, ))
         self.database.commit()