def _did_receive_response(self, response):
""" Called when a response is received """
try:
data = response.json()
except:
data = None
self._response = NURESTResponse(status_code=response.status_code,
headers=response.headers,
data=data,
reason=response.reason)
level = logging.WARNING if self._response.status_code >= 300 else logging.DEBUG
bambou_logger.info(
'< %s %s %s [%s] ' %
(self._request.method, self._request.url, self._request.params
if self._request.params else "", self._response.status_code))
bambou_logger.log(level, '< headers: %s' % self._response.headers)
bambou_logger.log(
level, '< data:\n%s' % json.dumps(self._response.data, indent=4))
self._callback(self)
return self
def _did_receive_response(self, connection):
""" Receive a response from the connection """
if connection.has_timeouted:
bambou_logger.info("NURESTConnection has timeout.")
return
has_callbacks = connection.has_callbacks()
should_post = not has_callbacks
if connection.handle_response_for_connection(should_post=should_post) and has_callbacks:
callback = connection.callbacks['local']
callback(connection)
def _did_receive_response(self, connection):
""" Receive a response from the connection """
if connection.has_timeouted:
bambou_logger.info("NURESTConnection has timeout.")
return
has_callbacks = connection.has_callbacks()
should_post = not has_callbacks
if connection.handle_response_for_connection(should_post=should_post) and has_callbacks:
callback = connection.callbacks['local']
callback(connection)
def _make_request(self, session=None):
""" Make a synchronous request """
if session is None:
session = NURESTSession.get_current_session()
self._has_timeouted = False
# Add specific headers
controller = session.login_controller
enterprise = controller.enterprise
user_name = controller.user
api_key = controller.api_key
certificate = controller.certificate
if self._root_object:
enterprise = self._root_object.enterprise_name
user_name = self._root_object.user_name
api_key = self._root_object.api_key
if self._uses_authentication:
self._request.set_header('X-Nuage-Organization', enterprise)
self._request.set_header(
'Authorization',
controller.get_authentication_header(user_name, api_key))
if controller.is_impersonating:
self._request.set_header('X-Nuage-ProxyUser',
controller.impersonation)
headers = self._request.headers
data = json.dumps(self._request.data)
bambou_logger.info(
'> %s %s %s' %
(self._request.method, self._request.url,
self._request.params if self._request.params else ""))
bambou_logger.debug('> headers: %s' % headers)
bambou_logger.debug('> data:\n %s' %
json.dumps(self._request.data, indent=4))
response = self.__make_request(
requests_session=session.requests_session,
method=self._request.method,
url=self._request.url,
params=self._request.params,
data=data,
headers=headers,
certificate=certificate)
retry_request = False
if response.status_code == HTTP_CODE_MULTIPLE_CHOICES and 'responseChoice' not in self._request.url:
if '?' in self._request.url:
self._request.url += '&responseChoice=1'
else:
self._request.url += '?responseChoice=1'
bambou_logger.debug(
'Bambou got [%s] response. Trying to force response choice' %
HTTP_CODE_MULTIPLE_CHOICES)
retry_request = True
elif response.status_code == HTTP_CODE_AUTHENTICATION_EXPIRED and session:
bambou_logger.debug(
'Bambou got [%s] response . Trying to reconnect your session that has expired'
% HTTP_CODE_AUTHENTICATION_EXPIRED)
session.reset()
session.start()
retry_request = True
if retry_request:
bambou_logger.info(
'> %s %s %s' %
(self._request.method, self._request.url,
self._request.params if self._request.params else ""))
bambou_logger.debug('> headers: %s' % headers)
bambou_logger.debug('> data:\n %s' %
json.dumps(self._request.data, indent=4))
response = self.__make_request(
requests_session=session.requests_session,
method=self._request.method,
url=self._request.url,
params=self._request.params,
data=data,
headers=headers,
certificate=certificate)
return self._did_receive_response(response)