fix: add correlation id

BHTOM-Team · Feb 1, 2024 · 0efe36f · 0efe36f
1 parent 52e5091
commit 0efe36f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/bhtom2/bhtom_catalogs/views.py b/bhtom2/bhtom_catalogs/views.py
@@ -5,6 +5,7 @@
 import requests
 from django.conf import settings
 import json
+from django_guid import get_guid
 
 from .forms import CatalogQueryForm
 from bhtom_base.bhtom_catalogs.harvester import MissingDataException
@@ -35,7 +36,10 @@ def form_valid(self, form):
                 'terms': term,
                 'harvester': service
             }
-            response = requests.post(settings.HARVESTER_URL + '/findTargetWithHarvester/', data=post_data)
+            header = {
+                "Correlation-ID" : get_guid(),
+            }
+            response = requests.post(settings.HARVESTER_URL + '/findTargetWithHarvester/', data=post_data, headers=header)
 
             if response.status_code == 200:
                 # Extract JSON from the response

diff --git a/bhtom2/bhtom_targets/utils.py b/bhtom2/bhtom_targets/utils.py
@@ -20,6 +20,7 @@
 from django.conf import settings
 import requests
 import json
+from django_guid import get_guid
 from math import radians
 from bhtom_base.bhtom_common.hooks import run_hook
 from astropy.coordinates import Angle
@@ -114,9 +115,12 @@ def import_targets(targets):
                         post_data = {
                         'terms': gaia_alerts_name,
                         'harvester': "Gaia Alerts"
-                        }            
+                        }       
+                        header = {
+                            "Correlation-ID" : get_guid(),
+                        }     
                         try:
-                            response = requests.post(settings.HARVESTER_URL + '/findTargetWithHarvester/', data=post_data)
+                            response = requests.post(settings.HARVESTER_URL + '/findTargetWithHarvester/', data=post_data, headers=header)
                             if response.status_code == 200:
                                 # Extract JSON from the response
                                 catalog_data = json.loads(response.text)