Initial commit: stock market platform

backend/app/models/alert.py

@@ -0,0 +1,29 @@
+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")