add the ability to customize --max-duration

Author: Ayc0

tool/AOC.py

@@ -100,6 +100,12 @@ def run(argv: list[str]) -> None:
             action="store_true",
             default=False,
         )
+        parser.add_argument(
+            "--max-duration",
+            help="max duration per script in seconds (default to 1s)",
+            default=1,
+            type=int
+        )
 
         args = parser.parse_args(argv)
 
@@ -115,6 +121,7 @@ def run(argv: list[str]) -> None:
             args.restricted,
             args.expand,
             args.times,
+            args.max_duration,
         )
 
     @staticmethod

tool/run.py

@@ -17,20 +17,18 @@
 from tool.leaderboard.leaderboard import generate_leaderboard
 
 
-def maxDurationHandler(signum, frame):
-    raise Exception("Maxiumum of 1s reached")
-
-signal.signal(signal.SIGALRM, maxDurationHandler)
-
-
-
 class DifferentAnswersException(Exception):
     pass
 
 
 class UnexpectedDebugLinesException(Exception):
     pass
 
+def maxDurationHandler(signum, frame):
+    raise Exception("Too long")
+
+signal.signal(signal.SIGALRM, maxDurationHandler)
+
 
 def run(
     days: list[int] | None,
@@ -44,6 +42,7 @@ def run(
     restricted: bool,
     expand: bool,
     print_time_dist: bool,
+    max_duration: int,
 ) -> None:
     problems = discovery.get_problems(days, parts, all_days_parts)
     printed_day_header: set[int] = set()
@@ -71,7 +70,7 @@ def run(
                 if restricted and input.author != submission.author.split(".")[0]:
                     continue
                 try:
-                    signal.alarm(1) # Start the 1s timer
+                    signal.alarm(max_duration) # Start timeout timer
                     result = run_submission(
                         problem, submission, input, previous, no_debug
                     )
@@ -82,7 +81,7 @@ def run(
                 except (DifferentAnswersException, UnexpectedDebugLinesException) as e:
                     errors.append(f"{BColor.RED}ERROR: {e}{BColor.ENDC}")
                 except Exception:
-                    errors.append(f"{BColor.RED}[{submission.author}] day-{submission.problem.day}/part-{submission.problem.part} ({submission.language}){BColor.ENDC}: Maxiumum of 1s reached (on input {BColor.BLUE}{input.author}{BColor.ENDC})")
+                    errors.append(f"{BColor.RED}[{submission.author}] day-{submission.problem.day}/part-{submission.problem.part} ({submission.language}){BColor.ENDC}: Maximum of {max_duration}s reached (on input {BColor.BLUE}{input.author}{BColor.ENDC})")
 
         for submission in submissions:
             if submission.runnable is not None: