30 lines
1.3 KiB
Python
30 lines
1.3 KiB
Python
from datetime import datetime
|
|
from sqlalchemy import String, Float, Boolean, ForeignKey, DateTime, func, Enum
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
import enum
|
|
from app.core.database import Base
|
|
|
|
|
|
class AlertType(str, enum.Enum):
|
|
PRICE_ABOVE = "price_above"
|
|
PRICE_BELOW = "price_below"
|
|
CHANGE_PCT_ABOVE = "change_pct_above"
|
|
CHANGE_PCT_BELOW = "change_pct_below"
|
|
|
|
|
|
class Alert(Base):
|
|
__tablename__ = "alerts"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True, index=True)
|
|
user_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"), index=True)
|
|
symbol: Mapped[str] = mapped_column(String(20), nullable=False, index=True)
|
|
name: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
alert_type: Mapped[AlertType] = mapped_column(Enum(AlertType), nullable=False)
|
|
threshold: Mapped[float] = mapped_column(Float, nullable=False)
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
triggered: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
triggered_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
user: Mapped["User"] = relationship(back_populates="alerts")
|