Refactor training day submissions to use managing contest participation

cms/db/submission.py

@@ -39,7 +39,7 @@
 
 from cmscommon.datetime import make_datetime
 from . import Filename, FilenameSchema, Digest, Base, Participation, Task, \
-    Dataset, Testcase
+    Dataset, Testcase, TrainingDay
 
 
 class Submission(Base):
@@ -85,6 +85,21 @@ class Submission(Base):
         Task,
         back_populates="submissions")
 
+    # Training day (id and object) that this submission was submitted via.
+    # If None, the submission was made directly to the task archive or
+    # via a regular contest (not a training day).
+    # When set, the submission was made via a training day's contest interface,
+    # but the participation points to the managing contest's participation.
+    training_day_id: int | None = Column(
+        Integer,
+        ForeignKey(TrainingDay.id,
+                   onupdate="CASCADE", ondelete="SET NULL"),
+        nullable=True,
+        index=True)
+    training_day: TrainingDay | None = relationship(
+        TrainingDay,
+        back_populates="submissions")
+
     # Time of the submission.
     timestamp: datetime = Column(
         DateTime,

cms/db/training_day.py

@@ -30,7 +30,7 @@
 from . import Base
 
 if typing.TYPE_CHECKING:
-    from . import Contest, TrainingProgram, Task, TrainingDayGroup
+    from . import Contest, TrainingProgram, Task, TrainingDayGroup, Submission
 
 
 class TrainingDay(Base):
@@ -88,3 +88,9 @@ class TrainingDay(Base):
         back_populates="training_day",
         cascade="all, delete-orphan",
     )
+
+    submissions: list["Submission"] = relationship(
+        "Submission",
+        back_populates="training_day",
+        passive_deletes=True,
+    )

cms/server/admin/handlers/__init__.py

@@ -104,7 +104,9 @@
 from .submissiondownload import \
     DownloadTaskSubmissionsHandler, \
     DownloadUserContestSubmissionsHandler, \
-    DownloadContestSubmissionsHandler
+    DownloadContestSubmissionsHandler, \
+    DownloadTrainingProgramSubmissionsHandler, \
+    DownloadTrainingProgramStudentSubmissionsHandler
 from .task import (
     AddTaskHandler,
     TaskHandler,
@@ -338,6 +340,8 @@
     (r"/training_program/([0-9]+)/ranking", TrainingProgramRankingHandler),
     (r"/training_program/([0-9]+)/ranking/([a-z]+)", TrainingProgramRankingHandler),
     (r"/training_program/([0-9]+)/submissions", TrainingProgramSubmissionsHandler),
+    (r"/training_program/([0-9]+)/submissions/download", DownloadTrainingProgramSubmissionsHandler),
+    (r"/training_program/([0-9]+)/student/([0-9]+)/submissions/download", DownloadTrainingProgramStudentSubmissionsHandler),
     (r"/training_program/([0-9]+)/announcements", TrainingProgramAnnouncementsHandler),
     (r"/training_program/([0-9]+)/announcement/([0-9]+)", TrainingProgramAnnouncementHandler),
     (r"/training_program/([0-9]+)/questions", TrainingProgramQuestionsHandler),

cms/server/admin/handlers/contestsubmission.py

@@ -39,18 +39,27 @@ def get(self, contest_id):
         contest = self.safe_get_item(Contest, contest_id)
         self.contest = contest
 
-        # For training day contests, tasks have training_day_id set
-        # but contest_id points to the managing contest.
-        # We need to filter by training_day_id for training day contests.
+        # Determine if this is a training program managing contest
+        is_training_program = contest.training_program is not None
+
+        # For training day contests, only show submissions made via that training day
+        # (submissions now have training_day_id set when submitted via a training day)
         if contest.training_day is not None:
-            query = self.sql_session.query(Submission).join(Task)\
-                .filter(Task.training_day_id == contest.training_day.id)
+            query = self.sql_session.query(Submission)\
+                .filter(Submission.training_day_id == contest.training_day.id)
         else:
+            # For regular contests and training program managing contests,
+            # show all submissions for tasks in this contest
             query = self.sql_session.query(Submission).join(Task)\
                 .filter(Task.contest == contest)
         page = int(self.get_query_argument("page", 0))
         self.render_params_for_submissions(query, page)
 
+        # Pass flag and training_program to template for training programs
+        self.r_params["is_training_program"] = is_training_program
+        if is_training_program:
+            self.r_params["training_program"] = contest.training_program
+
         self.render("contest_submissions.html", **self.r_params)