Browse Source

Make requests' Session persistent in ConnectionManager

Signed-off-by: Çağrı Ulaş <cagriulas@gmail.com>
hotfix/merge
Çağrı Ulaş 6 years ago
parent
commit
b5c36b56bd
  1. 45
      keycloak/connection.py

45
keycloak/connection.py

@ -45,6 +45,7 @@ class ConnectionManager(object):
self._headers = headers self._headers = headers
self._timeout = timeout self._timeout = timeout
self._verify = verify self._verify = verify
self._s = requests.Session()
@property @property
def base_url(self): def base_url(self):
@ -134,11 +135,11 @@ class ConnectionManager(object):
""" """
try: try:
return requests.get(urljoin(self.base_url, path),
params=kwargs,
headers=self.headers,
timeout=self.timeout,
verify=self.verify)
return self._s.get(urljoin(self.base_url, path),
params=kwargs,
headers=self.headers,
timeout=self.timeout,
verify=self.verify)
except Exception as e: except Exception as e:
raise KeycloakConnectionError( raise KeycloakConnectionError(
"Can't connect to server (%s)" % e) "Can't connect to server (%s)" % e)
@ -154,12 +155,12 @@ class ConnectionManager(object):
HttpError: Can't connect to server. HttpError: Can't connect to server.
""" """
try: try:
return requests.post(urljoin(self.base_url, path),
params=kwargs,
data=data,
headers=self.headers,
timeout=self.timeout,
verify=self.verify)
return self._s.post(urljoin(self.base_url, path),
params=kwargs,
data=data,
headers=self.headers,
timeout=self.timeout,
verify=self.verify)
except Exception as e: except Exception as e:
raise KeycloakConnectionError( raise KeycloakConnectionError(
"Can't connect to server (%s)" % e) "Can't connect to server (%s)" % e)
@ -175,12 +176,12 @@ class ConnectionManager(object):
HttpError: Can't connect to server. HttpError: Can't connect to server.
""" """
try: try:
return requests.put(urljoin(self.base_url, path),
params=kwargs,
data=data,
headers=self.headers,
timeout=self.timeout,
verify=self.verify)
return self._s.put(urljoin(self.base_url, path),
params=kwargs,
data=data,
headers=self.headers,
timeout=self.timeout,
verify=self.verify)
except Exception as e: except Exception as e:
raise KeycloakConnectionError( raise KeycloakConnectionError(
"Can't connect to server (%s)" % e) "Can't connect to server (%s)" % e)
@ -196,11 +197,11 @@ class ConnectionManager(object):
HttpError: Can't connect to server. HttpError: Can't connect to server.
""" """
try: try:
return requests.delete(urljoin(self.base_url, path),
params=kwargs,
headers=self.headers,
timeout=self.timeout,
verify=self.verify)
return self._s.delete(urljoin(self.base_url, path),
params=kwargs,
headers=self.headers,
timeout=self.timeout,
verify=self.verify)
except Exception as e: except Exception as e:
raise KeycloakConnectionError( raise KeycloakConnectionError(
"Can't connect to server (%s)" % e) "Can't connect to server (%s)" % e)
Loading…
Cancel
Save