def handle_end_of_body(self, stream):
HttpClient.handle_end_of_body(self, stream)
context = stream.opaque
extra = context.extra
if extra['requests'] <= 0:
raise RuntimeError('runner_mlabns: unexpected response')
extra['requests'] -= 1
tmp = context.headers.get(CONTENT_TYPE)
if context.code != CODE200 or tmp != APPLICATION_JSON:
logging.error('runner_mlabns: bad response')
stream.close()
return
content = six.bytes_to_string(context.body.getvalue(), 'utf-8')
response = json.loads(content)
http_utils.prettyprint_json(response, '<')
if extra['policy'] == 'random':
RUNNER_HOSTS.set_random_host(response)
else:
RUNNER_HOSTS.set_closest_host(response)
stream.close()
def write(self, data):
''' Write data on standard output '''
# Remember that with Python 3 we need to decode data
sys.stdout.write(six.bytes_to_string(data, 'utf-8'))
