Add on-call endpoints for overrides to the spec

examples/v2/on-call/CreateScheduleOverrides.py

@@ -0,0 +1,51 @@
+"""
+Create On-Call schedule overrides returns "Created" response
+"""
+
+from datetime import datetime
+from dateutil.relativedelta import relativedelta
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.on_call_api import OnCallApi
+from datadog_api_client.v2.model.create_override_request_attributes import CreateOverrideRequestAttributes
+from datadog_api_client.v2.model.create_override_request_data import CreateOverrideRequestData
+from datadog_api_client.v2.model.create_override_request_relationships import CreateOverrideRequestRelationships
+from datadog_api_client.v2.model.create_overrides_request import CreateOverridesRequest
+from datadog_api_client.v2.model.override_data_type import OverrideDataType
+from datadog_api_client.v2.model.override_relationships_user import OverrideRelationshipsUser
+from datadog_api_client.v2.model.override_relationships_user_data import OverrideRelationshipsUserData
+from datadog_api_client.v2.model.override_relationships_user_data_type import OverrideRelationshipsUserDataType
+
+# there is a valid "schedule" in the system
+SCHEDULE_DATA_ID = environ["SCHEDULE_DATA_ID"]
+
+# there is a valid "user" in the system
+USER_DATA_ID = environ["USER_DATA_ID"]
+
+body = CreateOverridesRequest(
+    data=[
+        CreateOverrideRequestData(
+            attributes=CreateOverrideRequestAttributes(
+                end=(datetime.now() + relativedelta(days=1)),
+                start=datetime.now(),
+            ),
+            relationships=CreateOverrideRequestRelationships(
+                user=OverrideRelationshipsUser(
+                    data=OverrideRelationshipsUserData(
+                        id=USER_DATA_ID,
+                        type=OverrideRelationshipsUserDataType.USERS,
+                    ),
+                ),
+            ),
+            type=OverrideDataType.OVERRIDES,
+        ),
+    ],
+)
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+with ApiClient(configuration) as api_client:
+    api_instance = OnCallApi(api_client)
+    response = api_instance.create_schedule_overrides(schedule_id=SCHEDULE_DATA_ID, body=body)
+
+    print(response)

examples/v2/on-call/DeleteScheduleOverride.py

@@ -0,0 +1,22 @@
+"""
+Delete On-Call schedule override returns "No Content" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.on_call_api import OnCallApi
+
+# there is a valid "schedule" in the system
+SCHEDULE_DATA_ID = environ["SCHEDULE_DATA_ID"]
+
+# there is a valid "override" in the system
+OVERRIDE_DATA_0_ID = environ["OVERRIDE_DATA_0_ID"]
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+with ApiClient(configuration) as api_client:
+    api_instance = OnCallApi(api_client)
+    api_instance.delete_schedule_override(
+        schedule_id=SCHEDULE_DATA_ID,
+        override_id=OVERRIDE_DATA_0_ID,
+    )

examples/v2/on-call/ListScheduleOverrides.py

@@ -0,0 +1,24 @@
+"""
+List On-Call schedule overrides returns "OK" response
+"""
+
+from datetime import datetime
+from dateutil.relativedelta import relativedelta
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.on_call_api import OnCallApi
+
+# there is a valid "schedule" in the system
+SCHEDULE_DATA_ID = environ["SCHEDULE_DATA_ID"]
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+with ApiClient(configuration) as api_client:
+    api_instance = OnCallApi(api_client)
+    response = api_instance.list_schedule_overrides(
+        schedule_id=SCHEDULE_DATA_ID,
+        filter_start=(datetime.now() + relativedelta(days=-1)),
+        filter_end=(datetime.now() + relativedelta(days=2)),
+    )
+
+    print(response)

examples/v2/on-call/ListScheduleOverrides_1773803881.py

@@ -0,0 +1,21 @@
+"""
+List On-Call schedule overrides returns "OK" response with pagination
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.on_call_api import OnCallApi
+from datetime import datetime
+from dateutil.tz import tzutc
+
+configuration = Configuration()
+configuration.access_token = environ["DD_BEARER_TOKEN"]
+with ApiClient(configuration) as api_client:
+    api_instance = OnCallApi(api_client)
+    items = api_instance.list_schedule_overrides_with_pagination(
+        schedule_id="3653d3c6-0c75-11ea-ad28-fb5701eabc7d",
+        filter_start=datetime(2024, 1, 7, 2, 53, 1, tzinfo=tzutc()),
+        filter_end=datetime(2024, 1, 14, 2, 53, 1, tzinfo=tzutc()),
+    )
+    for item in items:
+        print(item)