#!/usr/bin/env python3
"""
更新美股持仓数据库 - 批量更新
"""

import requests
import json
from datetime import datetime

# Notion API 配置
NOTION_API_KEY = "ntn_c43902219395mirQBetIfYoww1qKCAF14GBRUQeDee29o2"
DATABASE_ID = "2fb105ad-7873-8175-bbbd-e5b87cf101d9"

# 持仓数据
positions = [
    {"symbol": "BND", "shares": 6.7219, "cost": 501.83},
    {"symbol": "SGOV", "shares": 33.7072, "cost": 3380.48},
    {"symbol": "VOO", "shares": 3.1963, "cost": 1969.46}
]

# 当前价格（从搜索结果获取）
current_prices = {
    "BND": 74.22,
    "SGOV": 100.44,
    "VOO": 636.09
}

def get_current_price(symbol):
    """获取当前价格"""
    return current_prices.get(symbol, 0)

def calculate_pnl(shares, cost, current_price):
    """计算盈亏"""
    market_value = shares * current_price
    total_cost = cost
    pnl_amount = market_value - total_cost
    pnl_percent = (pnl_amount / total_cost * 100) if total_cost > 0 else 0
    return market_value, total_cost, pnl_amount, pnl_percent

def update_notion_page(page_id, symbol, shares, cost, current_price, market_value, pnl_amount, pnl_percent):
    """更新Notion页面"""
    url = f"https://api.notion.com/v1/pages/{page_id}"
    
    headers = {
        "Authorization": f"Bearer {NOTION_API_KEY}",
        "Content-Type": "application/json",
        "Notion-Version": "2022-06-28"
    }
    
    # 获取当前时间（UTC+8）
    now = datetime.utcnow()
    date_str = now.strftime("%Y-%m-%d %H:%M")
    
    data = {
        "properties": {
            "股票代码": {"title": [{"text": {"content": symbol}}]},
            "持仓数量": {"number": shares},
            "买入成本": {"number": cost},
            "当前价格": {"number": current_price},
            "总成本": {"number": cost},
            "当前市值": {"number": market_value},
            "盈亏金额": {"number": round(pnl_amount, 2)},
            "盈亏百分比": {"number": round(pnl_percent, 2)},
            "最后更新": {"date": {"start": date_str}}
        }
    }
    
    response = requests.patch(url, headers=headers, json=data)
    return response.status_code == 200

def find_page_by_symbol(symbol):
    """根据股票代码查找页面ID"""
    url = "https://api.notion.com/v1/databases/" + DATABASE_ID + "/query"
    
    headers = {
        "Authorization": f"Bearer {NOTION_API_KEY}",
        "Content-Type": "application/json",
        "Notion-Version": "2022-06-28"
    }
    
    data = {
        "filter": {
            "property": "股票代码",
            "title": {
                "equals": symbol
            }
        }
    }
    
    response = requests.post(url, headers=headers, json=data)
    
    if response.status_code == 200:
        results = response.json().get("results", [])
        if results:
            return results[0]["id"]
    return None

def main():
    print("开始更新美股持仓数据库...")
    print("=" * 50)
    
    total_market_value = 0
    total_cost = 0
    total_pnl = 0
    
    for position in positions:
        symbol = position["symbol"]
        shares = position["shares"]
        cost = position["cost"]
        
        print(f"\n处理 {symbol}...")
        
        # 获取当前价格
        current_price = get_current_price(symbol)
        print(f"  当前价格: ${current_price:.2f}")
        
        # 计算盈亏
        market_value, total_cost_val, pnl_amount, pnl_percent = calculate_pnl(shares, cost, current_price)
        print(f"  持仓数量: {shares}")
        print(f"  成本: ${cost:.2f}")
        print(f"  市值: ${market_value:.2f}")
        print(f"  盈亏: ${pnl_amount:.2f} ({pnl_percent:.2f}%)")
        
        # 查找页面ID
        page_id = find_page_by_symbol(symbol)
        
        if page_id:
            # 更新现有页面
            success = update_notion_page(page_id, symbol, shares, cost, current_price, market_value, pnl_amount, pnl_percent)
            if success:
                print(f"  ✅ Notion页面更新成功")
            else:
                print(f"  ❌ Notion页面更新失败")
        else:
            print(f"  ⚠️  未找到 {symbol} 的持仓记录")
        
        # 累计数据
        total_market_value += market_value
        total_cost += total_cost_val
        total_pnl += pnl_amount
    
    # 总计
    print("\n" + "=" * 50)
    print("总计:")
    print(f"  总市值: ${total_market_value:.2f}")
    print(f"  总成本: ${total_cost:.2f}")
    print(f"  总盈亏: ${total_pnl:.2f} ({total_pnl/total_cost*100:.2f}%)")
    print("=" * 50)
    print("更新完成！")

if __name__ == "__main__":
    main()