httpapi: Split off and save response text contents (#43305)

* Split off and save response text contents

* login might depend on httpapi
This commit is contained in:
Nathaniel Case 2018-07-26 10:12:34 -04:00 committed by GitHub
parent 13dd1acbec
commit ab40c72a3c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 22 additions and 17 deletions

View file

@ -206,12 +206,12 @@ class Connection(NetworkConnectionBase):
httpapi = httpapi_loader.get(self._network_os, self)
if httpapi:
display.vvvv('loaded API plugin for network_os %s' % self._network_os, host=self._play_context.remote_addr)
self._implementation_plugins.append(httpapi)
httpapi.set_become(self._play_context)
httpapi.login(self.get_option('remote_user'), self.get_option('password'))
display.vvvv('loaded API plugin for network_os %s' % self._network_os, host=self._play_context.remote_addr)
else:
raise AnsibleConnectionFailure('unable to load API plugin for network_os %s' % self._network_os)
self._implementation_plugins.append(httpapi)
cliconf = cliconf_loader.get(self._network_os, self)
if cliconf:
@ -258,7 +258,9 @@ class Connection(NetworkConnectionBase):
return self.send(path, data, **kwargs)
raise AnsibleConnectionFailure('Could not connect to {0}: {1}'.format(self._url, exc.reason))
# Try to assign a new auth token if one is given
self._auth = self.update_auth(response) or self._auth
response_text = response.read()
return response
# Try to assign a new auth token if one is given
self._auth = self.update_auth(response, response_text) or self._auth
return response, response_text

View file

@ -36,7 +36,7 @@ class HttpApiBase(AnsiblePlugin):
"""
pass
def update_auth(self, response):
def update_auth(self, response, response_text):
"""Return per-request auth token.
The response should be a dictionary that can be plugged into the

View file

@ -30,14 +30,13 @@ class HttpApi(HttpApiBase):
request = request_builder(data, output)
headers = {'Content-Type': 'application/json-rpc'}
response = self.connection.send('/command-api', request, headers=headers, method='POST')
response_text = to_text(response.read())
response, response_text = self.connection.send('/command-api', request, headers=headers, method='POST')
try:
response = json.loads(response_text)
response_text = json.loads(response_text)
except ValueError:
raise ConnectionError('Response was not valid JSON, got {0}'.format(response_text))
results = handle_response(response)
results = handle_response(response_text)
if self._become:
results = results[1:]
@ -50,8 +49,7 @@ class HttpApi(HttpApiBase):
# Fake a prompt for @enable_mode
if self._become:
return '#'
else:
return '>'
return '>'
# Imported from module_utils
def edit_config(self, config, commit=False, replace=False):
@ -113,7 +111,13 @@ class HttpApi(HttpApiBase):
responses = list()
def run_queue(queue, output):
response = to_list(self.send_request(queue, output=output))
try:
response = to_list(self.send_request(queue, output=output))
except Exception as exc:
if check_rc:
raise
return to_text(exc)
if output == 'json':
response = [json.loads(item) for item in response]
return response

View file

@ -27,14 +27,13 @@ class HttpApi(HttpApiBase):
request = request_builder(queue, output)
headers = {'Content-Type': 'application/json'}
response = self.connection.send('/ins', request, headers=headers, method='POST')
response_text = to_text(response.read())
response, response_text = self.connection.send('/ins', request, headers=headers, method='POST')
try:
response = json.loads(response_text)
response_text = json.loads(response_text)
except ValueError:
raise ConnectionError('Response was not valid JSON, got {0}'.format(response_text))
results = handle_response(response)
results = handle_response(response_text)
if self._become:
results = results[1:]