Merge pull request #633 from Gabriela20103967/issue-632

feat: Add a Delete Button in Budget-manager.

Author: Mrinank-Bhowmick

projects/Budget-manager/main.py

@@ -18,8 +18,11 @@
 conn.commit()
 
 
-# Function to add a transaction to the database
+# Function to add a transaction to the database.
 def add_transaction():
+    """
+    Adds a transaction to the database based on user input.
+    """
     date = date_entry.get()
     description = description_entry.get()
     amount = amount_entry.get()
@@ -39,25 +42,52 @@ def add_transaction():
         messagebox.showwarning("Warning", "Please fill in all fields.")
 
 
-# Function to clear input fields
+def delete_transaction():
+    """
+    Deletes a selected transaction from the database.
+    """
+    selected_index = transaction_listbox.curselection()
+    if selected_index:
+        transaction_id = transaction_listbox.get(selected_index)[0]
+        c.execute("DELETE FROM transactions WHERE id=?", (transaction_id,))
+        conn.commit()
+        update_transaction_list()
+        update_balance()
+        messagebox.showinfo("Success", "Transaction deleted successfully!")
+    else:
+        messagebox.showwarning("Warning", "Please select a transaction to delete")
+
+
+# Function to clear input fields.
+
 def clear_entries():
+    """
+    Clears all input fields.
+    """
     date_entry.delete(0, tk.END)
     description_entry.delete(0, tk.END)
     amount_entry.delete(0, tk.END)
     category_combobox.set("")
+    category_combobox.config(state="readonly")
 
 
-# Function to update the transaction list
+# Function to update the transaction list.
 def update_transaction_list():
+    """
+    Updates the displayed transaction list based on the database entries.
+    """
     transaction_listbox.delete(0, tk.END)
     c.execute("SELECT * FROM transactions")
     transactions = c.fetchall()
     for transaction in transactions:
         transaction_listbox.insert(tk.END, transaction)
 
 
-# Function to calculate and display the current balance
+# Function to calculate and display the current balance.
 def update_balance():
+    """
+    Calculates and displays the current balance based on income and expenses.
+    """
     c.execute("SELECT SUM(amount) FROM transactions")
     total_income = c.fetchone()[0]
     if total_income is None:
@@ -70,11 +100,15 @@ def update_balance():
     balance_label.config(text=f"Current Balance: ${balance:.2f}")
 
 
-# Create the main window
+# Create the main window.
 root = tk.Tk()
 root.title("Personal Budget Manager")
+root.columnconfigure(1, weight=1)
+root.columnconfigure(3, weight=1)
+root.columnconfigure(5, weight=1)
 
-# Create and configure widgets
+# Create and configure widgets.
+delete_button = tk.Button(root, text="Delete Transaction", command=delete_transaction)
 date_label = tk.Label(root, text="Date:")
 date_entry = tk.Entry(root, width=15)
 description_label = tk.Label(root, text="Description:")
@@ -99,7 +133,8 @@ def update_balance():
 transaction_listbox = tk.Listbox(root, width=50)
 balance_label = tk.Label(root, text="Current Balance: $0.00")
 
-# Place widgets in the window
+# Place widgets in the window.
+delete_button.grid(row=0, column=9, padx=10, pady=10)
 date_label.grid(row=0, column=0, padx=10, pady=10)
 date_entry.grid(row=0, column=1, padx=10, pady=10)
 description_label.grid(row=0, column=2, padx=10, pady=10)
@@ -112,7 +147,7 @@ def update_balance():
 transaction_listbox.grid(row=1, column=0, columnspan=9, padx=10, pady=10)
 balance_label.grid(row=2, column=0, columnspan=9, padx=10, pady=10)
 
-# Initialize the transaction list and balance
+# Initialize the transaction list and balance.
 update_transaction_list()
 update_balance()
 

projects/Budget-manager/test_DeleteButton.py

@@ -0,0 +1,29 @@
+import unittest
+
+
+class Budget:
+    def __init__(self):
+        self.transactions = []
+
+    def add_transaction(self, transaction):
+        self.transactions.append(transaction)
+
+    def delete_transaction(self, transaction):
+        self.transactions.remove(transaction)
+
+class TestBudgetFunctionality(unittest.TestCase):
+    def test_delete_transaction(self):
+        # Arrange
+        budget = Budget()
+        transaction_to_delete = {"date": "2023-01-01", "description": "Test Transaction", "amount": 100.0, "category": "Test"}
+        # Act
+        budget.add_transaction(transaction_to_delete)
+        budget.delete_transaction(transaction_to_delete)
+
+        # Assert ()
+        self.assertNotIn(transaction_to_delete, budget.transactions)
+
+if __name__ == '__main__':
+    unittest.main()
+
+