junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-07-12 13:29:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

set up asset list

Browse source
This commit is contained in:
Khalim Conn-Kowlessar 2024-04-10 12:08:47 +01:00
parent cf7627a8d7
commit b791ecb054
1 changed files with 20 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
etl/customers/immo/pilot/asset_list.py
View file

@ -41,4 +41,23 @@ def app():
raw_asset_list["address"] = raw_asset_list["Full Address"].str.split(",").str[0] 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["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)