Merge pull request #101 from RomanSemkin/using_cookies_for_sdk

ability to use cookies for sdk was added

Author: cuongssh

privx_api/base.py

@@ -70,6 +70,7 @@ def __init__(
         oauth_client_id: str,
         oauth_client_secret: str,
         re_auth_margin: int = 3,
+        use_cookies=False,
     ) -> None:
         self._access_token = ""
         self._oauth_client_id = oauth_client_id
@@ -84,6 +85,8 @@ def __init__(
         self._re_auth_deadline = None
         self._access_token_age = None
         self._re_auth_margin = re_auth_margin
+        self._use_cookies = use_cookies
+        self._cookies = None
 
     def _authenticate(self, username: str, password: str) -> None:
         # saving the creds for the re-auth purposes
@@ -122,6 +125,11 @@ def _authenticate(self, username: str, password: str) -> None:
             except (JSONDecodeError, TypeError) as e:
                 raise InternalAPIException(e) from e
 
+            # save and never change cookie`s value,
+            # in order to communicate with the same node
+            if self._cookies is None and self._use_cookies:
+                self._cookies = response.getheader("Set-Cookie")
+
             # privx response includes access token age in seconds
             self._access_token_age = data.get("expires_in")
             self._re_auth_deadline = (
@@ -174,6 +182,8 @@ def _get_headers(self) -> dict:
         }
         if self._access_token:
             headers["Authorization"] = "Bearer {}".format(self._access_token)
+        if self._cookies:
+            headers["Cookie"] = self._cookies
         return headers
 
     def _get_search_params(self, **kwargs: Union[str, int]) -> dict: