volsu-contests/backend/app/models/submission.py
2025-11-30 19:55:50 +03:00

36 lines
1.8 KiB
Python

from datetime import datetime, timezone
from sqlalchemy import String, Text, Integer, DateTime, ForeignKey, JSON
from sqlalchemy.orm import Mapped, mapped_column, relationship
from app.database import Base
def utcnow() -> datetime:
return datetime.now(timezone.utc)
class Submission(Base):
__tablename__ = "submissions"
id: Mapped[int] = mapped_column(primary_key=True)
user_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
problem_id: Mapped[int] = mapped_column(ForeignKey("problems.id", ondelete="CASCADE"), nullable=False)
contest_id: Mapped[int] = mapped_column(ForeignKey("contests.id", ondelete="CASCADE"), nullable=False)
source_code: Mapped[str] = mapped_column(Text, nullable=False)
language_id: Mapped[int] = mapped_column(Integer, nullable=False) # Judge0 language ID
language_name: Mapped[str | None] = mapped_column(String(50), nullable=True)
status: Mapped[str] = mapped_column(String(50), default="pending") # pending, judging, accepted, wrong_answer, etc.
score: Mapped[int] = mapped_column(Integer, default=0)
total_points: Mapped[int] = mapped_column(Integer, default=0)
tests_passed: Mapped[int] = mapped_column(Integer, default=0)
tests_total: Mapped[int] = mapped_column(Integer, default=0)
execution_time_ms: Mapped[int | None] = mapped_column(Integer, nullable=True)
memory_used_kb: Mapped[int | None] = mapped_column(Integer, nullable=True)
judge_response: Mapped[dict | None] = mapped_column(JSON, nullable=True)
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utcnow)
# Relationships
user = relationship("User", back_populates="submissions")
problem = relationship("Problem", back_populates="submissions")
contest = relationship("Contest", back_populates="submissions")