products.py

from storage import load_products, save_products
from utils import pause, get_menu_choice, get_positive_number, get_non_negative_int

def product_menu():
    while True:
        print("\nProduct Management")
        print("-" * 18)
        print("1. Add Product")
        print("2. View Products")
        print("3. Delete Product")
        print("4. Back to Main Menu")

        choice = get_menu_choice(1, 4)

        if choice == 1:
            add_product()
        elif choice == 2:
            view_products()
        elif choice == 3:
            delete_product()
        else:
            break

def add_product():
    df = load_products()

    product_id = input("Enter Product ID: ")
    if product_id in df["ID"].values:
        print("Product ID already exists")
        pause()
        return

    name = input("Enter Product Name: ")
    category = input("Enter Category: ")
    price = get_positive_number("Enter Price: ")
    qty = get_non_negative_int("Enter Quantity: ")

    df.loc[len(df)] = [product_id, name, category, price, qty]
    save_products(df)

    print("Product added successfully")
    pause()

def view_products():
    df = load_products()
    if df.empty:
        print("\nNo products found.")
    else:
        print("\nPRODUCT LIST")
        print("-" * 60)
        print(f"{'ID':<8}{'Name':<15}{'Category':<15}{'Price':<10}{'Qty'}")
        print("-" * 60)
        for _, row in df.iterrows():
            print(f"{row.ID:<8}{row.Name:<15}{row.Category:<15}{row.Price:<10}{row.Quantity}")
    pause()

def delete_product():
    df = load_products()
    df["ID"] = df["ID"].astype(str)

    pid = input("Enter Product ID to delete: ").strip()

    if pid not in df["ID"].values:
        print(" Product not found")
        pause()
        return

    confirm = input("Are you sure you want to delete this product? (y/n): ").lower()
    if confirm == "y":
        df = df[df["ID"] != pid]
        save_products(df)
        print("Product deleted successfully")
    else:
        print("Cancelled")

    pause()