1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
"""Portfolio endpoints."""
import logging
from fastapi import APIRouter, HTTPException, Query, Request
from sqlalchemy import select
from sqlalchemy.exc import OperationalError
from shared.sa_models import PositionRow
logger = logging.getLogger(__name__)
router = APIRouter()
@router.get("/positions")
async def get_positions(request: Request):
"""Get all current positions."""
try:
db = request.app.state.db
async with db.get_session() as session:
result = await session.execute(select(PositionRow))
rows = result.scalars().all()
return [
{
"symbol": r.symbol,
"quantity": float(r.quantity),
"avg_entry_price": float(r.avg_entry_price),
"current_price": float(r.current_price),
"unrealized_pnl": float(r.quantity * (r.current_price - r.avg_entry_price)),
}
for r in rows
]
except OperationalError as exc:
logger.error("Database error fetching positions: %s", exc)
raise HTTPException(status_code=503, detail="Database unavailable") from exc
except Exception as exc:
logger.error("Failed to get positions: %s", exc, exc_info=True)
raise HTTPException(status_code=500, detail="Failed to retrieve positions") from exc
@router.get("/snapshots")
async def get_snapshots(request: Request, days: int = Query(30, ge=1, le=365)):
"""Get portfolio snapshots for the last N days."""
try:
db = request.app.state.db
snapshots = await db.get_portfolio_snapshots(days=days)
return [
{
"total_value": float(s["total_value"]),
"realized_pnl": float(s["realized_pnl"]),
"unrealized_pnl": float(s["unrealized_pnl"]),
"snapshot_at": s["snapshot_at"].isoformat(),
}
for s in snapshots
]
except OperationalError as exc:
logger.error("Database error fetching snapshots: %s", exc)
raise HTTPException(status_code=503, detail="Database unavailable") from exc
except Exception as exc:
logger.error("Failed to get snapshots: %s", exc, exc_info=True)
raise HTTPException(status_code=500, detail="Failed to retrieve snapshots") from exc
|
