fix: set all waterings to included_in_map_layer = false first

technologiestiftung · May 28, 2024 · c9814c7 · c9814c7
1 parent fe8da4b
commit c9814c7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/harvester/src/mapbox_tree_update.py b/harvester/src/mapbox_tree_update.py
@@ -182,6 +182,16 @@ def update_tree_waterings(trees_watered, db_conn):
     print(f"Set included_in_map_layer = TRUE for {len(trees_watered)} watered trees...")
     tree_ids = [tree_watered[4] for tree_watered in trees_watered]
     with db_conn.cursor() as cur:
+        # Set included_in_map_layer = FALSE for all waterings
+        cur.execute(
+            """
+                UPDATE trees_watered SET included_in_map_layer = FALSE WHERE TRUE;
+            """,
+            (tree_ids,),
+        )
+        db_conn.commit()
+
+        # Set included_in_map_layer = FALSE for the waterings that are included in this round of the harvester
         cur.execute(
             """
                 UPDATE trees_watered SET included_in_map_layer = TRUE WHERE id = ANY (%s);

diff --git a/harvester/src/run_harvester.py b/harvester/src/run_harvester.py
@@ -90,4 +90,5 @@
         database_connection,
     )
 
+    # Update the tree waterings and flag the waterings which are now included in the mapbox layer with included_in_map_layer = TRUE
     update_tree_waterings(trees_watered, database_connection)