Add authorization token to use GCP IAP

Setting the header `Proxy-Authorization` does not work because it is deleted: https://github.com/psf/requests/blob/2a438c27b5a5828c8ea0dc958112eecffca70b12/src/requests/sessions.py#L318-L319.
google · Apr 5, 2024 · 993985f · 993985f
1 parent 2ceeb13
commit 993985f
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)