def acquire_token(self):
"""Acquire user token"""
logger = self.logger(self.acquire_token.__name__)
logger.debug('called')
try:
logger.debug("Acquiring Access Token..")
try:
from urlparse import urlparse # Python 2.X
except ImportError:
from urllib.parse import urlparse # Python 3+
url = urlparse(self.url)
options = {
'username': self.username,
'password': self.password,
'sts': self.sts,
'endpoint': url.scheme + '://' + url.hostname + self.login
}
self.acquire_service_token(options)
self.acquire_authentication_cookie(options)
return True
except requests.exceptions.RequestException as e:
self.error = "Error: {}".format(e)
return False
def acquire_token(self, **kwargs):
"""Acquire user token
"""
logger = self.logger(self.acquire_token.__name__)
logger.debug('acquire_token called')
try:
logger.debug("Acquiring Access Token..")
user_realm = self._get_user_realm(self.__username)
if user_realm.IsFederated:
token = self.acquire_service_token_from_adfs(
user_realm.STSAuthUrl, self.__username, self.__password)
else:
token = self._acquire_service_token(self.__username,
self.__password)
return self._acquire_authentication_cookie(token,
user_realm.IsFederated)
except requests.exceptions.RequestException as e:
self.error = "Error: {}".format(e)
return False
def get_authentication_cookie(self):
"""Acquire authentication cookie
"""
logger = self.logger(self.ensure_authentication_cookie.__name__)
logger.debug('get_authentication_cookie called')
try:
logger.debug("Acquiring Access Token..")
user_realm = self._get_user_realm()
if user_realm.IsFederated:
token = self._acquire_service_token_from_adfs(
user_realm.STSAuthUrl)
else:
token = self._acquire_service_token()
return self._get_authentication_cookie(token,
user_realm.IsFederated)
except requests.exceptions.RequestException as e:
logger.error(e.response.text)
self.error = "Error: {}".format(e)
raise ValueError(e.response.text)
