def __init__(self, url: str, token_url: str, tokens: list):
self.url = url
self.token_url = token_url
self.tokens = tokens
self.terminal_size = self._get_size_terminal()
self.token_generator = TokenGenerator(token_url)
self.SET_MAX_RETRY = 3
def __init__(self, endpoint_config: dict, environment: str, tokens: list):
self.endpoint_config = endpoint_config
self.environment = environment
self.requests_ssl_verify = suppress_insecure_request_warns(environment)
self.url = self._build_url(endpoint_config, environment)
self.tokens = tokens
self.terminal_size = self._get_size_terminal()
self.token_generator = TokenGenerator(endpoint_config,
environment=environment)
self.headers = None
self.SET_MAX_RETRY = 3
def __init__(self, endpoint_config: dict, environment: str, tokens: list):
self.endpoint_config = endpoint_config
self.environment = environment
self.url = self._build_url(endpoint_config, environment)
self.tokens = tokens
self.terminal_size = self._get_size_terminal()
self.token_generator = TokenGenerator(endpoint_config,
environment=environment)
self.headers = None
self.SET_MAX_RETRY = 3
def load_config(self, args):
"""
Load correct config and generate first tokens
:return (dict, str, list<str, str>)
"""
configure_logging(args.loglevel)
endpoint_config = self._load_config(self._CONFIG_ENDPOINTS)
main_url = endpoint_config['main'][args.env]
token_generator = TokenGenerator(endpoint_config, environment=args.env)
token_json = token_generator.get_token()
if token_json:
tokens = [f'Token {token_json["access_token"]}', f'Token {token_json["refresh_token"]}']
return endpoint_config, main_url, tokens
else:
logger.error("Couldn't generate Tokens, please check the login/password provided")
exit()
def test_auth():
engine = TokenGenerator(TEST_CONFIG, environment=TEST_ENV)
assert engine.url_token == 'https://datalake.com/api/v42/auth/token/'