Spaces:
Running
Running
File size: 3,255 Bytes
b1f38ad |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
"""
Additional simulated trading endpoints for trades and sessions
"""
def get_simulated_trades_endpoint(limit: int, current_user: str):
"""Get recent simulated trades for the current user"""
from models import Trade
from database import engine
from sqlmodel import Session, select
with Session(engine) as session:
statement = (
select(Trade)
.where(Trade.user_email == current_user)
.order_by(Trade.executed_at.desc())
.limit(limit)
)
trades = session.exec(statement).all()
trades_list = [
{
"symbol": trade.symbol,
"side": trade.side,
"price": trade.price,
"quantity": trade.quantity,
"total": trade.total,
"time": trade.executed_at.isoformat()
}
for trade in trades
]
return {"trades": trades_list}
def get_simulated_sessions_endpoint(current_user: str):
"""Get all simulated trading sessions for the current user"""
from models import TradingSession
from database import engine
from sqlmodel import Session, select
from datetime import datetime
from simulated_trading import simulated_sessions
with Session(engine) as session:
statement = (
select(TradingSession)
.where(TradingSession.user_email == current_user)
.order_by(TradingSession.start_time.desc())
)
sessions = session.exec(statement).all()
sessions_list = []
for s in sessions:
# Check if session is actually running in memory
is_actually_running = s.session_id in simulated_sessions
# If DB says running but not in memory, it expired/crashed
if s.is_running and not is_actually_running:
# Update DB to reflect reality
s.is_running = False
if s.end_time is None:
s.end_time = datetime.now()
session.add(s)
# Calculate elapsed and remaining time
now = datetime.now()
elapsed = (now - s.start_time).total_seconds() / 60 # minutes
# Calculate total duration based on duration_unit
if s.duration_unit == "days":
total_duration = s.duration_minutes * 24 * 60
else:
total_duration = s.duration_minutes
remaining = max(0, total_duration - elapsed)
sessions_list.append({
"session_id": s.session_id,
"strategy": s.strategy,
"symbol": s.symbol,
"trade_amount": s.trade_amount,
"is_running": is_actually_running, # Use real status from memory
"position": "NONE", # Default position
"trades_count": s.trades_count,
"pnl": s.total_pnl,
"elapsed_minutes": elapsed,
"remaining_minutes": remaining
})
session.commit() # Commit any DB updates
return {"sessions": sessions_list}
|