mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
import os
|
|
|
|
import pandas as pd
|
|
from dotenv import load_dotenv
|
|
|
|
from etl.spatial.OpenUprnClient import OpenUprnClient
|
|
from etl.route_march_data_pull.app import get_data
|
|
|
|
load_dotenv(dotenv_path="backend/.env")
|
|
EPC_AUTH_TOKEN = os.getenv("EPC_AUTH_TOKEN")
|
|
|
|
addresses = [
|
|
{"address": "3 Willis Road", "postcode": "CB1 2AQ"},
|
|
{"address": "22 Catharine Street", "postcode": "CB1 3AW"},
|
|
{"address": "332 Mill Road", "postcode": "CB1 3NN"},
|
|
{"address": "330 Mill Road", "postcode": "CB1 3NN"},
|
|
{"address": "328 Mill Road", "postcode": "CB1 3NN"},
|
|
{"address": "71 Mill Road", "postcode": "CB1 2AS"},
|
|
{"address": "78 Argyle Street", "postcode": "CB1 3LZ"},
|
|
{"address": "9 Graham Road", "postcode": "CB4 2ZE"},
|
|
{"address": "217 Mill Road", "postcode": "CB1 3BE"},
|
|
{"address": "374 Mill Road", "postcode": "CB1 3NN"},
|
|
{"address": "174 Thoday Street", "postcode": "CB1 3AX"},
|
|
{"address": "37 Abbey Road", "postcode": "CB5 8HH"},
|
|
{"address": "18 Upper Gwydir Street", "postcode": "CB1 2LR"},
|
|
{"address": "21 Fulbourn Road Fulbourn", "postcode": "CB1 9JL"},
|
|
{"address": "108 Argyle Street", "postcode": "CB1 3LS"},
|
|
{"address": "115 Victoria Road", "postcode": "CB4 3BS"},
|
|
{"address": "55 Ross Street", "postcode": "CB1 3BP"},
|
|
{"address": "16 Kingston Street", "postcode": "CB1 2NU"},
|
|
{"address": "13 Thoday Street", "postcode": "CB1 3AS"},
|
|
{"address": "103 York Street", "postcode": "CB1 2PZ"},
|
|
]
|
|
|
|
asset_list = pd.DataFrame(addresses)
|
|
asset_list["row_id"] = asset_list.index
|
|
|
|
epc_data, _, _ = get_data(
|
|
asset_list=asset_list, fulladdress_column="address", postcode_column="postcode", address1_column="address",
|
|
manual_uprn_map={}, epc_api_only=True
|
|
)
|
|
|
|
epc_df = pd.DataFrame(epc_data)
|
|
epc_df.shape
|
|
|
|
asset_list = asset_list.merge(
|
|
epc_df, how="left", on="row_id"
|
|
)
|
|
|
|
asset_list = asset_list.rename(columns={"address_x": "Address", "postcode_x": "Postcode"})
|
|
asset_list["uprn"] = asset_list["uprn"].astype(str)
|
|
|
|
spatial_data = OpenUprnClient.get_spatial_data([x["uprn"] for x in epc_data], bucket_name="retrofit-data-dev")
|
|
spatial_data["UPRN"] = spatial_data["UPRN"].astype(str)
|
|
|
|
asset_list = asset_list.merge(
|
|
spatial_data, how="left", left_on="uprn", right_on="UPRN"
|
|
)
|
|
|
|
asset_list.to_csv("/Users/khalimconn-kowlessar/Documents/hestia/Customers/Panacap/Acquisitions EPC Data.csv",
|
|
index=False)
|