33 lines
1.5 KiB
Python
33 lines
1.5 KiB
Python
|
from datetime import datetime
|
||
|
|
from enum import Enum
|
||
|
|
from sqlalchemy import String, Text, DateTime, ForeignKey, Integer
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||
|
|
|
||
|
|
from app.core.database import Base
|
||
|
|
|
||
|
|
|
||
|
|
class AssignmentStatus(str, Enum):
|
||
|
|
ACTIVE = "active"
|
||
|
|
COMPLETED = "completed"
|
||
|
|
DROPPED = "dropped"
|
||
|
|
|
||
|
|
|
||
|
|
class Assignment(Base):
|
||
|
|
__tablename__ = "assignments"
|
||
|
|
|
||
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
|
participant_id: Mapped[int] = mapped_column(ForeignKey("participants.id", ondelete="CASCADE"), index=True)
|
||
|
|
challenge_id: Mapped[int] = mapped_column(ForeignKey("challenges.id", ondelete="CASCADE"))
|
||
|
|
status: Mapped[str] = mapped_column(String(20), default=AssignmentStatus.ACTIVE.value)
|
||
|
|
proof_path: Mapped[str | None] = mapped_column(String(500), nullable=True)
|
||
|
|
proof_url: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
|
|
proof_comment: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
|
|
points_earned: Mapped[int] = mapped_column(Integer, default=0)
|
||
|
|
streak_at_completion: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
||
|
|
started_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||
|
|
completed_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
||
|
|
|
||
|
|
# Relationships
|
||
|
|
participant: Mapped["Participant"] = relationship("Participant", back_populates="assignments")
|
||
|
|
challenge: Mapped["Challenge"] = relationship("Challenge", back_populates="assignments")
|