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,13 +17,6 @@
 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
 
@@ -44,6 +37,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 +65,11 @@ def run(
                 if restricted and input.author != submission.author.split(".")[0]:
                     continue
                 try:
-                    signal.alarm(1) # Start the 1s timer
+                    def maxDurationHandler(signum, frame):
+                        raise Exception(f"Maxiumum of {max_duration}s reached")
+
+                    signal.signal(signal.SIGALRM, maxDurationHandler)
+                    signal.alarm(max_duration) # Start timeout timer
                     result = run_submission(
                         problem, submission, input, previous, no_debug
                     )