From d34a4d4d963d349877d63a44753549186247a64d Mon Sep 17 00:00:00 2001
From: Khalim Conn-Kowlessar <kconnkowlessar@gmail.com>
Date: Thu, 28 Mar 2024 14:32:29 +0000
Subject: [PATCH] allowing passage of uprn to Searcher in api

---
 .idea/Model.iml             | 2 +-
 .idea/misc.xml              | 2 +-
 backend/app/plan/router.py  | 4 ++++
 backend/app/plan/schemas.py | 2 ++
 4 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/.idea/Model.iml b/.idea/Model.iml
index b0f9c00d..4413bb06 100644
--- a/.idea/Model.iml
+++ b/.idea/Model.iml
@@ -7,7 +7,7 @@
       <sourceFolder url="file://$MODULE_DIR$/open_uprn" isTestSource="false" />
       <sourceFolder url="file://$MODULE_DIR$/recommendations" isTestSource="false" />
     </content>
-    <orderEntry type="jdk" jdkName="Python 3.10 (model_data)" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Python 3.10 (backend)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
   <component name="PyNamespacePackagesService">
diff --git a/.idea/misc.xml b/.idea/misc.xml
index 1122b380..6f308057 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -3,7 +3,7 @@
   <component name="Black">
     <option name="sdkName" value="Python 3.10 (backend)" />
   </component>
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (model_data)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (backend)" project-jdk-type="Python SDK" />
   <component name="PythonCompatibilityInspectionAdvertiser">
     <option name="version" value="3" />
   </component>
diff --git a/backend/app/plan/router.py b/backend/app/plan/router.py
index 0b98cf2c..5456cdb6 100644
--- a/backend/app/plan/router.py
+++ b/backend/app/plan/router.py
@@ -91,10 +91,14 @@ async def trigger_plan(body: PlanTriggerRequest):
         input_properties = []
         for config in tqdm(plan_input):
             # We validate each record in the file. If the record is NOT valid, we need to handle this accordingly
+            uprn = config.get("uprn", None)
+            if uprn:
+                uprn = int(float(uprn))
 
             epc_searcher = SearchEpc(
                 address1=config["address"],
                 postcode=config["postcode"],
+                uprn=uprn,
                 auth_token=get_settings().EPC_AUTH_TOKEN,
                 os_api_key=get_settings().ORDNANCE_SURVEY_API_KEY
             )
diff --git a/backend/app/plan/schemas.py b/backend/app/plan/schemas.py
index 9801375f..1e95fb2f 100644
--- a/backend/app/plan/schemas.py
+++ b/backend/app/plan/schemas.py
@@ -8,3 +8,5 @@ class PlanTriggerRequest(BaseModel):
     goal_value: str
     portfolio_id: int
     trigger_file_path: str
+    # optional exclusions list
+    exclusions: list[str] | None = None