152 lines
4.6 KiB
Python
152 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
添加新的美股持仓到Notion数据库
|
||
"""
|
||
|
||
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", "name": "Vanguard Total Bond Market ETF", "shares": 6.7219, "cost": 501.83, "current_price": 74.22},
|
||
{"symbol": "SGOV", "name": "iShares 0-3 Month Treasury Bond ETF", "shares": 33.7072, "cost": 3380.48, "current_price": 100.44},
|
||
{"symbol": "VOO", "name": "Vanguard S&P 500 ETF", "shares": 3.1963, "cost": 1969.46, "current_price": 636.09}
|
||
]
|
||
|
||
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 add_position_to_notion(symbol, name, shares, cost, current_price):
|
||
"""添加持仓到Notion"""
|
||
url = "https://api.notion.com/v1/pages"
|
||
|
||
headers = {
|
||
"Authorization": f"Bearer {NOTION_API_KEY}",
|
||
"Content-Type": "application/json",
|
||
"Notion-Version": "2022-06-28"
|
||
}
|
||
|
||
# 计算盈亏
|
||
market_value, total_cost, pnl_amount, pnl_percent = calculate_pnl(shares, cost, current_price)
|
||
|
||
# 获取当前时间（UTC）
|
||
now = datetime.utcnow()
|
||
date_str = now.strftime("%Y-%m-%dT%H:%M:00Z")
|
||
|
||
data = {
|
||
"parent": {
|
||
"database_id": DATABASE_ID
|
||
},
|
||
"properties": {
|
||
"股票代码": {
|
||
"title": [
|
||
{
|
||
"text": {
|
||
"content": symbol
|
||
}
|
||
}
|
||
]
|
||
},
|
||
"公司名称": {
|
||
"rich_text": [
|
||
{
|
||
"text": {
|
||
"content": name
|
||
}
|
||
}
|
||
]
|
||
},
|
||
"持仓数量": {
|
||
"number": shares
|
||
},
|
||
"买入成本": {
|
||
"number": cost
|
||
},
|
||
"当前价格": {
|
||
"number": current_price
|
||
},
|
||
"总成本": {
|
||
"number": total_cost
|
||
},
|
||
"当前市值": {
|
||
"number": market_value
|
||
},
|
||
"盈亏金额": {
|
||
"number": round(pnl_amount, 2)
|
||
},
|
||
"盈亏百分比": {
|
||
"number": round(pnl_percent, 2)
|
||
},
|
||
"最后更新": {
|
||
"date": {
|
||
"start": date_str
|
||
}
|
||
},
|
||
"备注": {
|
||
"rich_text": [
|
||
{
|
||
"text": {
|
||
"content": "自动添加"
|
||
}
|
||
}
|
||
]
|
||
}
|
||
}
|
||
}
|
||
|
||
response = requests.post(url, headers=headers, json=data)
|
||
|
||
if response.status_code == 200:
|
||
result = response.json()
|
||
page_id = result["id"]
|
||
print(f" ✅ 成功添加 {symbol} - 页面ID: {page_id}")
|
||
return page_id
|
||
else:
|
||
print(f" ❌ 添加失败 {symbol}: {response.status_code}")
|
||
print(f" 响应: {response.text}")
|
||
return None
|
||
|
||
def main():
|
||
print("开始添加新的美股持仓到Notion...")
|
||
print("=" * 60)
|
||
|
||
for position in positions:
|
||
symbol = position["symbol"]
|
||
name = position["name"]
|
||
shares = position["shares"]
|
||
cost = position["cost"]
|
||
current_price = position["current_price"]
|
||
|
||
print(f"\n添加 {symbol} ({name})...")
|
||
|
||
# 计算盈亏
|
||
market_value, total_cost, pnl_amount, pnl_percent = calculate_pnl(shares, cost, current_price)
|
||
print(f" 持仓数量: {shares}")
|
||
print(f" 买入成本: ${cost:.2f}")
|
||
print(f" 当前价格: ${current_price:.2f}")
|
||
print(f" 当前市值: ${market_value:.2f}")
|
||
print(f" 盈亏: ${pnl_amount:.2f} ({pnl_percent:.2f}%)")
|
||
|
||
# 添加到Notion
|
||
page_id = add_position_to_notion(symbol, name, shares, cost, current_price)
|
||
|
||
if page_id:
|
||
print(f" 📄 页面链接: https://www.notion.so/{page_id.replace('-', '')}")
|
||
|
||
print("\n" + "=" * 60)
|
||
print("所有持仓添加完成！")
|
||
print("=" * 60)
|
||
|
||
if __name__ == "__main__":
|
||
main()
|