fix(fhir): Resolved issue with GCP not using private URLs when return…

…ing paging URLs
hms-dbmi · Jul 11, 2024 · 6a02233 · 6a02233
1 parent 6028ad3
commit 6a02233
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ppmutils/fhir.py b/ppmutils/fhir.py
@@ -2779,6 +2779,12 @@ def _query_bundle(resource_type: str, query: dict[str, Any] = None) -> Bundle:
                 if link["relation"] == "next":
                     url = link["url"]
 
+                    # Swap domain if necessary
+                    if furl(url).host != furl(PPM.fhir_url()).host:
+
+                        # Set it
+                        url = furl(url).set(host=furl(PPM.fhir_url().host)).url
+
         return Bundle(total_bundle)
 
     @staticmethod