inserting county

backend/app/plan/router.py

@@ -241,6 +241,10 @@ def create_epc_records(epc_searcher: SearchEpc, energy_assessment: dict):
     epc = energy_assessment["epc"]
     energy_assessment_date = epc["inspection-date"].strftime("%Y-%m-%d")
 
+    # We insert county into the epc, since right now this isn't something that we pull out from the energy
+    # assessment
+    epc["county"] = epc_searcher.newest_epc["county"]
+
     # We check if the energy assessment is newer than the newest EPC
     if pd.to_datetime(energy_assessment_date) > pd.to_datetime(epc_searcher.newest_epc["inspection-date"]):
         # In this case, our energy assessment is newer than the EPCs available for this property

etl/xml_survey_extraction/app.py

@@ -29,6 +29,8 @@ def main():
     #      The data is stored in a folder called {surveyors}/{project_code}/{uprn}
     #       We'll need to get the uprn from the folder name, which we can do with EpcSearcher class
 
+    # TODO: Pull out county, as in create_epc_records in the router, we pull it from the latest EPC, but we should
+    #       be able to deduce it from just the address
     #
     energy_assessments = list_files_and_subfolders_in_s3_folder(
         bucket_name=BUCKET, folder_name=f"{SURVEYORS}/{PROJECT_CODE}/"