chore: polished code

Author: ca20110820

projects/Expense-Tracker/app.py

@@ -1,5 +1,3 @@
-import tkinter as tk
-from tkinter import ttk
 from tkinter import messagebox
 import customtkinter as ctk
 
@@ -361,3 +359,9 @@ def on_row_selected(self, e):
             category=values[4],
             subcategory=values[5]
         )
+
+
+if __name__ == '__main__':
+    DB_PATH = str(Path(__file__).resolve().parent / 'items.json')
+    app = App(DB_PATH)
+    app.mainloop()

projects/Expense-Tracker/expense_income_stats.py

@@ -2,7 +2,6 @@
 from item import ItemsDB
 import pandas as pd
 import statistics
-from reportlab.pdfgen import canvas
 
 
 class ExpenseIncomeStats:

projects/Expense-Tracker/gui_widgets.py

@@ -1,10 +1,8 @@
 import tkinter as tk
 from tkinter import ttk
-import customtkinter as ctk
-import os
 from pathlib import Path
 
-from item import ItemsDB, Item, Category
+from item import ItemsDB
 from expense_income_stats import ExpenseIncomeStats
 
 
@@ -29,53 +27,8 @@ def __init__(self, parent, items_db: ItemsDB, *args, **kwargs):
         self._parent = parent
         self._items_db = items_db
 
-        # For Testing
-        self._test_add_items_to_db()
         self.load_items()
 
-    def _test_add_items_to_db(self):
-        items = \
-            [Item.create_income_item('Bitcoin',
-                                     500,
-                                     "Income from Bitcoin",
-                                     '2023-06-25',
-                                     Category('Personal Finance', 'Investing')
-                                     ),
-             Item.create_income_item('Youtube Ads',
-                                     5,
-                                     "Income from Youtube Ads",
-                                     '2024-1-10',
-                                     Category('Youtube')
-                                     ),
-             Item.create_income_item('Gift',
-                                     30,
-                                     "Gift from Relatives",
-                                     '2024-03-04'
-                                     ),
-             Item.create_expense_item('Pizza',
-                                      50,
-                                      "Pizza from Pizza Hut",
-                                      '2023-04-20',
-                                      Category('Food', 'Junk')
-                                      ),
-             Item.create_expense_item('Bus',
-                                      20,
-                                      "Travel Expenses by Bus",
-                                      '2024-02-05',
-                                      Category('Transportation')
-                                      ),
-             Item.create_expense_item('Bitcoin',
-                                      -300,
-                                      "Loss on Bitcoin Gambling.",
-                                      '2024-04-20'
-                                      ),
-             ]
-        self._items_db.insert_items(items)
-
-    def clear_test(self):
-        if os.path.exists(self._items_db.db_path):
-            os.remove(self._items_db.db_path)
-
     def clear_items(self):
         # Clearing all the rows in Items Table (Not Deleting!)
         self.delete(*self.get_children())

projects/Expense-Tracker/main.py

@@ -0,0 +1,65 @@
+from pathlib import Path
+import os
+
+from app import App
+from item import ItemsDB, Item, Category
+
+DB_PATH = str(Path(__file__).resolve().parent / 'items.json')
+
+
+def mock_add_items():
+    items = \
+        [Item.create_income_item('Bitcoin',
+                                 500,
+                                 "Income from Bitcoin",
+                                 '2023-06-25',
+                                 Category('Personal Finance', 'Investing')
+                                 ),
+         Item.create_income_item('Youtube Ads',
+                                 5,
+                                 "Income from Youtube Ads",
+                                 '2024-1-10',
+                                 Category('Youtube')
+                                 ),
+         Item.create_income_item('Gift',
+                                 30,
+                                 "Gift from Relatives",
+                                 '2024-03-04'
+                                 ),
+         Item.create_expense_item('Pizza',
+                                  50,
+                                  "Pizza from Pizza Hut",
+                                  '2023-04-20',
+                                  Category('Food', 'Junk')
+                                  ),
+         Item.create_expense_item('Bus',
+                                  20,
+                                  "Travel Expenses by Bus",
+                                  '2024-02-05',
+                                  Category('Transportation')
+                                  ),
+         Item.create_expense_item('Bitcoin',
+                                  -300,
+                                  "Loss on Bitcoin Gambling.",
+                                  '2024-04-20'
+                                  ),
+         ]
+    ItemsDB(DB_PATH).insert_items(items)
+
+
+def mock_clear_test():
+    if os.path.exists(DB_PATH):
+        os.remove(DB_PATH)
+
+    if os.path.exists(str(Path(__file__).resolve().parent / 'report.xlsx')):
+        os.remove(str(Path(__file__).resolve().parent / 'report.xlsx'))
+
+    if os.path.exists(str(Path(__file__).resolve().parent / 'report.pdf')):
+        os.remove(str(Path(__file__).resolve().parent / 'report.pdf'))
+
+
+if __name__ == '__main__':
+    mock_add_items()
+    app = App(DB_PATH)
+    app.mainloop()
+    mock_clear_test()