refactor: make session auth to use base service's http client

Author: eiri

ibmcloudant/cloudant_base_service.py

@@ -98,7 +98,8 @@ def __init__(
         if isinstance(authenticator, CouchDbSessionAuthenticator):
             # Replacing BaseService's http.cookiejar.CookieJar as RequestsCookieJar supports update(CookieJar)
             self.jar = RequestsCookieJar(self.jar)
-            self.authenticator.set_jar(self.jar)  # Authenticators don't have access to cookie jars by default
+            # Make token manager of CouchDbSessionAuthenticator to use the same http client as main service
+            self.authenticator.set_http_client(self.get_http_client())
         add_hooks(self)
 
     def set_service_url(self, service_url: str):

ibmcloudant/couchdb_session_authenticator.py

@@ -1,6 +1,6 @@
 # coding: utf-8
 
-# © Copyright IBM Corporation 2020, 2022.
+# © Copyright IBM Corporation 2020, 2025.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,7 +16,7 @@
 """
 Module for handling session authentication
 """
-from requests import Request
+from requests import Request, Session
 
 from ibm_cloud_sdk_core.authenticators import Authenticator
 from .couchdb_session_token_manager import CouchDbSessionTokenManager
@@ -46,20 +46,21 @@ def __init__(self,
         if not isinstance(disable_ssl_verification, bool):
             raise TypeError('disable_ssl_verification must be a bool')
 
-        self.jar = None
-
         self.token_manager = CouchDbSessionTokenManager(
             username,
             password,
             disable_ssl_verification=disable_ssl_verification
         )
         self.validate()
 
-    def set_jar(self, jar):
-        """Sets the cookie jar for the authenticator.
+    def set_http_client(self, http_client: Session) -> None:
+        """Sets base serivice's http client for the authenticator.
         This is an internal method called by BaseService. Not to be called directly.
         """
-        self.jar = jar
+        if isinstance(http_client, Session):
+            self.token_manager.http_client = http_client
+        else:
+            raise TypeError("http_client parameter must be a requests.sessions.Session")
 
     def validate(self):
         """Validates the username, and password for session token requests.
@@ -82,11 +83,7 @@ def authenticate(self, req: Request):
         Args:
             req: Ignored. BaseService uses the cookie jar for every request
         """
-        jar = self.token_manager.get_token()
-        # Requests seem to save cookies only for Sessions. BaseService is
-        # hard-coded to work with "regular" requests requests so updating
-        # the jar manually is necessary
-        self.jar.update(jar)
+        self.token_manager.get_token()
 
     def authentication_type(self) -> str:
         """Returns this authenticator's type ('COUCHDB_SESSION')."""

ibmcloudant/couchdb_session_token_manager.py

@@ -1,6 +1,6 @@
 # coding: utf-8
 
-# © Copyright IBM Corporation 2020, 2022.
+# © Copyright IBM Corporation 2020, 2025.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,6 +16,8 @@
 """
 Module for managing session authentication token
 """
+from requests import Session
+
 from ibm_cloud_sdk_core.token_managers.token_manager import TokenManager
 
 
@@ -52,6 +54,7 @@ def __init__(self, username: str, password: str,
 
         self.token = None
 
+        self.http_client = Session()
         self.http_config = {}
         self.headers = None
 
@@ -63,7 +66,7 @@ def request_token(self):
              A CookieJar of Cookies the server sent back.
         """
 
-        response = self._request(
+        response = self.http_client.request(
             method='POST',
             url=self.url + "/_session",
             headers=self.headers,