Skip to content
Snippets Groups Projects
Commit f4648d46 authored by Cresson Remi's avatar Cresson Remi
Browse files

Update 2 files

- /setup.py
- /dinamis_sdk/auth.py
parent c59cc354
No related branches found
No related tags found
No related merge requests found
Pipeline #189138 passed with warnings
...@@ -251,20 +251,19 @@ class OAuth2Session: ...@@ -251,20 +251,19 @@ class OAuth2Session:
class TokenServer: class TokenServer:
def __init__(self, endpoint: str, total_retry=5, backoff_factor=0.8): def __init__(self, endpoint: str, total_retry=5, backoff_factor=0.8):
session = requests.Session() self.session = requests.Session()
retry = urllib3.util.retry.Retry( retry = urllib3.util.retry.Retry(
total=total_retry, total=total_retry,
backoff_factor=backoff_factor, backoff_factor=backoff_factor,
status_forcelist=[404, 429, 500, 502, 503, 504], status_forcelist=[404, 429, 500, 502, 503, 504],
) )
adapter = requests.adapters.HTTPAdapter(max_retries=retry) adapter = requests.adapters.HTTPAdapter(max_retries=retry)
session.mount("https://", adapter) self.session.mount("https://", adapter)
session.mount("http://", adapter) self.session.mount("http://", adapter)
self.endpoint = endpoint log.info("Using Token Server: %s", self.endpoint)
log.info("Using Token Server: %s", endpoint)
def get_access_token() -> str: def get_access_token(self) -> str:
return session.get(self.endpoint, timeout=10).json() return self.session.get(self.endpoint, timeout=10).json()
session = TokenServer(TOKEN_SERVER) if TOKEN_SERVER else OAuth2Session() session = TokenServer(TOKEN_SERVER) if TOKEN_SERVER else OAuth2Session()
......
...@@ -13,7 +13,7 @@ install_requires = [ ...@@ -13,7 +13,7 @@ install_requires = [
setup( setup(
name="dinamis-sdk", name="dinamis-sdk",
version="0.1.5", version="0.1.6",
description="DINAMIS SDK", description="DINAMIS SDK",
python_requires=">=3.8", python_requires=">=3.8",
author="Remi Cresson", author="Remi Cresson",
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment