From b791ecb054f0e5be39f91f78771f74ed80fe904d Mon Sep 17 00:00:00 2001
From: Khalim Conn-Kowlessar <kconnkowlessar@gmail.com>
Date: Wed, 10 Apr 2024 12:08:47 +0100
Subject: [PATCH] set up asset list

---
 etl/customers/immo/pilot/asset_list.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/etl/customers/immo/pilot/asset_list.py b/etl/customers/immo/pilot/asset_list.py
index 33f79729..269ffe00 100644
--- a/etl/customers/immo/pilot/asset_list.py
+++ b/etl/customers/immo/pilot/asset_list.py
@@ -41,4 +41,23 @@ def app():
     raw_asset_list["address"] = raw_asset_list["Full Address"].str.split(",").str[0]
     raw_asset_list["postcode"] = raw_asset_list["Full Address"].str.split(",").str[-1].str.strip()
 
-    raw_asset_list[["address", "postcode"]].to_dict("records")
+    council_tax_bands = pd.DataFrame(council_tax_bands)
+    asset_list = raw_asset_list.merge(council_tax_bands, how="left", on=["address", "postcode"])
+
+    # Store the data in s3
+    filename = f"{USER_ID}/{PORTFOLIO_ID}/pilot.csv"
+    save_csv_to_s3(
+        dataframe=asset_list,
+        bucket_name="retrofit-plan-inputs-dev",
+        file_name=filename
+    )
+
+    body = {
+        "portfolio_id": str(PORTFOLIO_ID),
+        "housing_type": "Private",
+        "goal": "Increase EPC",
+        "goal_value": "A",
+        "trigger_file_path": filename,
+        "budget": None,
+    }
+    print(body)