init

backend/app/models/activity.py

@@ -0,0 +1,84 @@
+import uuid
+from datetime import datetime
+
+from sqlalchemy import String, Float, Integer, DateTime, ForeignKey, func
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from backend.app.core.database import Base
+
+
+class Activity(Base):
+    __tablename__ = "activities"
+
+    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    rider_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("riders.id"))
+    name: Mapped[str | None] = mapped_column(String(200), nullable=True)
+    activity_type: Mapped[str] = mapped_column(String(50), default="road")
+    date: Mapped[datetime] = mapped_column(DateTime(timezone=True))
+    duration: Mapped[int] = mapped_column(Integer)  # seconds
+    distance: Mapped[float | None] = mapped_column(Float, nullable=True)  # meters
+    elevation_gain: Mapped[float | None] = mapped_column(Float, nullable=True)  # meters
+    file_path: Mapped[str | None] = mapped_column(String(500), nullable=True)
+
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
+
+    rider = relationship("Rider", back_populates="activities")
+    metrics = relationship("ActivityMetrics", back_populates="activity", uselist=False, lazy="joined")
+    intervals = relationship("Interval", back_populates="activity", lazy="selectin")
+    data_points = relationship("DataPoint", back_populates="activity", lazy="noload")
+
+
+class ActivityMetrics(Base):
+    __tablename__ = "activity_metrics"
+
+    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    activity_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("activities.id"), unique=True)
+
+    tss: Mapped[float | None] = mapped_column(Float, nullable=True)
+    normalized_power: Mapped[float | None] = mapped_column(Float, nullable=True)
+    intensity_factor: Mapped[float | None] = mapped_column(Float, nullable=True)
+    variability_index: Mapped[float | None] = mapped_column(Float, nullable=True)
+    avg_power: Mapped[float | None] = mapped_column(Float, nullable=True)
+    max_power: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    avg_hr: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    max_hr: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    avg_cadence: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    avg_speed: Mapped[float | None] = mapped_column(Float, nullable=True)  # m/s
+    calories: Mapped[int | None] = mapped_column(Integer, nullable=True)
+
+    activity = relationship("Activity", back_populates="metrics")
+
+
+class DataPoint(Base):
+    __tablename__ = "data_points"
+
+    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
+    activity_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("activities.id"))
+    timestamp: Mapped[datetime] = mapped_column(DateTime(timezone=True), index=True)
+
+    power: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    heart_rate: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    cadence: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    speed: Mapped[float | None] = mapped_column(Float, nullable=True)
+    latitude: Mapped[float | None] = mapped_column(Float, nullable=True)
+    longitude: Mapped[float | None] = mapped_column(Float, nullable=True)
+    altitude: Mapped[float | None] = mapped_column(Float, nullable=True)
+    temperature: Mapped[int | None] = mapped_column(Integer, nullable=True)
+
+    activity = relationship("Activity", back_populates="data_points")
+
+
+class Interval(Base):
+    __tablename__ = "intervals"
+
+    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    activity_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("activities.id"))
+    start_ts: Mapped[datetime] = mapped_column(DateTime(timezone=True))
+    end_ts: Mapped[datetime] = mapped_column(DateTime(timezone=True))
+    interval_type: Mapped[str] = mapped_column(String(50))  # work / rest / climb
+    avg_power: Mapped[float | None] = mapped_column(Float, nullable=True)
+    avg_hr: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    duration: Mapped[int | None] = mapped_column(Integer, nullable=True)  # seconds
+
+    activity = relationship("Activity", back_populates="intervals")