101 lines
3.4 KiB
Python
101 lines
3.4 KiB
Python
"""Paper trades router — take, list, and close simulated trades."""
|
|
|
|
from fastapi import APIRouter, Depends, Query
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.dependencies import get_db, require_access, require_admin
|
|
from app.models.user import User
|
|
from app.schemas.common import APIEnvelope
|
|
from app.schemas.paper_trade import (
|
|
ExitPolicyUpdate,
|
|
PaperTradeClose,
|
|
PaperTradeCreate,
|
|
PaperTradeResponse,
|
|
)
|
|
from app.services import paper_trade_service
|
|
|
|
router = APIRouter(tags=["paper-trades"])
|
|
|
|
|
|
def _resp(trade, symbol: str, current_price=None) -> dict:
|
|
return PaperTradeResponse(
|
|
id=trade.id,
|
|
symbol=symbol,
|
|
direction=trade.direction,
|
|
entry_price=trade.entry_price,
|
|
shares=trade.shares,
|
|
stop_loss=trade.stop_loss,
|
|
target=trade.target,
|
|
status=trade.status,
|
|
opened_at=trade.opened_at,
|
|
close_price=trade.close_price,
|
|
closed_at=trade.closed_at,
|
|
current_price=current_price,
|
|
).model_dump(mode="json")
|
|
|
|
|
|
@router.get("/paper-trades", response_model=APIEnvelope)
|
|
async def list_paper_trades(
|
|
status: str | None = Query(default=None, pattern=r"^(open|closed)$"),
|
|
user: User = Depends(require_access),
|
|
db: AsyncSession = Depends(get_db),
|
|
) -> APIEnvelope:
|
|
rows = await paper_trade_service.list_trades(db, user.id, status=status)
|
|
data = [PaperTradeResponse(**r).model_dump(mode="json") for r in rows]
|
|
return APIEnvelope(status="success", data=data)
|
|
|
|
|
|
@router.get("/paper-trades/exit-policy", response_model=APIEnvelope)
|
|
async def read_exit_policy(
|
|
_user: User = Depends(require_access),
|
|
db: AsyncSession = Depends(get_db),
|
|
) -> APIEnvelope:
|
|
"""The active auto-exit policy for open paper trades (shown in the UI)."""
|
|
return APIEnvelope(status="success", data=await paper_trade_service.get_exit_policy(db))
|
|
|
|
|
|
@router.put("/paper-trades/exit-policy", response_model=APIEnvelope)
|
|
async def write_exit_policy(
|
|
body: ExitPolicyUpdate,
|
|
_user: User = Depends(require_admin),
|
|
db: AsyncSession = Depends(get_db),
|
|
) -> APIEnvelope:
|
|
"""Change the auto-exit policy (admin)."""
|
|
data = await paper_trade_service.set_exit_policy(
|
|
db,
|
|
mode=body.mode,
|
|
trailing_pct=body.trailing_pct,
|
|
atr_multiplier=body.atr_multiplier,
|
|
hold_days=body.hold_days,
|
|
)
|
|
return APIEnvelope(status="success", data=data)
|
|
|
|
|
|
@router.post("/paper-trades", response_model=APIEnvelope, status_code=201)
|
|
async def create_paper_trade(
|
|
body: PaperTradeCreate,
|
|
user: User = Depends(require_access),
|
|
db: AsyncSession = Depends(get_db),
|
|
) -> APIEnvelope:
|
|
trade = await paper_trade_service.create_trade(
|
|
db, user.id,
|
|
symbol=body.symbol,
|
|
direction=body.direction,
|
|
entry_price=body.entry_price,
|
|
shares=body.shares,
|
|
stop_loss=body.stop_loss,
|
|
target=body.target,
|
|
)
|
|
return APIEnvelope(status="success", data=_resp(trade, body.symbol.strip().upper()))
|
|
|
|
|
|
@router.post("/paper-trades/{trade_id}/close", response_model=APIEnvelope)
|
|
async def close_paper_trade(
|
|
trade_id: int,
|
|
body: PaperTradeClose,
|
|
user: User = Depends(require_access),
|
|
db: AsyncSession = Depends(get_db),
|
|
) -> APIEnvelope:
|
|
trade = await paper_trade_service.close_trade(db, user.id, trade_id, body.close_price)
|
|
return APIEnvelope(status="success", data={"id": trade.id, "status": trade.status})
|