Add authorization token to use GCP IAP (#16)

Using the header `Proxy-Authorization` does not work because it is deleted by https://github.com/psf/requests/blob/2a438c27b5a5828c8ea0dc958112eecffca70b12/src/requests/sessions.py#L318-L319.
hnhdev · Apr 8, 2024 · 476b3f8 · 476b3f8
1 parent 9053f9a
commit 476b3f8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/api_client/python/timesketch_api_client/client.py b/api_client/python/timesketch_api_client/client.py
@@ -340,6 +340,10 @@ def _create_session(
 
         session = requests.Session()
 
+        # GCP IAP
+        if token := os.getenv("AUTHORIZATION_TOKEN"):
+            session.headers = {"Authorization": f"Bearer {token}"}
+
         # If using HTTP Basic auth, add the user/pass to the session
         if auth_mode == "http-basic":
             session.auth = (username, password)