78 lines
3 KiB
Python
78 lines
3 KiB
Python
from fastapi import APIRouter, Query, HTTPException, Body
|
|
from typing import Optional
|
|
from app.services.bybit_service import get_bybit_client
|
|
import json
|
|
import os
|
|
|
|
router = APIRouter()
|
|
|
|
ACCOUNTS_FILE = os.path.join(os.path.dirname(__file__), '../../accounts.json')
|
|
|
|
def get_user_api(user_id: str):
|
|
try:
|
|
with open(ACCOUNTS_FILE, 'r', encoding='utf-8') as f:
|
|
accounts = json.load(f)
|
|
for acc in accounts:
|
|
if acc['id'] == user_id:
|
|
return acc['bybit_api_key'], acc['bybit_api_secret']
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Lỗi đọc file accounts: {e}")
|
|
raise HTTPException(status_code=404, detail="Không tìm thấy userId")
|
|
|
|
@router.get("/orders", tags=["Order"])
|
|
def get_orders(
|
|
userId: str = Query(..., description="ID của user để lấy API key/secret"),
|
|
symbol: str = Query(..., description="Mã giao dịch, ví dụ: BTCUSDT"),
|
|
category: str = Query("linear", description="Loại lệnh: linear (future) hoặc spot")
|
|
):
|
|
api_key, api_secret = get_user_api(userId)
|
|
client = get_bybit_client(api_key, api_secret)
|
|
res = client.get_open_orders(
|
|
category=category, # "linear" cho future, "spot" cho spot
|
|
symbol=symbol
|
|
)
|
|
return res
|
|
|
|
# Submit order
|
|
@router.post("/orders", tags=["Order"])
|
|
def submit_order(
|
|
userId: str = Body(..., embed=True, description="ID của user để lấy API key/secret"),
|
|
symbol: str = Body(..., embed=True, description="Mã giao dịch, ví dụ: BTCUSDT"),
|
|
side: str = Body(..., embed=True, description="side: Buy hoặc Sell"),
|
|
orderType: str = Body(..., embed=True, description="Loại lệnh: Market hoặc Limit"),
|
|
qty: float = Body(..., embed=True, description="Số lượng"),
|
|
category: str = Body("linear", embed=True, description="Loại lệnh: linear (future) hoặc spot"),
|
|
price: Optional[float] = Body(None, embed=True, description="Giá (bắt buộc với Limit)")
|
|
):
|
|
api_key, api_secret = get_user_api(userId)
|
|
client = get_bybit_client(api_key, api_secret)
|
|
params = {
|
|
"category": category,
|
|
"symbol": symbol,
|
|
"side": side,
|
|
"orderType": orderType,
|
|
"qty": qty
|
|
}
|
|
if orderType.lower() == "limit":
|
|
if price is None:
|
|
raise HTTPException(status_code=400, detail="Thiếu giá cho lệnh Limit")
|
|
params["price"] = price
|
|
res = client.place_order(**params)
|
|
return res
|
|
|
|
# Cancel order
|
|
@router.delete("/orders/{order_id}", tags=["Order"])
|
|
def cancel_order(
|
|
order_id: str,
|
|
userId: str = Query(..., description="ID của user để lấy API key/secret"),
|
|
symbol: str = Query(..., description="Mã giao dịch, ví dụ: BTCUSDT"),
|
|
category: str = Query("linear", description="Loại lệnh: linear (future) hoặc spot")
|
|
):
|
|
api_key, api_secret = get_user_api(userId)
|
|
client = get_bybit_client(api_key, api_secret)
|
|
res = client.cancel_order(
|
|
category=category,
|
|
symbol=symbol,
|
|
orderId=order_id
|
|
)
|
|
return res
|