# yfinance_test.py
# A standalone script to diagnose the persistent yfinance issue.

import yfinance as yf
import logging

# Set up a simple logger to see detailed error tracebacks
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s'
)

# A list of tickers to test. One very common one, and two from your logs.
TICKERS_TO_TEST = ["MSFT", "AEBI", "AEHR"]

print("--- Starting YFINANCE Diagnostic Test ---")

for ticker_symbol in TICKERS_TO_TEST:
    print(f"\n--- Testing Ticker: {ticker_symbol} ---")

    # --- Test 1: The Ticker().info method ---
    try:
        logging.info(f"Attempting to create Ticker object and get .info for {ticker_symbol}...")
        ticker_obj = yf.Ticker(ticker_symbol)
        market_cap = ticker_obj.info.get('marketCap')
        if market_cap is not None:
            logging.info(f"SUCCESS: Got market cap for {ticker_symbol}: {market_cap}")
        else:
            logging.warning(f"PARTIAL SUCCESS: .info call for {ticker_symbol} worked, but no market cap was found.")
    except Exception:
        logging.error(f"FAILURE: An error occurred during the Ticker().info call for {ticker_symbol}.", exc_info=True)


    # --- Test 2: The yf.download() method ---
    try:
        logging.info(f"Attempting yf.download() for {ticker_symbol}...")
        data = yf.download(
            ticker_symbol,
            period="2d",
            progress=False,
            auto_adjust=False
        )
        if not data.empty:
            logging.info(f"SUCCESS: yf.download() for {ticker_symbol} returned {len(data)} rows of data.")
        else:
            logging.warning(f"PARTIAL SUCCESS: yf.download() for {ticker_symbol} worked, but returned no data (likely delisted).")
    except Exception:
        logging.error(f"FAILURE: An error occurred during the yf.download() call for {ticker_symbol}.", exc_info=True)

print("\n--- YFINANCE Diagnostic Test Complete ---")