Add global mute flag support

README.md

@@ -143,6 +143,10 @@ Options:
   --force-missing-dependencies                Force importing and syncing resources that
                                               could be potential dependencies to the
                                               requested resources. [sync only]
+  --create-global-downtime                    Scheduled downtime is meant to be removed
+                                              during failover when user determines
+                                              monitors have enough telemetry to trigger
+                                              appropriately.
   --help                                      Show this message and exit.
 
 Commands:

datadog_sync/commands/sync.py

@@ -33,6 +33,15 @@
     help="Force importing and syncing resources that could be potential dependencies to the requested resources.",
     cls=CustomOptionClass,
 )
+@option(
+    "--create-global-downtime",
+    required=False,
+    is_flag=True,
+    default=False,
+    help="Scheduled downtime is meant to be removed during failover when "
+    "user determines monitors have enough telemetry to trigger appropriately.",
+    cls=CustomOptionClass,
+)
 def sync(**kwargs):
     """Sync Datadog resources to destination."""
     cfg = build_config(CMD_SYNC, **kwargs)

datadog_sync/utils/configuration.py

@@ -44,6 +44,7 @@ class Configuration(object):
     skip_failed_resource_connections: bool
     max_workers: int
     cleanup: int
+    create_global_downtime: bool
     resources: Dict[str, BaseResource] = field(default_factory=dict)
     resources_arg: List[str] = field(default_factory=list)
 
@@ -88,6 +89,7 @@ def build_config(cmd: str, **kwargs: Optional[Any]) -> Configuration:
     force_missing_dependencies = kwargs.get("force_missing_dependencies")
     skip_failed_resource_connections = kwargs.get("skip_failed_resource_connections")
     max_workers = kwargs.get("max_workers", 10)
+    create_global_downtime = kwargs.get("create_global_downtime")
 
     cleanup = kwargs.get("cleanup")
     if cleanup:
@@ -108,6 +110,7 @@ def build_config(cmd: str, **kwargs: Optional[Any]) -> Configuration:
         skip_failed_resource_connections=skip_failed_resource_connections,
         max_workers=max_workers,
         cleanup=cleanup,
+        create_global_downtime=create_global_downtime,
     )
 
     # Initialize resources

datadog_sync/utils/resource_utils.py

@@ -92,6 +92,38 @@ def give_up_diffing(self, level, diff_instance) -> bool:
         return False
 
 
+def create_global_downtime(config: Configuration):
+    """Create global downtime"""
+    payload = {
+        "data": {
+            "attributes": {
+                "message": "Downtime created by datadog-sync-cli to mute all monitors synced. "
+                "To be manually removed during failover when monitors have enough telemetry"
+                "to trigger appropriately.",
+                "monitor_identifier": {"monitor_tags": DEFAULT_TAGS},
+                "scope": "*",
+                "schedule": {
+                    "start": None,
+                },
+            },
+            "type": "downtime",
+        }
+    }
+
+    try:
+        resp = config.destination_client.post(
+            config.resources["downtime_schedules"].resource_config.base_path, payload
+        ).json()
+        config.logger.info(f"Global downtime for datadog-sync-cli created successfully - {resp['data']['id']}")
+    except CustomClientHTTPError as e:
+        if e.status_code == 400 and "downtime being created is a duplicate" in str(e):
+            config.logger.info(f"Global downtime for datadog-sync-cli already exists: {str(e)}")
+        else:
+            config.logger.error(f"Error creating global downtime for datadog-sync-cli: {str(e)}")
+    except Exception as e:
+        config.logger.error(f"Error creating global downtime for datadog-sync-cli: {str(e)}")
+
+
 def find_attr(keys_list_str: str, resource_to_connect: str, r_obj: Any, connect_func: Callable) -> Optional[List[str]]:
     if not r_obj:
         return None

datadog_sync/utils/resources_handler.py

@@ -19,6 +19,7 @@
     ResourceConnectionError,
     SkipResource,
     check_diff,
+    create_global_downtime,
     dump_resources,
     prep_resource,
     thread_pool_executor,
@@ -88,6 +89,11 @@ def apply_resources(self) -> Tuple[int, int]:
         # Run pre-apply hooks
         for resource_type in set(self.resources_manager.all_resources.values()):
             futures.append(parralel_executor.submit(self.config.resources[resource_type]._pre_apply_hook))
+
+        # Additional pre-apply actions
+        if self.config.create_global_downtime:
+            futures.append(parralel_executor.submit(create_global_downtime, self.config))
+
         wait(futures)
         for future in futures:
             try:

tests/conftest.py

@@ -132,6 +132,7 @@ def config():
         force_missing_dependencies=False,
         skip_failed_resource_connections=True,
         cleanup=False,
+        create_global_downtime=False,
     )
 
     resources = init_resources(cfg)