154 lines
4.5 KiB
Python
154 lines
4.5 KiB
Python
#!/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()
|