41 lines
1.0 KiB
Python
Raw Normal View History

"""
Drawdown related Pydantic schemas.
"""
from datetime import date
from decimal import Decimal
from typing import List, Optional
from pydantic import BaseModel, Field
from app.schemas.portfolio import FloatDecimal
class DrawdownDataPoint(BaseModel):
date: date
total_value: FloatDecimal
peak: FloatDecimal
drawdown_pct: FloatDecimal
class DrawdownResponse(BaseModel):
portfolio_id: int
current_drawdown_pct: FloatDecimal
max_drawdown_pct: FloatDecimal
max_drawdown_date: date | None = None
peak_value: FloatDecimal | None = None
peak_date: date | None = None
trough_value: FloatDecimal | None = None
trough_date: date | None = None
alert_threshold_pct: FloatDecimal = Decimal("20")
class DrawdownHistoryResponse(BaseModel):
portfolio_id: int
data: List[DrawdownDataPoint] = []
max_drawdown_pct: FloatDecimal
current_drawdown_pct: FloatDecimal
class DrawdownSettingsUpdate(BaseModel):
alert_threshold_pct: FloatDecimal = Field(..., gt=0, le=100)