Refactor usage of token in requests

Author: Henry Chan

workos/audit_log.py

@@ -51,13 +51,18 @@ def create_event(self, event, idempotency_key=None):
             dict: Response from WorkOS
         """
         if len(event.get("metadata", {})) > METADATA_LIMIT:
-            raise Exception("Number of metadata keys exceeds %d." % METADATA_LIMIT)
+            raise Exception(
+                "Number of metadata keys exceeds {}.".format(METADATA_LIMIT)
+            )
 
         headers = {
-            "Authorization": "Bearer %s" % workos.api_key,
             "idempotency_key": idempotency_key,
         }
 
         return self.request_helper.request(
-            EVENTS_PATH, method=REQUEST_METHOD_POST, params=event, headers=headers
+            EVENTS_PATH,
+            method=REQUEST_METHOD_POST,
+            params=event,
+            headers=headers,
+            token=workos.api_key,
         )

workos/utils/request.py

@@ -33,7 +33,9 @@ def set_base_api_url(self, base_api_url):
     def generate_api_url(self, path):
         return self.base_api_url.format(path)
 
-    def request(self, path, method=REQUEST_METHOD_GET, params=None, headers=None):
+    def request(
+        self, path, method=REQUEST_METHOD_GET, params=None, headers=None, token=None
+    ):
         """Executes a request against the WorkOS API.
 
         Args:
@@ -42,12 +44,17 @@ def request(self, path, method=REQUEST_METHOD_GET, params=None, headers=None):
         Kwargs:
             method (str): One of the supported methods as defined by the REQUEST_METHOD_X constants
             params (dict): Query params to be added to the request
+            token (str): Bearer token
 
         Returns:
             dict: Response from WorkOS
         """
         if headers is None:
             headers = {}
+
+        if token:
+            headers["Authorization"] = "Bearer {}".format(token)
+
         headers.update(BASE_HEADERS)
         url = self.generate_api_url(path)