Merge pull request Peter-Slump#32 from gt-novelt/auto-refresh-admin-token

Auto refresh admin token

Author: marcospereirampj

keycloak/connection.py

@@ -27,6 +27,7 @@
     from urlparse import urljoin
 
 import requests
+from requests.adapters import HTTPAdapter
 
 from .exceptions import (KeycloakConnectionError)
 
@@ -47,6 +48,17 @@ def __init__(self, base_url, headers={}, timeout=60, verify=True):
         self._verify = verify
         self._s = requests.Session()
 
+        # retry once to reset connection with Keycloak after  tomcat's ConnectionTimeout
+        # see https://github.com/marcospereirampj/python-keycloak/issues/36
+        for protocol in ('https://', 'http://'):
+            adapter = HTTPAdapter(max_retries=1)
+            # adds POST to retry whitelist
+            method_whitelist = set(adapter.max_retries.method_whitelist)
+            method_whitelist.add('POST')
+            adapter.max_retries.method_whitelist = frozenset(method_whitelist)
+
+            self._s.mount(protocol, adapter)
+
     @property
     def base_url(self):
         """ Return base url in use for requests to the server. """