371 lines
10 KiB
Python
371 lines
10 KiB
Python
from core.tracker import Tracker
|
|
from config import ADMIN_PASSWORD
|
|
|
|
|
|
class CLI:
|
|
def __init__(self, tracker) -> None:
|
|
self.tracker = tracker
|
|
self.current_user = None
|
|
|
|
def run(self):
|
|
while True:
|
|
self._clear_screen()
|
|
self._show_main_menu()
|
|
|
|
def _clear_screen(self):
|
|
import os
|
|
|
|
os.system("cls" if os.name == "nt" else "clear")
|
|
|
|
def _show_main_menu(self):
|
|
print("=" * 50)
|
|
print("DEBT TRACKER")
|
|
print("=" * 50)
|
|
|
|
if self.current_user:
|
|
user = self.tracker.get_user_by_id(self.current_user)
|
|
debt = self.tracker.get_user_debt(self.current_user)
|
|
print(f"\nLogged in as: {user.name}")
|
|
print(f"Current Debt: ${debt:.2f}\n")
|
|
|
|
print("1. Select user")
|
|
print("2. View all debts")
|
|
print("3. Admin panel")
|
|
print("4. Exit")
|
|
print()
|
|
|
|
choice = input("Choose Option: ").strip()
|
|
|
|
if choice == "1":
|
|
self._select_user()
|
|
elif choice == "2":
|
|
self._view_all_debts()
|
|
elif choice == "3":
|
|
self._admin_panel()
|
|
elif choice == "4":
|
|
print("\nGoodbye!")
|
|
exit()
|
|
else:
|
|
self._invalid_choice()
|
|
|
|
def _select_user(self):
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("SELECT USER")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
users = self.tracker.get_all_users()
|
|
user_list = list(users.items())
|
|
|
|
for i, (user_id, user) in enumerate(user_list, 1):
|
|
debt = self.tracker.get_user_debt(user_id)
|
|
print(f"{i}. {user.name} (Debt: ${debt:.2f})")
|
|
|
|
print(f"{len(user_list) + 1}. Back to main menu")
|
|
print()
|
|
|
|
try:
|
|
choice = int(input("Choose user: ").strip())
|
|
|
|
if choice == len(user_list) + 1:
|
|
return
|
|
|
|
if 1 <= choice <= len(user_list):
|
|
self.current_user = user_list[choice - 1][0]
|
|
self._user_menu()
|
|
else:
|
|
self._invalid_choice()
|
|
except ValueError:
|
|
self._invalid_choice()
|
|
|
|
def _user_menu(self):
|
|
while True:
|
|
self._clear_screen()
|
|
user = self.tracker.get_user_by_id(self.current_user)
|
|
debt = self.tracker.get_user_debt(self.current_user)
|
|
|
|
print("=" * 50)
|
|
print(f"USER MENU - {user.name}")
|
|
print("=" * 50)
|
|
print(f"Current debt: ${debt:.2f}\n")
|
|
|
|
print("1. Add purchase")
|
|
print("2. View my debt")
|
|
print("3. Logout")
|
|
print()
|
|
|
|
choice = input("Choose option: ").strip()
|
|
|
|
if choice == "1":
|
|
self._add_purchase()
|
|
elif choice == "2":
|
|
self._view_user_debt()
|
|
elif choice == "3":
|
|
self.current_user = None
|
|
break
|
|
else:
|
|
self._invalid_choice()
|
|
|
|
def _add_purchase(self):
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("ADD PURCHASE")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
menu_items = self.tracker.get_items()
|
|
item_list = list(menu_items.items())
|
|
|
|
for i, (item_id, item) in enumerate(item_list, 1):
|
|
print(f"{i}. {item.name} - ${item.price:.2f}")
|
|
|
|
print(f"{len(item_list) + 1}. Cancel")
|
|
print()
|
|
|
|
try:
|
|
choice = int(input("Choose item: ").strip())
|
|
|
|
if choice == len(item_list) + 1:
|
|
return
|
|
|
|
if 1 <= choice <= len(item_list):
|
|
selected_item_id = item_list[choice - 1][0]
|
|
self._confirm_purchase(selected_item_id)
|
|
else:
|
|
self._invalid_choice()
|
|
except ValueError:
|
|
self._invalid_choice()
|
|
|
|
def _confirm_purchase(self, item_id):
|
|
self._clear_screen()
|
|
|
|
user = self.tracker.get_user_by_id(self.current_user)
|
|
item = self.tracker.get_items()[item_id]
|
|
|
|
print("=" * 50)
|
|
print("CONFIRM PURCHASE")
|
|
print("=" * 50)
|
|
print()
|
|
print(f"User: {user.name}")
|
|
print(f"Item: {item.name}")
|
|
print(f"Price: ${item.price:.2f}")
|
|
print()
|
|
|
|
confirm = input("Add to debt? (y/n): ").strip().lower()
|
|
|
|
if confirm == "y":
|
|
result = self.tracker.add_purchase(self.current_user, item_id)
|
|
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("PURCHASE ADDED")
|
|
print("=" * 50)
|
|
print()
|
|
print(f"✓ {result['message']}")
|
|
print(f"New debt: ${result['new_debt']:.2f}")
|
|
print()
|
|
|
|
input("Press Enter to continue...")
|
|
else:
|
|
print("Purchase cancelled.")
|
|
input("Press Enter to continue...")
|
|
|
|
def _view_user_debt(self):
|
|
self._clear_screen()
|
|
|
|
user = self.tracker.get_user_by_id(self.current_user)
|
|
debt = self.tracker.get_user_debt(self.current_user)
|
|
|
|
print("=" * 50)
|
|
print(f"DEBT - {user.name}")
|
|
print("=" * 50)
|
|
print()
|
|
print(f"Amount owed: ${debt:.2f}")
|
|
print()
|
|
|
|
input("Press Enter to continue...")
|
|
|
|
def _view_all_debts(self):
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("ALL DEBTS")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
all_debts = self.tracker.get_all_debts()
|
|
|
|
if not all_debts:
|
|
print("No users with debt.")
|
|
else:
|
|
total = 0
|
|
for user_id, debt_info in all_debts.items():
|
|
amount = debt_info["debt"]
|
|
print(f"{debt_info['name']:.<30} ${amount:>7.2f}")
|
|
total += amount
|
|
|
|
print("-" * 50)
|
|
print(f"{'TOTAL':.<30} ${total:>7.2f}")
|
|
|
|
print()
|
|
input("Press Enter to continue...")
|
|
|
|
def _admin_panel(self):
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("ADMIN PANEL")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
password = input("Enter admin password: ").strip()
|
|
|
|
if password != ADMIN_PASSWORD:
|
|
print("Invalid password!")
|
|
input("Press Enter to continue...")
|
|
return
|
|
|
|
while True:
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("ADMIN PANEL")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
print("1. View all debts")
|
|
print("2. Adjust user debt")
|
|
print("3. Clear user debt")
|
|
print("4. View transaction history")
|
|
print("5. Logout")
|
|
print()
|
|
|
|
choice = input("Choose option: ").strip()
|
|
|
|
if choice == "1":
|
|
self._view_all_debts()
|
|
elif choice == "2":
|
|
self._adjust_debt()
|
|
elif choice == "3":
|
|
self._clear_user_debt()
|
|
elif choice == "4":
|
|
self._view_transaction_history()
|
|
elif choice == "5":
|
|
break
|
|
else:
|
|
self._invalid_choice()
|
|
|
|
def _adjust_debt(self):
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("ADJUST DEBT")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
users = self.tracker.get_all_users()
|
|
user_list = list(users.items())
|
|
|
|
for i, (user_id, user) in enumerate(user_list, 1):
|
|
debt = self.tracker.get_user_debt(user_id)
|
|
print(f"{i}. {user.name} - ${debt:.2f}")
|
|
|
|
print()
|
|
|
|
try:
|
|
choice = int(input("Choose user: ").strip())
|
|
|
|
if 1 <= choice <= len(user_list):
|
|
user_id = user_list[choice - 1][0]
|
|
|
|
amount_str = input("Amount paid: $").strip()
|
|
amount = float(amount_str)
|
|
|
|
result = self.tracker.adjust_debt(user_id, amount, ADMIN_PASSWORD)
|
|
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("DEBT ADJUSTED")
|
|
print("=" * 50)
|
|
print()
|
|
print(f"{result['message']}")
|
|
print(f"New debt: ${result['new_debt']:.2f}")
|
|
print()
|
|
|
|
input("Press Enter to continue...")
|
|
else:
|
|
self._invalid_choice()
|
|
except ValueError:
|
|
print("Invalid input!")
|
|
input("Press Enter to continue...")
|
|
|
|
def _clear_user_debt(self):
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("CLEAR USER DEBT")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
users = self.tracker.get_all_users()
|
|
user_list = list(users.items())
|
|
|
|
for i, (user_id, user) in enumerate(user_list, 1):
|
|
debt = self.tracker.get_user_debt(user_id)
|
|
print(f"{i}. {user.name} - ${debt:.2f}")
|
|
|
|
print()
|
|
|
|
try:
|
|
choice = int(input("Choose user: ").strip())
|
|
|
|
if 1 <= choice <= len(user_list):
|
|
user_id = user_list[choice - 1][0]
|
|
debt_amount = self.tracker.get_user_debt(user_id)
|
|
|
|
confirm = input(f"Clear ${debt_amount:.2f}? (y/n): ").strip().lower()
|
|
|
|
if confirm == "y":
|
|
result = self.tracker.adjust_debt(
|
|
user_id, debt_amount, ADMIN_PASSWORD
|
|
)
|
|
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("DEBT CLEARED")
|
|
print("=" * 50)
|
|
print()
|
|
print(
|
|
f"Debt cleared for {self.tracker.get_user_by_id(user_id).name}"
|
|
)
|
|
print()
|
|
|
|
input("Press Enter to continue...")
|
|
else:
|
|
self._invalid_choice()
|
|
except ValueError:
|
|
print("Invalid input!")
|
|
input("Press Enter to continue...")
|
|
|
|
def _view_transaction_history(self):
|
|
self._clear_screen()
|
|
print("=" * 50)
|
|
print("TRANSACTION HISTORY")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
transactions = self.tracker.transactions
|
|
|
|
if not transactions:
|
|
print("No transactions yet.")
|
|
else:
|
|
for t in sorted(transactions, key=lambda x: x.timestamp, reverse=True):
|
|
user = self.tracker.get_user_by_id(t.user_id)
|
|
item = self.tracker.get_items()[t.item_id]
|
|
print(
|
|
f"{t.timestamp[:10]} | {user.name:.<20} {item.name:.<15} ${t.amount:.2f}"
|
|
)
|
|
|
|
print()
|
|
input("Press Enter to continue...")
|
|
|
|
def _invalid_choice(self):
|
|
print("Invalid choice. Please try again.")
|
|
input("Press Enter to continue...")
|