feat: ssm deadline window

closes: #571
Change-Id: I841d34e594e8dca4bf8b8772ceadb3e30ad3e6fe

Author: grafuls

conf/selfservice.yml

@@ -8,3 +8,9 @@ ssm_default_lifetime: 5
 ssm_user_cloud_limit: 2
 # Set to true to create a Jira ticket when a self-schedule is created
 ssm_jira_create_ticket: false
+
+# Set to true to enable the scheduling window
+ssm_enable_sched_window: true
+# Set the end date for the scheduling window
+ssm_deadline_day: sunday
+ssm_deadline_hour: 21

src/quads/server/blueprints/schedules.py

@@ -1,4 +1,5 @@
 from datetime import datetime, timedelta
+from calendar import day_name
 
 from flask import Blueprint, Response, g, jsonify, make_response, request
 
@@ -158,7 +159,20 @@ def create_schedule() -> Response:
             return make_response(jsonify(response), 400)
 
         start = datetime.now()
-        end = start + timedelta(days=Config.get("ssm_default_lifetime", 1))
+
+        ssm_deadline_day = Config.get("ssm_deadline_day", "sunday").lower()
+        ssm_deadline_hour = Config.get("ssm_deadline_hour", "21")
+        ssm_default_lifetime = Config.get("ssm_default_lifetime", 1)
+
+        day_mapping = {day.lower(): i for i, day in enumerate(day_name)}
+        target_day = day_mapping.get(ssm_deadline_day)
+        current_day = start.weekday()
+
+        days_ahead = target_day - current_day
+        if days_ahead < ssm_default_lifetime:
+            end = start.replace(hour=ssm_deadline_hour, minute=0, second=0, microsecond=0) + timedelta(days=days_ahead)
+        else:
+            end = start + timedelta(days=ssm_default_lifetime)
     else:
         start = data.get("start")
         end = data.get("end")

tests/api/test_schedules.py

@@ -256,6 +256,7 @@ def test_valid_self(self, mock_datetime, test_client, auth, prefill):
         del response_dict["host"]["default_cloud"]["last_redefined"]
         del response_dict["assignment"]["created_at"]
         del response_dict["assignment"]["cloud"]["last_redefined"]
+        SELF_SCHEDULE_1_RESPONSE["end"] = "Sun, 04 Jun 2023 21:00:00 GMT"
         assert response_dict == SELF_SCHEDULE_1_RESPONSE
 
     @pytest.mark.parametrize("prefill", prefill_self_schedule, indirect=True)