get_nhs_number

Author: JamesW1-NHS

lambdas/backend/src/repository/fhir_repository.py

@@ -16,7 +16,7 @@
 from common.models.constants import Constants
 from common.models.errors import ResourceNotFoundError
 from common.models.immunization_record_metadata import ImmunizationRecordMetadata
-from common.models.utils.generic_utils import get_contained_patient
+from common.models.utils.generic_utils import get_contained_patient, get_nhs_number
 from common.models.utils.validation_utils import (
     get_vaccine_type,
 )
@@ -47,14 +47,6 @@ def _query_identifier(table, index, pk, identifier):
         return queryresponse
 
 
-def get_nhs_number(imms):
-    try:
-        nhs_number = [x for x in imms["contained"] if x["resourceType"] == "Patient"][0]["identifier"][0]["value"]
-    except (KeyError, IndexError):
-        nhs_number = "TBC"
-    return nhs_number
-
-
 @dataclass
 class RecordAttributes:
     pk: str

lambdas/recordforwarder/src/repository/fhir_batch_repository.py

@@ -15,6 +15,7 @@
     ResourceNotFoundError,
     UnhandledResponseError,
 )
+from common.models.utils.generic_utils import get_nhs_number
 
 
 def create_table(region_name="eu-west-2"):
@@ -56,14 +57,6 @@ def _query_identifier(table, index, pk, identifier, is_present):
             return queryresponse
 
 
-def get_nhs_number(imms):
-    try:
-        nhs_number = [x for x in imms["contained"] if x["resourceType"] == "Patient"][0]["identifier"][0]["value"]
-    except (KeyError, IndexError):
-        nhs_number = "TBC"
-    return nhs_number
-
-
 @dataclass
 class RecordAttributes:
     pk: str

lambdas/shared/src/common/models/utils/generic_utils.py

@@ -20,6 +20,14 @@
 from common.models.constants import Constants
 
 
+def get_nhs_number(imms: dict):
+    try:
+        nhs_number = [x for x in imms["contained"] if x["resourceType"] == "Patient"][0]["identifier"][0]["value"]
+    except (KeyError, IndexError):
+        nhs_number = "TBC"
+    return nhs_number
+
+
 def get_contained_resource(imms: dict, resource: Literal["Patient", "Practitioner", "QuestionnaireResponse"]):
     """Extract and return the requested contained resource from the FHIR Immunization Resource JSON data"""
     return [x for x in imms.get("contained") if x.get("resourceType") == resource][0]