예제 #1
0
 def __init__(self,
              host,
              username,
              password,
              identifier,
              validate_ssl=True,
              timeout=10,
              max_retries=5,
              auth=None):
     basic_token = utils.generate_http_basic_token(username=username,
                                                   password=password)
     self.request_params = {
         'headers': {
             'X-Requested-By': identifier,
             'Authorization': 'Basic {}'.format(basic_token)
         },
         #'auth': (username, password),
         'verify': validate_ssl,
         'timeout': timeout,
     }
     # automatically retry requests on connection errors
     self.session = requests.Session()
     self.session.auth = auth
     adapter = requests.adapters.HTTPAdapter(max_retries=max_retries)
     self.session.mount(host, adapter)
예제 #2
0
 def __init__(self,
              host,
              identifier,
              username=None,
              password=None,
              oidc_token=None,
              validate_ssl=True,
              timeout=10,
              max_retries=5,
              auth=None):
     if oidc_token:
         auth_header = f'Bearer {oidc_token}'
     elif username and password:
         basic_token = utils.generate_http_basic_token(username=username,
                                                       password=password)
         auth_header = f'Basic {basic_token}'
     else:
         raise BadHttpAuthArg
     self.request_params = {
         'headers': {
             'X-Requested-By': identifier,
             'Authorization': auth_header
         },
         'verify': validate_ssl,
         'timeout': timeout,
     }
     # automatically retry requests on connection errors
     self.session = requests.Session()
     self.session.auth = auth
     adapter = requests.adapters.HTTPAdapter(max_retries=max_retries)
     self.session.mount(host, adapter)