feat: Authorization header can now be customized in AuthenticatedClient [#660]. Thanks @supermihi!

Co-authored-by: Dylan Anthony <[email protected]>

Author: supermihi

end_to_end_tests/golden-record/my_test_api_client/client.py

@@ -42,7 +42,10 @@ class AuthenticatedClient(Client):
     """A Client which has been authenticated for use on secured endpoints"""
 
     token: str
+    prefix: str = "Bearer"
+    auth_header_name: str = "Authorization"
 
     def get_headers(self) -> Dict[str, str]:
+        auth_header_value = f"{self.prefix} {self.token}" if self.prefix else self.token
         """Get headers to be used in authenticated endpoints"""
-        return {"Authorization": f"Bearer {self.token}", **self.headers}
+        return {self.auth_header_name: auth_header_value, **self.headers}

integration-tests/integration_tests/client.py

@@ -42,7 +42,10 @@ class AuthenticatedClient(Client):
     """A Client which has been authenticated for use on secured endpoints"""
 
     token: str
+    prefix: str = "Bearer"
+    auth_header_name: str = "Authorization"
 
     def get_headers(self) -> Dict[str, str]:
+        auth_header_value = f"{self.prefix} {self.token}" if self.prefix else self.token
         """Get headers to be used in authenticated endpoints"""
-        return {"Authorization": f"Bearer {self.token}", **self.headers}
+        return {self.auth_header_name: auth_header_value, **self.headers}

openapi_python_client/templates/client.py.jinja

@@ -39,7 +39,10 @@ class AuthenticatedClient(Client):
     """ A Client which has been authenticated for use on secured endpoints """
 
     token: str
+    prefix: str = "Bearer"
+    auth_header_name: str = "Authorization"
 
     def get_headers(self) -> Dict[str, str]:
-        """ Get headers to be used in authenticated endpoints """
-        return {"Authorization": f"Bearer {self.token}", **self.headers}
+        auth_header_value = f"{self.prefix} {self.token}" if self.prefix else self.token
+        """Get headers to be used in authenticated endpoints"""
+        return {self.auth_header_name: auth_header_value, **self.headers}