97 lines
2.5 KiB
Python
97 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from pydantic import BaseModel
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.core.database import get_db
|
|
from app.models.stock import Trade
|
|
from app.services.trading import trading_service
|
|
|
|
router = APIRouter(prefix="/api/trading", tags=["trading"])
|
|
|
|
|
|
class OrderRequest(BaseModel):
|
|
stock_code: str
|
|
side: str # buy / sell
|
|
qty: int
|
|
price: int = 0
|
|
order_type: str = "00"
|
|
|
|
|
|
class CancelRequest(BaseModel):
|
|
order_no: str
|
|
stock_code: str
|
|
qty: int
|
|
|
|
|
|
@router.post("/order")
|
|
async def place_order(data: OrderRequest, db: Session = Depends(get_db)) -> dict:
|
|
result = await trading_service.place_order(
|
|
stock_code=data.stock_code,
|
|
side=data.side,
|
|
qty=data.qty,
|
|
price=data.price,
|
|
order_type=data.order_type,
|
|
)
|
|
|
|
trade = Trade(
|
|
order_no=result.get("order_no", ""),
|
|
stock_code=data.stock_code,
|
|
stock_name="",
|
|
side=data.side,
|
|
qty=data.qty,
|
|
price=data.price,
|
|
order_type=data.order_type,
|
|
status="filled" if result.get("rt_cd") == "0" else "rejected",
|
|
)
|
|
db.add(trade)
|
|
db.commit()
|
|
|
|
return result
|
|
|
|
|
|
@router.post("/cancel")
|
|
async def cancel_order(data: CancelRequest) -> dict:
|
|
result = await trading_service.cancel_order(
|
|
order_no=data.order_no,
|
|
stock_code=data.stock_code,
|
|
qty=data.qty,
|
|
)
|
|
return result
|
|
|
|
|
|
@router.get("/history")
|
|
def get_trade_history(
|
|
stock_code: str | None = None,
|
|
limit: int = 50,
|
|
db: Session = Depends(get_db),
|
|
) -> list[dict]:
|
|
query = db.query(Trade).order_by(Trade.created_at.desc())
|
|
if stock_code:
|
|
query = query.filter(Trade.stock_code == stock_code)
|
|
trades = query.limit(limit).all()
|
|
return [
|
|
{
|
|
"id": t.id,
|
|
"order_no": t.order_no,
|
|
"stock_code": t.stock_code,
|
|
"stock_name": t.stock_name,
|
|
"side": t.side,
|
|
"qty": t.qty,
|
|
"price": t.price,
|
|
"order_type": t.order_type,
|
|
"status": t.status,
|
|
"strategy_id": t.strategy_id,
|
|
"created_at": t.created_at.isoformat() if t.created_at else None,
|
|
"filled_at": t.filled_at.isoformat() if t.filled_at else None,
|
|
}
|
|
for t in trades
|
|
]
|
|
|
|
|
|
@router.get("/account")
|
|
async def get_account() -> dict:
|
|
from app.services.account import account_service
|
|
return await account_service.get_balance()
|