fixup! Add tracking headers

datatrails · Jun 12, 2024 · 507b124 · 507b124
1 parent d486fd4
commit 507b124
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/archivist/archivistpublic.py b/archivist/archivistpublic.py
@@ -42,6 +42,7 @@
     HEADERS_TOTAL_COUNT,
     PARTNER_ID,
     USER_AGENT,
+    USER_AGENT_PREFIX,
 )
 from .dictmerge import _deepmerge, _dotdict
 from .errors import (
@@ -185,7 +186,7 @@ def __copy__(self):
 
     def _add_headers(self, headers: "dict[str, str]|None") -> "dict[str, str]":
         newheaders = {**headers} if headers is not None else {}
-        newheaders[USER_AGENT] = f"datatrails/pysdk/{self.version}"
+        newheaders[USER_AGENT] = f"{USER_AGENT_PREFIX}{self.version}"
         p = self.partner_id
         if p:
             newheaders[PARTNER_ID] = p

diff --git a/archivist/constants.py b/archivist/constants.py
@@ -13,9 +13,9 @@
     "RecordEvidence",
 ]
 
-USER_AGENT = "Archivist-User-Agent"
-USER_AGENT_PREFIX = "datatrails/pysdk/"
-PARTNER_ID = "Archivist-Partner-ID"
+USER_AGENT = "DataTrails-User-Agent"
+USER_AGENT_PREFIX = "pysdk/"
+PARTNER_ID = "DataTrails-Partner-ID"
 
 # define in MIME canonical form
 HEADERS_REQUEST_TOTAL_COUNT = "X-Request-Total-Count"