#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
获取股票价格和标普500指数（最终版）
"""

import requests
import json
from datetime import datetime

# Yahoo Finance API 配置
YAHOO_API_URL = "https://query1.finance.yahoo.com/v8/finance/chart/"

# 指数配置
INDEXES = {
    "^GSPC": "标普500",
    "^IXIC": "纳斯达克",
    "^DJI": "道琼斯"
}

# 股票配置（从Notion数据库读取）
STOCKS = [
    {"symbol": "MSFT", "name": "Microsoft Corporation", "quantity": 1, "cost_basis": 439.00}
]

def get_stock_price(symbol):
    """获取股票实时价格"""
    
    try:
        url = f"{YAHOO_API_URL}{symbol}"
        params = {
            "interval": "1d",
            "range": "1d"
        }
        
        headers = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
        }
        
        response = requests.get(url, params=params, headers=headers, timeout=10)
        
        if response.status_code == 200:
            data = response.json()
            
            # 检查响应结构
            if "chart" in data and "result" in data["chart"]:
                result = data["chart"]["result"]
                if result and len(result) > 0:
                    meta = result[0].get("meta", {})
                    current_price = meta.get("regularMarketPrice")
                    previous_close = meta.get("previousClose")
                    
                    if current_price and previous_close:
                        change = current_price - previous_close
                        change_percent = (change / previous_close) * 100
                        
                        return {
                            "symbol": symbol,
                            "price": current_price,
                            "previous_close": previous_close,
                            "change": change,
                            "change_percent": change_percent,
                            "currency": meta.get("currency", "USD")
                        }
        
        return None
        
    except Exception as e:
        print(f"获取 {symbol} 价格失败: {e}")
        return None

def get_index_prices():
    """获取指数价格"""
    
    print("=" * 60)
    print("获取指数价格")
    print("=" * 60)
    
    indexes = {}
    
    for symbol, name in INDEXES.items():
        price_data = get_stock_price(symbol)
        if price_data:
            indexes[symbol] = {
                "name": name,
                "price": price_data["price"],
                "change": price_data["change"],
                "change_percent": price_data["change_percent"]
            }
            print(f"{name}: {price_data['price']:.2f} ({price_data['change_percent']:+.2f}%)")
        else:
            print(f"{name}: 获取失败")
    
    print("=" * 60)
    return indexes

def get_stock_prices():
    """获取持仓股票价格"""
    
    print("=" * 60)
    print("获取持仓股票价格")
    print("=" * 60)
    
    stock_prices = {}
    
    for stock in STOCKS:
        symbol = stock["symbol"]
        price_data = get_stock_price(symbol)
        if price_data:
            stock_prices[symbol] = {
                "name": stock["name"],
                "quantity": stock["quantity"],
                "cost_basis": stock["cost_basis"],
                "current_price": price_data["price"],
                "change": price_data["change"],
                "change_percent": price_data["change_percent"]
            }
            print(f"{symbol}: ${price_data['price']:.2f} ({price_data['change_percent']:+.2f}%)")
        else:
            print(f"{symbol}: 获取失败")
    
    print("=" * 60)
    return stock_prices

def main():
    """主函数"""
    print("开始获取股票价格和指数...")
    print()
    
    # 获取指数价格
    indexes = get_index_prices()
    
    # 获取持仓股票价格
    stock_prices = get_stock_prices()
    
    # 打印汇总
    print("\n📊 数据汇总:")
    print("=" * 60)
    
    if indexes:
        print(f"\n指数数据: {len(indexes)} 个")
        for symbol, data in indexes.items():
            print(f"  {data['name']}: {data['price']:.2f} ({data['change_percent']:+.2f}%)")
    
    if stock_prices:
        print(f"\n持仓股票: {len(stock_prices)} 个")
        for symbol, data in stock_prices.items():
            print(f"  {data['name']}: ${data['current_price']:.2f} ({data['change_percent']:+.2f}%)")
    
    print("=" * 60)

if __name__ == "__main__":
    main()