feat: add district to layer

technologiestiftung · Sep 30, 2024 · e5bed7b · e5bed7b
1 parent 23aa9d5
commit e5bed7b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/harvester/src/mapbox_tree_update.py b/harvester/src/mapbox_tree_update.py
@@ -64,6 +64,7 @@ def generate_trees_csv(temp_dir, db_conn):
                     trees.radolan_sum,
                     trees.pflanzjahr,
                     COALESCE(SUM(w.amount), 0) AS watering_sum
+                    trees.bezirk as district
                 FROM
                     trees
                 LEFT JOIN
@@ -87,7 +88,9 @@ def generate_trees_csv(temp_dir, db_conn):
         logging.info(f"Creating trees.csv file for {len(trees)} trees...")
 
         # Build CSV file with all trees in it
-        header = "id,lat,lng,radolan_sum,age,watering_sum,total_water_sum_liters"
+        header = (
+            "id,lat,lng,radolan_sum,age,watering_sum,total_water_sum_liters,district"
+        )
         lines = []
         for tree in tqdm(trees):
             id = tree[0]