update milestones when leaderboard gets updated

Author: ngc92

src/kernelbot/cogs/admin_cog.py

@@ -807,7 +807,7 @@ async def _create_update_plan(  # noqa: C901
 
         return update_list, create_list
 
-    async def update_competition(
+    async def update_competition(  # noqa: C901
         self, interaction: discord.Interaction, spec_file: Path, force: bool = False
     ):
         try:
@@ -857,6 +857,21 @@ async def update_competition(
                         entry["name"], self._parse_deadline(entry["deadline"]), task
                     )
                     new_lb: LeaderboardItem = db.get_leaderboard(entry["name"])
+                    # delete old milestones
+                    db.delete_milestones(new_lb["id"])
+                    # and (re)-create new ones
+                    for milestone in task.milestones:
+                        db.create_milestone(
+                            new_lb["id"],
+                            milestone.name,
+                            milestone.code,
+                            description=milestone.description,
+                            exclude_gpus=milestone.exclude_gpus,
+                        )
+
+                    #  and finally trigger re-run
+                    if task.milestones:
+                        await self._submit_milestones(interaction, new_lb["name"])
 
                 forum_id = new_lb["forum_id"]
                 try:

src/libkernelbot/leaderboard_db.py

@@ -313,6 +313,16 @@ def delete_milestone_runs(self, leaderboard_name: str):
         )
         self.connection.commit()
 
+    def delete_milestones(self, leaderboard_id: int):
+        self.cursor.execute(
+            """
+            DELETE FROM leaderboard.milestones
+            WHERE leaderboard_id = %s;
+            """,
+            (leaderboard_id,),
+        )
+        self.connection.commit()
+
     def get_runs_generic(
         self, *, milestone_id: Optional[int] = None, submission_id: Optional[int] = None
     ) -> List["RunItem"]: