quant/galaxis-po
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add rebalancing API endpoints

Browse Source
This commit is contained in:
zephyrdark 2026-02-03 07:09:26 +09:00
parent a45c44740e
commit 5f3c6061c9
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
backend/app/api/portfolio.py
View File

@ -14,7 +14,9 @@ from app.schemas.portfolio import (
TargetCreate, TargetResponse,
HoldingCreate, HoldingResponse,
TransactionCreate, TransactionResponse,
RebalanceResponse, RebalanceSimulationRequest, RebalanceSimulationResponse,
)
from app.services.rebalance import RebalanceService
router = APIRouter(prefix="/api/portfolios", tags=["portfolios"])
@ -289,3 +291,28 @@ async def add_transaction(
db.commit()
db.refresh(transaction)
return transaction
@router.get("/{portfolio_id}/rebalance", response_model=RebalanceResponse)
async def calculate_rebalance(
portfolio_id: int,
current_user: CurrentUser,
db: Session = Depends(get_db),
):
"""Calculate rebalancing for a portfolio."""
portfolio = _get_portfolio(db, portfolio_id, current_user.id)
service = RebalanceService(db)
return service.calculate_rebalance(portfolio)
@router.post("/{portfolio_id}/rebalance/simulate", response_model=RebalanceSimulationResponse)
async def simulate_rebalance(
portfolio_id: int,
data: RebalanceSimulationRequest,
current_user: CurrentUser,
db: Session = Depends(get_db),
):
"""Simulate rebalancing with additional investment amount."""
portfolio = _get_portfolio(db, portfolio_id, current_user.id)
service = RebalanceService(db)
return service.calculate_rebalance(portfolio, additional_amount=data.additional_amount)