47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import Boolean, DateTime, Float, ForeignKey, String, Text, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.database import Base
|
|
|
|
|
|
class DimensionScore(Base):
|
|
__tablename__ = "dimension_scores"
|
|
__table_args__ = (
|
|
UniqueConstraint("ticker_id", "dimension", name="uq_dimension_score_ticker_dimension"),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
ticker_id: Mapped[int] = mapped_column(
|
|
ForeignKey("tickers.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
dimension: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
score: Mapped[float] = mapped_column(Float, nullable=False)
|
|
is_stale: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
|
computed_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), nullable=False
|
|
)
|
|
|
|
ticker = relationship("Ticker", back_populates="dimension_scores")
|
|
|
|
|
|
class CompositeScore(Base):
|
|
__tablename__ = "composite_scores"
|
|
__table_args__ = (
|
|
UniqueConstraint("ticker_id", name="uq_composite_score_ticker"),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
ticker_id: Mapped[int] = mapped_column(
|
|
ForeignKey("tickers.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
score: Mapped[float] = mapped_column(Float, nullable=False)
|
|
is_stale: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
|
weights_json: Mapped[str] = mapped_column(Text, nullable=False)
|
|
computed_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), nullable=False
|
|
)
|
|
|
|
ticker = relationship("Ticker", back_populates="composite_scores")
|