diff --git a/etl/ownership/Ownership.py b/etl/ownership/Ownership.py index 25ba7cea..8221f8f7 100644 --- a/etl/ownership/Ownership.py +++ b/etl/ownership/Ownership.py @@ -109,6 +109,9 @@ class Ownership: self.portfolio_properties_filepath = ( f"ownership/{self.project_name}/{self.run_timestamp}/portfolio_properties.xlsx" ) + self.portfolio_epc_data_filepath = ( + f"ownership/{self.project_name}/{self.run_timestamp}/portfolio_epc_data.xlsx" + ) # Data self.epc_data = None @@ -999,3 +1002,19 @@ class Ownership: bucket_name=self.bucket, file_key=self.portfolio_epc_data_filepath, ) + + def get_asset_list(self): + """ + From the EPC data, creates the asset list + :return: + """ + + asset_list = self.portfolio_epc_data[["UPRN", "ADDRESS1", "POSTCODE"]].copy().rename( + columns={ + "UPRN": "uprn", + "ADDRESS1": "address", + "POSTCODE": "postcode" + } + ) + + return asset_list diff --git a/etl/ownership/projects/midlands_portfolio/app.py b/etl/ownership/projects/midlands_portfolio/app.py index 2165dc94..905ff0e4 100644 --- a/etl/ownership/projects/midlands_portfolio/app.py +++ b/etl/ownership/projects/midlands_portfolio/app.py @@ -122,3 +122,6 @@ def app(): ownership_instance.pipeline(column_filters=epc_column_filters) # Create the project, if a portfolio doesn't exist for the project name + + # Create the asset list and the body of the portfolio + asset_list = ownership_instance.get_asset_list()