283 lines
8.7 KiB
Python
283 lines
8.7 KiB
Python
import uuid
|
|
import asyncio
|
|
from datetime import datetime
|
|
from fastapi import APIRouter, Depends, HTTPException, status, BackgroundTasks
|
|
from sqlalchemy.orm import Session
|
|
from typing import List, Dict, Any, Optional
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
|
|
from .auth import get_current_user
|
|
from ..core.database import get_db
|
|
from ..core.config import get_settings
|
|
from ..db.schemas import BacktestCreate, BacktestResponse
|
|
from ..db.models import Bot, Backtest, Signal, User
|
|
|
|
router = APIRouter()
|
|
|
|
running_backtests: Dict[str, Any] = {}
|
|
executor = ThreadPoolExecutor(max_workers=4)
|
|
|
|
|
|
def run_backtest_sync(
|
|
backtest_id: str, db_url: str, bot_id: str, config: Dict[str, Any]
|
|
):
|
|
import asyncio
|
|
import json
|
|
from ..services.backtest.engine import BacktestEngine
|
|
from ..core.database import SessionLocal
|
|
|
|
async def _run():
|
|
engine = BacktestEngine(config)
|
|
engine.run_id = backtest_id
|
|
running_backtests[backtest_id] = engine
|
|
try:
|
|
results = await engine.run()
|
|
|
|
# Convert datetime objects to ISO strings for JSON serialization
|
|
def convert_datetime(obj):
|
|
if isinstance(obj, datetime):
|
|
return obj.isoformat()
|
|
elif isinstance(obj, dict):
|
|
return {k: convert_datetime(v) for k, v in obj.items()}
|
|
elif isinstance(obj, list):
|
|
return [convert_datetime(i) for i in obj]
|
|
return obj
|
|
|
|
results = convert_datetime(results)
|
|
|
|
db = SessionLocal()
|
|
try:
|
|
backtest = db.query(Backtest).filter(Backtest.id == backtest_id).first()
|
|
if backtest:
|
|
backtest.status = engine.status
|
|
backtest.ended_at = datetime.utcnow()
|
|
backtest.result = results
|
|
db.commit()
|
|
|
|
for signal in engine.signals:
|
|
signal_data = convert_datetime(signal)
|
|
db_signal = Signal(
|
|
id=signal_data["id"],
|
|
bot_id=signal_data["bot_id"],
|
|
run_id=signal_data["run_id"],
|
|
signal_type=signal_data["signal_type"],
|
|
token=signal_data["token"],
|
|
price=signal_data["price"],
|
|
confidence=signal_data.get("confidence"),
|
|
reasoning=signal_data.get("reasoning"),
|
|
executed=signal_data.get("executed", False),
|
|
created_at=signal["created_at"], # Use original datetime, not converted string
|
|
)
|
|
db.add(db_signal)
|
|
db.commit()
|
|
finally:
|
|
db.close()
|
|
finally:
|
|
if backtest_id in running_backtests:
|
|
del running_backtests[backtest_id]
|
|
|
|
asyncio.run(_run())
|
|
|
|
|
|
@router.post(
|
|
"/bots/{bot_id}/backtest",
|
|
response_model=BacktestResponse,
|
|
status_code=status.HTTP_201_CREATED,
|
|
)
|
|
async def start_backtest(
|
|
bot_id: str,
|
|
config: BacktestCreate,
|
|
background_tasks: BackgroundTasks,
|
|
current_user: User = Depends(get_current_user),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
bot = db.query(Bot).filter(Bot.id == bot_id).first()
|
|
if not bot:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND, detail="Bot not found"
|
|
)
|
|
if bot.user_id != current_user.id:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN, detail="Not authorized"
|
|
)
|
|
|
|
settings = get_settings()
|
|
backtest_id = str(uuid.uuid4())
|
|
|
|
backtest_config = {
|
|
"bot_id": bot_id,
|
|
"token": config.token,
|
|
"chain": config.chain,
|
|
"timeframe": config.timeframe,
|
|
"start_date": config.start_date,
|
|
"end_date": config.end_date,
|
|
"strategy_config": bot.strategy_config,
|
|
"ave_api_key": settings.AVE_API_KEY,
|
|
"ave_api_plan": settings.AVE_API_PLAN,
|
|
"initial_balance": 10000.0,
|
|
}
|
|
|
|
backtest = Backtest(
|
|
id=backtest_id,
|
|
bot_id=bot_id,
|
|
started_at=datetime.utcnow(),
|
|
status="running",
|
|
config={
|
|
"token": config.token,
|
|
"chain": config.chain,
|
|
"timeframe": config.timeframe,
|
|
"start_date": config.start_date,
|
|
"end_date": config.end_date,
|
|
},
|
|
)
|
|
db.add(backtest)
|
|
db.commit()
|
|
db.refresh(backtest)
|
|
|
|
db_url = str(settings.DATABASE_URL)
|
|
background_tasks.add_task(
|
|
run_backtest_sync, backtest_id, db_url, bot_id, backtest_config
|
|
)
|
|
|
|
return backtest
|
|
|
|
|
|
@router.get("/bots/{bot_id}/backtest/{run_id}", response_model=BacktestResponse)
|
|
def get_backtest(
|
|
bot_id: str,
|
|
run_id: str,
|
|
current_user: User = Depends(get_current_user),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
bot = db.query(Bot).filter(Bot.id == bot_id).first()
|
|
if not bot:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND, detail="Bot not found"
|
|
)
|
|
if bot.user_id != current_user.id:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN, detail="Not authorized"
|
|
)
|
|
|
|
backtest = (
|
|
db.query(Backtest)
|
|
.filter(Backtest.id == run_id, Backtest.bot_id == bot_id)
|
|
.first()
|
|
)
|
|
if not backtest:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND, detail="Backtest not found"
|
|
)
|
|
|
|
# Add progress from running engine if available
|
|
if backtest.status == "running" and run_id in running_backtests:
|
|
engine = running_backtests[run_id]
|
|
backtest.progress = engine.progress
|
|
|
|
return backtest
|
|
|
|
|
|
@router.get("/bots/{bot_id}/backtest/{run_id}/trades")
|
|
def get_backtest_trades(
|
|
bot_id: str,
|
|
run_id: str,
|
|
current_user: User = Depends(get_current_user),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
"""Get trade history for a specific backtest."""
|
|
bot = db.query(Bot).filter(Bot.id == bot_id).first()
|
|
if not bot:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND, detail="Bot not found"
|
|
)
|
|
if bot.user_id != current_user.id:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN, detail="Not authorized"
|
|
)
|
|
|
|
backtest = (
|
|
db.query(Backtest)
|
|
.filter(Backtest.id == run_id, Backtest.bot_id == bot_id)
|
|
.first()
|
|
)
|
|
if not backtest:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND, detail="Backtest not found"
|
|
)
|
|
|
|
# Get trades from result
|
|
result = backtest.result or {}
|
|
trades = result.get("trades", [])
|
|
|
|
return {
|
|
"backtest_id": run_id,
|
|
"trades": trades,
|
|
"total_trades": len(trades),
|
|
}
|
|
|
|
@router.get("/bots/{bot_id}/backtests", response_model=List[BacktestResponse])
|
|
def list_backtests(
|
|
bot_id: str,
|
|
current_user: User = Depends(get_current_user),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
bot = db.query(Bot).filter(Bot.id == bot_id).first()
|
|
if not bot:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND, detail="Bot not found"
|
|
)
|
|
if bot.user_id != current_user.id:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN, detail="Not authorized"
|
|
)
|
|
|
|
backtests = (
|
|
db.query(Backtest)
|
|
.filter(Backtest.bot_id == bot_id)
|
|
.order_by(Backtest.started_at.desc())
|
|
.all()
|
|
)
|
|
return backtests
|
|
|
|
|
|
@router.post("/bots/{bot_id}/backtest/{run_id}/stop")
|
|
def stop_backtest(
|
|
bot_id: str,
|
|
run_id: str,
|
|
current_user: User = Depends(get_current_user),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
bot = db.query(Bot).filter(Bot.id == bot_id).first()
|
|
if not bot:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND, detail="Bot not found"
|
|
)
|
|
if bot.user_id != current_user.id:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN, detail="Not authorized"
|
|
)
|
|
|
|
backtest = (
|
|
db.query(Backtest)
|
|
.filter(Backtest.id == run_id, Backtest.bot_id == bot_id)
|
|
.first()
|
|
)
|
|
if not backtest:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND, detail="Backtest not found"
|
|
)
|
|
|
|
if run_id in running_backtests:
|
|
engine = running_backtests[run_id]
|
|
engine.running = False # Direct sync access to running flag
|
|
backtest.status = "stopped"
|
|
backtest.ended_at = datetime.utcnow()
|
|
db.commit()
|
|
elif backtest.status == "running":
|
|
# Engine already finished but status not updated
|
|
backtest.status = "stopped"
|
|
backtest.ended_at = datetime.utcnow()
|
|
db.commit()
|
|
|
|
return {"status": "stopping", "run_id": run_id}
|