29 lines
1006 B
Python
29 lines
1006 B
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import DateTime, ForeignKey, String, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.database import Base
|
|
|
|
|
|
class WatchlistEntry(Base):
|
|
__tablename__ = "watchlist_entries"
|
|
__table_args__ = (
|
|
UniqueConstraint("user_id", "ticker_id", name="uq_watchlist_user_ticker"),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
user_id: Mapped[int] = mapped_column(
|
|
ForeignKey("users.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
ticker_id: Mapped[int] = mapped_column(
|
|
ForeignKey("tickers.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
entry_type: Mapped[str] = mapped_column(String(10), nullable=False)
|
|
added_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), default=datetime.utcnow, nullable=False
|
|
)
|
|
|
|
user = relationship("User", back_populates="watchlist_entries")
|
|
ticker = relationship("Ticker", back_populates="watchlist_entries")
|