Move downtime calculations to their own file

Author: encukou

master/custom/worker_downtime.py

@@ -0,0 +1,48 @@
+from datetime import datetime, timedelta
+
+
+def no_builds_between(start, end, *, day_of_week=None, tz=None):
+    """A function for "builder.canStartBuild" that implements planned downtime
+
+    Avoid a build to be started between start and end time and delay such
+    builds until end time.
+    """
+
+    start = datetime.strptime(start, "%H:%M").time()
+    end = datetime.strptime(end, "%H:%M").time()
+    def canStartBuild(builder, wfb, request):
+        now_dt = datetime.now(tz=tz)
+        if day_of_week is not None and now_dt.weekday() != day_of_week:
+            return True
+        now = now_dt.time()
+        if is_within_time_range(now, start, end):
+            delay = get_delay(now, end)
+            # Adapted from: https://docs.buildbot.net/current/manual/customization.html#canstartbuild-functions
+            wfb.worker.quarantine_timeout = delay
+            wfb.worker.putInQuarantine()
+            # This does not take the worker out of quarantine, it only resets
+            # the timeout value to default (restarting the default
+            # exponential backoff)
+            wfb.worker.resetQuarantine()
+            return False
+        # Schedule the build now
+        return True
+    return canStartBuild
+
+def is_within_time_range(now, start, end):
+    if start <= end:
+        return start <= now <= end
+    else:
+        return now >= start or now <= end
+
+
+def get_delay(now, end):
+    today = datetime.today()
+    now = datetime.combine(today, now)
+    end = datetime.combine(today, end)
+
+    if now > end:
+        end += timedelta(days=1)
+
+    difference = end - now
+    return difference.total_seconds()

master/master.cfg

@@ -14,7 +14,7 @@ import os
 import subprocess
 import sys
 
-from datetime import datetime, timedelta
+from datetime import timedelta
 from functools import partial
 from zoneinfo import ZoneInfo
 
@@ -52,6 +52,7 @@ from custom.builders import (  # noqa: E402
     STABLE,
     ONLY_MAIN_BRANCH,
 )
+from custom.worker_downtime import no_builds_between
 
 
 def set_up_sentry():
@@ -194,50 +195,6 @@ def is_important_change(change):
     return any(is_important_file(filename) for filename in change.files)
 
 
-def is_within_time_range(now, start, end):
-    if start <= end:
-        return start <= now <= end
-    else:
-        return now >= start or now <= end
-
-
-def get_delay(now, end):
-    today = datetime.today()
-    now = datetime.combine(today, now)
-    end = datetime.combine(today, end)
-
-    if now > end:
-        end += timedelta(days=1)
-
-    difference = end - now
-    return difference.total_seconds()
-
-
-# Avoid a build to be started between start and end time and delay such build
-# at end time
-def no_builds_between(start, end, *, day_of_week=None, tz=None):
-    start = datetime.strptime(start, "%H:%M").time()
-    end = datetime.strptime(end, "%H:%M").time()
-    def canStartBuild(builder, wfb, request):
-        now_dt = datetime.now(tz=tz)
-        if day_of_week is not None and now_dt.weekday() != day_of_week:
-            return True
-        now = now_dt.time()
-        if is_within_time_range(now, start, end):
-            delay = get_delay(now, end)
-            # Adapted from: https://docs.buildbot.net/current/manual/customization.html#canstartbuild-functions
-            wfb.worker.quarantine_timeout = delay
-            wfb.worker.putInQuarantine()
-            # This does not take the worker out of quarantine, it only resets
-            # the timeout value to default (restarting the default
-            # exponential backoff)
-            wfb.worker.resetQuarantine()
-            return False
-        # Schedule the build now
-        return True
-    return canStartBuild
-
-
 github_status_builders = []
 release_status_builders = []
 mail_status_builders = []