Added header

Author: cody-constine-ttd

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "uid2_client"
-version = "1.0.0"
+version = "2.0.0"
 authors = [
     { name = "Cody Constine", email = "cody.constine@thetradedesk.com" }
 ]

tests/sharing_test.py

@@ -138,13 +138,6 @@ def test_cannot_encrypt_if_no_key_from_default_keyset(self):
 
         self.assertRaises(EncryptionError, encrypt_key, _example_uid, IdentityScope.UID2, keys)
 
-#    def test_cannot_encrypt_if_theres_no_default_keyset_header(self):
-#        client = Uid2Client("endpoint", "authkey", _client_secret)
-#        json_body = self._key_set_to_json_for_sharing_with_header("", 99999, [_master_key, _site_key])
-#        keys = client.refresh_json(json_body)
-#
-#        self.assertRaises(EncryptionError, encrypt_key, _example_uid, IdentityScope.UID2, keys)
-
     def test_expiry_in_token_matches_expiry_in_reponse(self):
         client = Uid2Client("endpoint", "authkey", _client_secret)
         json_body = self._key_set_to_json_for_sharing_with_header('"default_keyset_id": 99999, "token_expiry_seconds": 2,', 99999, [_master_key, _site_key])

uid2_client/client.py

@@ -11,6 +11,7 @@
 import json
 import os
 import urllib.request as request
+import pkg_resources
 
 from .keys import EncryptionKey, EncryptionKeysCollection
 from .encryption import _decrypt_gcm, _encrypt_gcm
@@ -93,7 +94,7 @@ def _make_url(self, path):
         return self._base_url + path
 
     def _auth_headers(self):
-        return {'Authorization': 'Bearer ' + self._auth_key}
+        return {'Authorization': 'Bearer ' + self._auth_key, "X-UID2-Client-Version": pkg_resources.get_distribution("uid2_client").version}
 
     def _make_v2_request(self, now):
         payload = int.to_bytes(int(now.timestamp() * 1000), 8, 'big')