proj2.py

import csv
 
# Hardcoded dictionary of stock prices
STOCK_PRICES = {
    "AAPL": 180,
    "TSLA": 250,
    "GOOGL": 140,
    "AMZN": 175,
    "MSFT": 420,
    "META": 480,
}
 
 
def display_available_stocks():
    print("\nAvailable Stocks and Prices ($):")
    print("-" * 30)
    for stock, price in STOCK_PRICES.items():
        print(f"{stock:<10} ${price}")
    print("-" * 30)
 
 
def get_portfolio():
    """Collect stock names and quantities from the user."""
    portfolio = {}
 
    print("\nEnter the stocks you own. Type 'done' when finished.")
    while True:
        stock = input("\nEnter stock symbol (or 'done' to finish): ").upper().strip()
 
        if stock == "DONE":
            break
 
        if stock not in STOCK_PRICES:
            print(f"'{stock}' not found in our price list. Please choose from the available stocks.")
            continue
 
        try:
            quantity = int(input(f"Enter quantity of {stock} you own: "))
            if quantity < 0:
                print("Quantity cannot be negative. Try again.")
                continue
        except ValueError:
            print("Please enter a valid whole number for quantity.")
            continue
 
        # Add to portfolio (accumulate if stock entered more than once)
        portfolio[stock] = portfolio.get(stock, 0) + quantity
 
    return portfolio
 
 
def calculate_investment(portfolio):
    """Calculate value per stock and total investment."""
    details = []
    total = 0
    for stock, quantity in portfolio.items():
        price = STOCK_PRICES[stock]
        value = price * quantity
        total += value
        details.append((stock, quantity, price, value))
    return details, total
 
 
def display_summary(details, total):
    print("\n" + "=" * 50)
    print("PORTFOLIO SUMMARY")
    print("=" * 50)
    print(f"{'Stock':<10}{'Qty':<8}{'Price($)':<12}{'Value($)':<10}")
    print("-" * 50)
    for stock, quantity, price, value in details:
        print(f"{stock:<10}{quantity:<8}{price:<12}{value:<10}")
    print("-" * 50)
    print(f"Total Investment: ${total:,.2f}")
    print("=" * 50)
 
 
def save_to_file(details, total):
    """Save the portfolio summary to a .txt or .csv file."""
    choice = input("\nSave results to a file? (y/n): ").lower().strip()
    if choice != "y":
        return
 
    file_format = input("Choose format - 'txt' or 'csv': ").lower().strip()
 
    if file_format == "csv":
        filename = "portfolio_summary.csv"
        with open(filename, mode="w", newline="") as f:
            writer = csv.writer(f)
            writer.writerow(["Stock", "Quantity", "Price ($)", "Value ($)"])
            for stock, quantity, price, value in details:
                writer.writerow([stock, quantity, price, value])
            writer.writerow([])
            writer.writerow(["Total Investment", "", "", total])
        print(f"Portfolio saved to {filename}")
 
    elif file_format == "txt":
        filename = "portfolio_summary.txt"
        with open(filename, mode="w") as f:
            f.write("PORTFOLIO SUMMARY\n")
            f.write("=" * 50 + "\n")
            f.write(f"{'Stock':<10}{'Qty':<8}{'Price($)':<12}{'Value($)':<10}\n")
            f.write("-" * 50 + "\n")
            for stock, quantity, price, value in details:
                f.write(f"{stock:<10}{quantity:<8}{price:<12}{value:<10}\n")
            f.write("-" * 50 + "\n")
            f.write(f"Total Investment: ${total:,.2f}\n")
        print(f"Portfolio saved to {filename}")
 
    else:
        print("Invalid format. Skipping file save.")
 
 
def main():
    print("=" * 50)
    print("Welcome to the Stock Portfolio Tracker")
    print("=" * 50)
 
    display_available_stocks()
    portfolio = get_portfolio()
 
    if not portfolio:
        print("\nNo stocks entered. Exiting.")
        return
 
    details, total = calculate_investment(portfolio)
    display_summary(details, total)
    save_to_file(details, total)
 
 
if __name__ == "__main__":
    main()