From e9471b338c0ef6113d6133dffd4fdd2f437c05e7 Mon Sep 17 00:00:00 2001
From: Jona <jona.raphael@gmail.com>
Date: Sun, 26 Nov 2023 15:28:08 -0500
Subject: [PATCH] Avoid deactivating slicks if in DRYRUN

---
 .../cloud_run_orchestrator/handler.py         | 90 ++++++++++---------
 1 file changed, 47 insertions(+), 43 deletions(-)

diff --git a/cerulean_cloud/cloud_run_orchestrator/handler.py b/cerulean_cloud/cloud_run_orchestrator/handler.py
index 244f5ce2..8a19478c 100644
--- a/cerulean_cloud/cloud_run_orchestrator/handler.py
+++ b/cerulean_cloud/cloud_run_orchestrator/handler.py
@@ -345,53 +345,57 @@ async def _orchestrate(
 
     # write to DB
     async with DatabaseClient(db_engine) as db_client:
-        try:
-            async with db_client.session.begin():
-                trigger = await db_client.get_trigger(trigger=payload.trigger)
-                layers = [await db_client.get_layer(layer) for layer in model.layers]
-                sentinel1_grd = await db_client.get_sentinel1_grd(
-                    payload.sceneid,
-                    scene_info,
-                    titiler_client.get_base_tile_url(
+        if not payload.dry_run:
+            try:
+                async with db_client.session.begin():
+                    trigger = await db_client.get_trigger(trigger=payload.trigger)
+                    layers = [
+                        await db_client.get_layer(layer) for layer in model.layers
+                    ]
+                    sentinel1_grd = await db_client.get_sentinel1_grd(
                         payload.sceneid,
-                        rescale=(0, 255),
-                    ),
-                )
-                stale_slick_count = (
-                    await db_client.deactivate_stale_slicks_from_scene_id(
-                        payload.sceneid
+                        scene_info,
+                        titiler_client.get_base_tile_url(
+                            payload.sceneid,
+                            rescale=(0, 255),
+                        ),
                     )
-                )
-                print(
-                    f"{start_time}: Deactivating {stale_slick_count} slicks from stale runs on {payload.sceneid}."
-                )
-                orchestrator_run = await db_client.add_orchestrator(
-                    start_time,
-                    start_time,
-                    ntiles,
-                    noffsettiles,
-                    os.getenv("GIT_HASH"),
-                    os.getenv("GIT_TAG"),
-                    make_cloud_log_url(
-                        os.getenv("CLOUD_RUN_NAME"), start_time, os.getenv("PROJECT_ID")
-                    ),
-                    zoom,
-                    scale,
-                    scene_bounds,
-                    trigger,
-                    model,
-                    sentinel1_grd,
-                )
-        except:  # noqa: E722
-            await db_client.session.close()
-            raise
+                    stale_slick_count = (
+                        await db_client.deactivate_stale_slicks_from_scene_id(
+                            payload.sceneid
+                        )
+                    )
+                    print(
+                        f"{start_time}: Deactivating {stale_slick_count} slicks from stale runs on {payload.sceneid}."
+                    )
+                    orchestrator_run = await db_client.add_orchestrator(
+                        start_time,
+                        start_time,
+                        ntiles,
+                        noffsettiles,
+                        os.getenv("GIT_HASH"),
+                        os.getenv("GIT_TAG"),
+                        make_cloud_log_url(
+                            os.getenv("CLOUD_RUN_NAME"),
+                            start_time,
+                            os.getenv("PROJECT_ID"),
+                        ),
+                        zoom,
+                        scale,
+                        scene_bounds,
+                        trigger,
+                        model,
+                        sentinel1_grd,
+                    )
+            except:  # noqa: E722
+                await db_client.session.close()
+                raise
 
-        inference_parms = {
-            "model_type": model.type,
-            "thresholds": model.thresholds,
-        }
+            inference_parms = {
+                "model_type": model.type,
+                "thresholds": model.thresholds,
+            }
 
-        if not payload.dry_run:
             success = True
             try:
                 print(f"{start_time}: Instantiating inference client.")