added uprn extraction to extract_epc_data:

backend/Property.py

@@ -48,15 +48,17 @@ class Property(Definitions):
 
     spatial = None
 
-    def __init__(self, id, data=None, old_data=None, full_sap_epc=None):
+    def __init__(self, id, address, postcode, data=None, old_data=None, full_sap_epc=None):
         self.id = id
 
+        self.address = address
+        self.postcode = postcode
         self.data = data
         self.old_data = old_data
         self.full_sap_epc = full_sap_epc
         self.property_dimensions = None
 
-        self.uprn = None
+        self.uprn = None if data is not None else data["uprn"]
 
         self.in_conservation_area, self.is_listed, self.is_heritage = None, None, None
         self.restricted_measures = False

backend/SearchEpc.py

@@ -364,7 +364,13 @@ class SearchEpc:
         # Retrieve postcode and address
         address_epc, postcode_epc = self.format_address(newest_epc=newest_epc)
 
-        return newest_epc, older_epcs, full_sap_epc, address_epc, postcode_epc
+        # Ge the uprn from the newest record for this home
+        uprns = {r["uprn"] for r in rows if r["uprn"]}
+        if len(uprns) != 1:
+            raise ValueError("Multiple UPRNs found - investigate me")
+        uprn = uprns.pop()
+
+        return newest_epc, older_epcs, full_sap_epc, address_epc, postcode_epc, uprn
 
     @staticmethod
     def filter_newest_epc(list_of_epcs: List):
@@ -626,7 +632,7 @@ class SearchEpc:
 
         if response["status"] == 200:
             (
-                self.newest_epc, self.older_epcs, self.full_sap_epc, self.address_clean, self.postcode_clean
+                self.newest_epc, self.older_epcs, self.full_sap_epc, self.address_clean, self.postcode_clean, self.uprn
             ) = self.extract_epc_data(address=self.full_address)
             return
 
@@ -642,7 +648,7 @@ class SearchEpc:
         response = self.get_epc()
         if response["status"] == 200:
             (
-                self.newest_epc, self.older_epcs, self.full_sap_epc, self.address_clean, self.postcode_clean
+                self.newest_epc, self.older_epcs, self.full_sap_epc, self.address_clean, self.postcode_clean, self.uprn
             ) = self.extract_epc_data()
             return
 

backend/app/plan/router.py

@@ -98,6 +98,8 @@ async def trigger_plan(body: PlanTriggerRequest):
             input_properties.append(
                 Property(
                     id=property_id,
+                    address=epc_searcher.address_clean,
+                    postcode=epc_searcher.postcode_clean,
                     data=epc_searcher.newest_epc,
                     old_data=epc_searcher.older_epcs,
                     full_sap_epc=epc_searcher.full_sap_epc,