feat: gracefully handle existing environments during creation and add a 10-second delay after listing target DAGs.

ddeleo · ddeleo · commit 24a6d8359756 · 2026-02-18T15:40:29.000-05:00
diff --git a/composer/tools/composer_migrate.py b/composer/tools/composer_migrate.py
@@ -78,6 +78,10 @@ def create_environment_from_config(self, config: Any) -> Any:
         # The original code did: POST .../environments with body.
         
         response = self.session.post(url, json=config)
+        if response.status_code == 409:
+            logger.info("Environment already exists, skipping creation.")
+            return
+
         if response.status_code != 200:
             raise RuntimeError(
                 f"Failed to create environment: {response.text}"
@@ -488,6 +492,7 @@ def main(
         target_env_dag_ids = [dag["dag_id"] for dag in target_env_dags]
         all_dags_present = set(source_env_dag_ids) == set(target_env_dag_ids)
         logger.info("List of DAGs in the target environment: %s", target_env_dag_ids)
+        time.sleep(10)
     # Unpause only DAGs that were not paused in the source environment.
     # Optimization: if all DAGs were unpaused in source, use bulk unpause.
     if not any(d["is_paused"] for d in source_env_dags):
