total weight in metric tonnes

db/db_utils.py

@@ -78,6 +78,7 @@ def initialize_db(conn):
             donations_discount REAL,
             total_emissions_reduction_pre REAL,
             total_emissions_reduction REAL,
+            total_weight_metric_tonnes REAL,
             created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
         )

rollups.py

@@ -74,21 +74,30 @@ for run_meta_row in run_meta_rows:
     else:
         total_emissions_reduction_pre = 0
 
+    # calculate total_weight
+    db_cursor.execute("SELECT sum(weight_metric_tonnes) FROM results WHERE run_key = %s", (run_key,))
+    weight_result = db_cursor.fetchone()
+    if weight_result and weight_result[0]:
+        total_weight_metric_tonnes = weight_result[0]
+    else:
+        total_weight_metric_tonnes = 0
+
     #  divide the donations_from by 2 because we attribute the emissions reductions to both the donor and the recipient
     donations_discount = donations / 2
     total_emissions_reduction = total_emissions_reduction_pre - donations_discount
 
     # store this data in the rollups table
     db_cursor.execute("""
-        INSERT INTO rollups (run_key, year, donations_discount, total_emissions_reduction_pre, total_emissions_reduction)
-        VALUES (%s, %s, %s, %s, %s)
+        INSERT INTO rollups (run_key, year, donations_discount, total_emissions_reduction_pre, total_emissions_reduction, total_weight_metric_tonnes)
+        VALUES (%s, %s, %s, %s, %s, %s)
         ON CONFLICT (run_key)
         DO UPDATE SET
             year = EXCLUDED.year,
             donations_discount = EXCLUDED.donations_discount,
             total_emissions_reduction_pre = EXCLUDED.total_emissions_reduction_pre,
-            total_emissions_reduction = EXCLUDED.total_emissions_reduction
-    """, (run_key, year, donations_discount, total_emissions_reduction_pre, total_emissions_reduction))
+            total_emissions_reduction = EXCLUDED.total_emissions_reduction,
+            total_weight_metric_tonnes = EXCLUDED.total_weight_metric_tonnes
+    """, (run_key, year, donations_discount, total_emissions_reduction_pre, total_emissions_reduction, total_weight_metric_tonnes, ))
     db_conn.commit()
 
 db_cursor.close()

run.py

@@ -17,8 +17,9 @@ if __name__ == "__main__":
     # remove test.csv from raw_files
     raw_files = [f for f in raw_files if f != 'test.csv']
 
-    # for raw_file_name in ['test.csv']:
-    for raw_file_name in raw_files:
+    # for raw_file_name in ['sharing-excess-2020-raw-data.csv', 'sharing-excess-2021-raw-data.csv', 'sharing-excess-2022-raw-data.csv', 'sharing-excess-2023-raw-data.csv']:
+    for raw_file_name in ['spoonfuls-2023-Raw-Data.csv']:
+    # for raw_file_name in raw_files:
         if not raw_file_name.endswith('.csv'):
             continue