71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from datetime import datetime
|
||
|
|
from decimal import Decimal
|
||
|
|
from enum import Enum
|
||
|
|
from typing import Optional
|
||
|
|
from uuid import UUID
|
||
|
|
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class Direction(str, Enum):
|
||
|
|
long = "long"
|
||
|
|
short = "short"
|
||
|
|
|
||
|
|
|
||
|
|
# ─── Request schemas ──────────────────────────────────────────────
|
||
|
|
|
||
|
|
class PositionCreate(BaseModel):
|
||
|
|
symbol: str
|
||
|
|
direction: Direction
|
||
|
|
entry_price: Decimal
|
||
|
|
quantity: Decimal
|
||
|
|
exchange: str
|
||
|
|
metadata: Optional[dict] = None
|
||
|
|
|
||
|
|
|
||
|
|
class PositionUpdate(BaseModel):
|
||
|
|
entry_price: Optional[Decimal] = None
|
||
|
|
exit_price: Optional[Decimal] = None
|
||
|
|
quantity: Optional[Decimal] = None
|
||
|
|
metadata: Optional[dict] = None
|
||
|
|
|
||
|
|
|
||
|
|
class WebhookTrade(BaseModel):
|
||
|
|
"""Payload from automated scanner scripts."""
|
||
|
|
symbol: str
|
||
|
|
direction: Direction
|
||
|
|
entry_price: Decimal
|
||
|
|
quantity: Decimal
|
||
|
|
exchange: str
|
||
|
|
strategy: Optional[str] = None
|
||
|
|
|
||
|
|
|
||
|
|
# ─── Response schemas ─────────────────────────────────────────────
|
||
|
|
|
||
|
|
class PositionOut(BaseModel):
|
||
|
|
id: UUID
|
||
|
|
symbol: str
|
||
|
|
direction: Direction
|
||
|
|
entry_price: Decimal
|
||
|
|
exit_price: Optional[Decimal]
|
||
|
|
quantity: Decimal
|
||
|
|
exchange: str
|
||
|
|
opened_at: datetime
|
||
|
|
closed_at: Optional[datetime]
|
||
|
|
pnl: Optional[Decimal]
|
||
|
|
metadata: Optional[dict]
|
||
|
|
|
||
|
|
model_config = {"from_attributes": True}
|
||
|
|
|
||
|
|
|
||
|
|
class PnLSnapshot(BaseModel):
|
||
|
|
today_pnl: Decimal
|
||
|
|
week_pnl: Decimal
|
||
|
|
month_pnl: Decimal
|
||
|
|
all_time_pnl: Decimal
|
||
|
|
total_trades: int
|
||
|
|
open_positions: int
|
||
|
|
|