Kill the judging submission first with a keyboard interrupt on CLI

DMOJ · Jun 13, 2020 · 0c6832f · 0c6832f
1 parent f66008d
commit 0c6832f
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/dmoj/commands/submit.py b/dmoj/commands/submit.py
@@ -73,10 +73,13 @@ def execute(self, line):
 
         self.judge.submission_id_counter += 1
         self.judge.graded_submissions.append((problem_id, language_id, src, time_limit, memory_limit))
-        self.judge.begin_grading(
-            Submission(
-                self.judge.submission_id_counter, problem_id, language_id, src, time_limit, memory_limit, False, {}
-            ),
-            blocking=True,
-            report=print,
-        )
+        try:
+            self.judge.begin_grading(
+                Submission(
+                    self.judge.submission_id_counter, problem_id, language_id, src, time_limit, memory_limit, False, {}
+                ),
+                blocking=True,
+                report=print,
+            )
+        except KeyboardInterrupt:
+            self.judge.abort_grading()