70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
|
"""Data models for trade persistence."""
|
||
|
|
from dataclasses import dataclass, asdict
|
||
|
|
from typing import Optional
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Trade:
|
||
|
|
"""Represents a completed trade."""
|
||
|
|
|
||
|
|
trade_id: str
|
||
|
|
symbol: str
|
||
|
|
side: str
|
||
|
|
entry_price: float
|
||
|
|
size: float
|
||
|
|
size_usdt: float
|
||
|
|
entry_time: str
|
||
|
|
exit_price: Optional[float] = None
|
||
|
|
pnl_usd: Optional[float] = None
|
||
|
|
pnl_pct: Optional[float] = None
|
||
|
|
exit_time: Optional[str] = None
|
||
|
|
hold_duration_hours: Optional[float] = None
|
||
|
|
reason: Optional[str] = None
|
||
|
|
order_id_entry: Optional[str] = None
|
||
|
|
order_id_exit: Optional[str] = None
|
||
|
|
id: Optional[int] = None
|
||
|
|
|
||
|
|
def to_dict(self) -> dict:
|
||
|
|
"""Convert to dictionary."""
|
||
|
|
return asdict(self)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_row(cls, row: tuple, columns: list[str]) -> "Trade":
|
||
|
|
"""Create Trade from database row."""
|
||
|
|
data = dict(zip(columns, row))
|
||
|
|
return cls(**data)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class DailySummary:
|
||
|
|
"""Daily trading summary."""
|
||
|
|
|
||
|
|
date: str
|
||
|
|
total_trades: int = 0
|
||
|
|
winning_trades: int = 0
|
||
|
|
total_pnl_usd: float = 0.0
|
||
|
|
max_drawdown_usd: float = 0.0
|
||
|
|
id: Optional[int] = None
|
||
|
|
|
||
|
|
def to_dict(self) -> dict:
|
||
|
|
"""Convert to dictionary."""
|
||
|
|
return asdict(self)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Session:
|
||
|
|
"""Trading session metadata."""
|
||
|
|
|
||
|
|
start_time: str
|
||
|
|
end_time: Optional[str] = None
|
||
|
|
starting_balance: Optional[float] = None
|
||
|
|
ending_balance: Optional[float] = None
|
||
|
|
total_pnl: Optional[float] = None
|
||
|
|
total_trades: int = 0
|
||
|
|
id: Optional[int] = None
|
||
|
|
|
||
|
|
def to_dict(self) -> dict:
|
||
|
|
"""Convert to dictionary."""
|
||
|
|
return asdict(self)
|